tests/testthat/test-duckdb.R

In bcgov/rems: Get Data from British Columbia's Environmental Monitoring System

library(dplyr)

test_that("duckdb is created from a csv file", {
  expect_true(write_test_db())
})

test_that("connecting and attaching historic duckdb works",{
  con <- connect_historic_db(write_db_path())
  on.exit(disconnect_historic_db(con))
  tbl <- attach_historic_data(con)
  expect_s3_class(tbl, "tbl_duckdb_connection")
  collected <- collect(tbl)
  ref <- read_ems_data("test_historic.csv")
  expect_equal(dim(collected), dim(ref))
  expect_equal(names(collected), names(ref))

  # Test columns including time zones
  expect_equal(attributes(collected$COLLECTION_START)[["tzone"]], "Etc/GMT+8")
  expect_equal(attributes(collected$COLLECTION_END)[["tzone"]], "Etc/GMT+8")
  expect_equal(collected, ref)
})

test_that("read_historic_data works", {
  dat <- read_historic_data()
  expect_s3_class(dat, "data.frame")
  expect_equal(dim(dat), c(10L, 24L))

  dat <- read_historic_data(emsid = "0400034", from_date = as.Date("1975-07-27"),
                     to_date = as.Date("1975-07-29"), cols = "all", check_db = FALSE)
  expect_s3_class(dat, "data.frame")
  expect_equal(dim(dat), c(2L, 60L))
  expect_equal(attributes(dat$COLLECTION_START)[["tzone"]], "Etc/GMT+8")
  expect_equal(attributes(dat$COLLECTION_END)[["tzone"]], "Etc/GMT+8")
})

test_that("removing historic works works", {
  expect_equal(._remsenv_$cache$list(), c("2yr", "4yr", "cache_dates"))
  remove_data_cache("historic")
  withr::defer(write_test_db())
  expect_error(connect_historic_db(write_db_path()), "Please download the historic data")
})

test_that("download_historic_data works", {
  expect_message(ret <- download_historic_data(force = FALSE, ask = FALSE),
                 "you already have the most up-to date version")
  expect_equal(ret, write_db_path())

  set_cache_date("historic", as.POSIXct("2020-12-31"))
  withr::defer(set_cache_date("historic", Sys.time() + 1))

  expect_warning(ret <- download_historic_data(force = FALSE, ask = FALSE, dont_update = TRUE),
                 "you have asked not to update")
  expect_equal(ret, write_db_path())
})