tests/testthat/test-termvectors.R

In elastic: General Purpose Interface to 'Elasticsearch'

context("termvectors")

x <- connect(port = Sys.getenv("TEST_ES_PORT"), warn = FALSE)

if (!index_exists(x, 'omdb')) {
  omdb <- system.file("examples", "omdb.json", package = "elastic")
  invisible(docs_bulk(x, omdb))
}

test_that("termvectors works", {
  skip_on_travis()
  if (gsub("\\.", "", x$ping()$version$number) < 130) skip('feature not in this ES version')

  body <- '{
    "fields" : ["Plot"],
    "offsets" : true,
    "positions" : true,
    "term_statistics" : true,
    "field_statistics" : true
  }'

  id <- vapply(Search(x, "omdb", size = 1)$hits$hits, "[[", "", "_id")
  aa <- termvectors(x, 'omdb', 'omdb', id, body = body)


  expect_is(aa, 'list')
  expect_equal(aa$`_index`, "omdb")
  expect_equal(aa$`_type`, "omdb")
  expect_is(aa$`_id`, "character")

  expect_is(aa$term_vectors, "list")
  expect_named(aa$term_vectors, 'Plot')
  expect_named(aa$term_vectors$Plot, c('field_statistics', 'terms'))
  expect_is(aa$term_vectors$Plot$field_statistics, "list")
  expect_is(aa$term_vectors$Plot$terms, "list")
  expect_is(aa$term_vectors$Plot$terms[[1]], "list")
  expect_named(aa$term_vectors$Plot$terms[[1]]$tokens[[1]], c('position', 'start_offset', 'end_offset'))
})

test_that("termvectors fails well", {
  skip_on_travis()
  if (gsub("\\.", "", x$ping()$version$number) < 130) skip('feature not in this ES version')

  expect_error(termvectors(x), "argument \"index\" is missing")
  # expect_error(termvectors(x, "omdb"), "argument \"type\" is missing")
  expect_error(termvectors(x, "omdb", "omdb"), "Validation Failed")

  body <- '{
     "fields" : ["Plot"],
     "offsets" : true,
     "positions" : true,
     "term_statistics" : true,
    "field_statistics" : true
  }'

  expect_error(termvectors(x, 'omdb', 'omdb', body = body),
               "Validation Failed")
  expect_equal(length(termvectors(x, 'omdb', 'omdb', 'AVXdx8Eqg_0Z_tpMDyP_')$term_vectors),
               0)
})