context("cdc")

library(jsonlite)


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)
  local_mocked_bindings(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_from_json <- mock()
  mock_read_csv <- mock()
  local_mocked_bindings(re_authenticate_if_necessary = function(...) { })
  local_mocked_bindings(fromJSON = mock_from_json)
  local_mocked_bindings(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_from_json <- mock()
  mock_read_csv <- mock()
  local_mocked_bindings(re_authenticate_if_necessary = function(...) { })
  local_mocked_bindings(fromJSON = mock_from_json)
  local_mocked_bindings(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
})