inst/unitTests/test_SolrCore.R

In rsolr: R to Solr Interface

checkResponseIdentical <- function(response, input) {
  checkIdentical(unmeta(response), input)
}

checkResponseEquals <- function(response, input, tolerance=1) {
  response[,"price_c"] <- NULL
  checkEquals(unmeta(response), input, tolerance=tolerance)
}

test_SolrCore_accessors <- function() {
  solr <- TestSolr()
  sc <- SolrCore(solr$uri)
  checkIdentical(name(sc), "example")
  checkIdentical(ndoc(sc), 32L)
  checkIdentical(uniqueKey(schema(sc)), "id")
  checkIdentical(version(sc), package_version("5.5.0"))

  delete(sc)
  purgeCache(sc)
  checkIdentical(ndoc(sc), 0L)
  
  doc <- list(id="1112211111", name="my name!")
  dc <- as(list(doc), "DocCollection")
  update(sc, dc, commit=FALSE)
  q <- SolrQuery(id == .(doc$id))
  checkIdentical(read(sc, q), dc[NULL])

  checkIdentical(commit(sc), 0L)
  checkIdentical(read(sc, q), dc[NULL]) # still NULL for 30s because of cache
  purgeCache(sc)
  ids(dc) <- doc$id
  checkResponseIdentical(read(sc, q), dc)
  checkIdentical(ndoc(sc), length(dc))
  
  docs <- list(
    list(id="2", inStock=TRUE, price=2, timestamp_dt=Sys.time()),
    list(id="3", inStock=FALSE, price=3, timestamp_dt=Sys.time()),
    list(id="4", price=4, timestamp_dt=Sys.time()),
    list(id="5", inStock=FALSE, price=5, timestamp_dt=Sys.time())
    )

  update(sc, docs)
  fn <- c("id", "name", "price", "inStock", "text", "timestamp_dt", "price_c")
  checkIdentical(fieldNames(sc), fn)
  checkIdentical(fieldNames(sc, "*_c"), "price_c")
  checkIdentical(fieldNames(sc, "text*", includeStatic=TRUE),
                 c("text", "text_rev"))
  checkIdentical(fieldNames(sc, onlyStored=TRUE), setdiff(fn, "text"))
  unindexed <- setdiff(fieldNames(sc, includeStatic=TRUE),
                       fieldNames(sc, includeStatic=TRUE, onlyIndexed=TRUE))
  checkIdentical(unindexed, "content")
  checkIdentical(fieldNames(sc, character(0L)), character(0L))

  docs <- as(docs, "DocCollection")
  docs[,"timestamp_dt"] <- structure(docs[,"timestamp_dt"], tzone="UTC")
  ids(docs) <- docs[,"id"]
  docs <- docs[,fieldNames(sc)]

  q <- SolrQuery(id %in% as.character(2:4))
  checkIdentical(ndoc(sc, q), 3L)
  checkResponseEquals(read(sc, q), docs[1:3])

  q <- SolrQuery(id == .(doc$id))
  delete(sc, q)
  checkResponseEquals(read(sc), docs)
  del <- list()
  del[as.character(2:4)] <- list(NULL)
  update(sc, del)
  purgeCache(sc)
  checkResponseEquals(read(sc), docs[4])
  delete(sc)
  purgeCache(sc)
  checkResponseEquals(read(sc), new("DocList"))

  docs[,"id"] <- NULL
  update(sc, docs)
  docs[,"id"] <- ids(docs)
  purgeCache(sc)
  docs <- docs[,fieldNames(sc)]
  checkResponseEquals(read(sc), docs)

  del <- c(del, list(doc))
  update(sc, del)
  purgeCache(sc)
  checkResponseEquals(read(sc), c(dc, docs["5"]))
}