tests/testthat/test-setup-download.R

In septic-tank/ricu: Intensive Care Unit Data with R

curl_mock_fm <- function(url, handle) {

  file <- system.file("testdata", paste0(basename(url), ".rds"),
                      package = "ricu")

  if (identical(file, "")) stop("file ", basename(url), "not found")

  readRDS(file)
}

curl_mock_fd <- function(url, path, handle) {

  dat <- curl_mock_fm(url, handle)

  # if e.g. timecondition opt can be queried from handle, a 304 could be
  # returned to mimic the no-change scenario

  writeBin(dat[["content"]], path)

  dat
}

tmp <- tempfile()

setup(dir.create(tmp))
teardown(unlink(tmp, recursive = TRUE))

test_that("file download", {

  dat_mem <- with_mock(
    `curl::curl_fetch_memory` = curl_mock_fm,
    `curl::curl_fetch_disk` = function(...) stop("error"),
    `curl::curl_fetch_stream` = function(...) stop("error"),
    download_pysionet_file("foo/bar/SHA256SUMS.txt")
  )

  expect_is(dat_mem, "raw")
  expect_length(list.files(tmp), 0L)

  dat_disk <- with_mock(
    `curl::curl_fetch_memory` = function(...) stop("error"),
    `curl::curl_fetch_disk` = curl_mock_fd,
    `curl::curl_fetch_stream` = function(...) stop("error"),
    download_pysionet_file("foo/bar/SHA256SUMS.txt",
                           file.path(tmp, "SHA256SUMS.txt"))
  )

  expect_is(dat_disk, "NULL")
  expect_length(list.files(tmp, "SHA256SUMS.txt"), 1L)
})

test_that("hash checking", {

  sha <- with_mock(
    `curl::curl_fetch_memory` = curl_mock_fm,
    `curl::curl_fetch_disk` = function(...) stop("error"),
    `curl::curl_fetch_stream` = function(...) stop("error"),
    get_sha256("foo/bar")
  )

  expect_is(sha, "list")
  for (x in sha) {
    expect_is(x, "character")
    expect_length(x, 2L)
  }

  res <- with_mock(
    `curl::curl_fetch_memory` = function(...) stop("error"),
    `curl::curl_fetch_disk` = curl_mock_fd,
    `curl::curl_fetch_stream` = function(...) stop("error"),
    download_pysionet_file("foo/bar/patients.csv",
                           file.path(tmp, sha[[2L]][2L]))
  )

  expect_null(res)

  expect_true(
    check_file_sha256(file.path(tmp, sha[[2L]][2L]), sha[[2L]][1L])
  )
  expect_false(
    check_file_sha256(file.path(tmp, sha[[2L]][2L]), sha[[1L]][1L])
  )
})