context("cdc")
library(jsonlite)
library(testthat)
library(zoltr)
test_that("job_data() calls are correct", {
zoltar_connection <- new_connection("http://example.com")
cdc_file <- "data/20161023-KoTstable-20161109-small.cdc.csv" # file doesn't matter for this test, but does below
exp_data_frame <- utils::read.csv(cdc_file, stringsAsFactors = FALSE, colClasses = 'character') # "NA" -> NA
mock_reauthenticate <- mock(exp_data_frame)
testthat::with_mock("zoltr::get_resource" = mock_reauthenticate, {
act_data_frame <- job_data(zoltar_connection, "http://example.com/api/job/2/", "forecasts")
expect_equal(length(mock_calls(mock_reauthenticate)), 1)
expect_equal(mock_args(mock_reauthenticate)[[1]][[2]], "http://example.com/api/job/2/data/")
expect_is(act_data_frame, "data.frame")
expect_equal(act_data_frame, exp_data_frame)
})
})
test_that("job_data() column types are correct for forecast CSV data", { # JSON data tested many other places
zoltar_connection <- new_connection("http://example.com")
mock_reauthenticate <- mock()
mock_from_json <- mock()
mock_read_csv <- mock()
testthat::with_mock("zoltr::re_authenticate_if_necessary" = mock_reauthenticate, {
testthat::with_mock("jsonlite::fromJSON" = mock_from_json, {
testthat::with_mock("readr::read_csv" = mock_read_csv, {
webmockr::stub_request("get", uri = "http://example.com/api/job/2/data/") %>%
to_return(
status = 200,
headers = list('Content-Type' = 'text/csv'))
job_data(zoltar_connection, "http://example.com/api/job/2/", "forecasts")
expect_equal(length(mock_calls(mock_from_json)), 0)
expect_equal(length(mock_calls(mock_read_csv)), 1)
expect_equal(mock_args(mock_read_csv)[[1]]$col_types, "cDcccc????d????")
# note: we do not test the column types of the actual data returned by job_data b/c we trust that
# readr::read_csv correctly applies col_types
})
})
})
})
test_that("job_data() column types are correct for truth CSV data", { # JSON data tested many other places
zoltar_connection <- new_connection("http://example.com")
mock_reauthenticate <- mock()
mock_from_json <- mock()
mock_read_csv <- mock()
testthat::with_mock("zoltr::re_authenticate_if_necessary" = mock_reauthenticate, {
testthat::with_mock("jsonlite::fromJSON" = mock_from_json, {
testthat::with_mock("readr::read_csv" = mock_read_csv, {
webmockr::stub_request("get", uri = "http://example.com/api/job/2/data/") %>%
to_return(
status = 200,
headers = list('Content-Type' = 'text/csv'))
job_data(zoltar_connection, "http://example.com/api/job/2/", "truth")
expect_equal(length(mock_calls(mock_from_json)), 0)
expect_equal(length(mock_calls(mock_read_csv)), 1)
expect_equal(mock_args(mock_read_csv)[[1]]$col_types, "Dcc?")
# note: we do not test the column types of the actual data returned by job_data b/c we trust that
# readr::read_csv correctly applies col_types
})
})
})
})
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.