tests/testthat/test_tables.R

In civisanalytics/civis-r: R Client for the 'Civis Platform API'

context("tables")
library(civis)

options(civis.default_db = "sample_db")

test_that("transfer_table succeeds", {
  res <- with_mock(
    `civis::default_credential` = function(...) 1,
    `civis::get_database_id` = function(...) 32,
    `civis::imports_post` = function(...) list(id = 999),
    `civis::imports_post_syncs` = function(...) NULL,
    `civis::imports_post_runs` = function(...) list(runId = 999),
    `civis::imports_get_files_runs` = function(...) list(state = "succeeded"),
    transfer_table("db1", "db2", "sc.tb1", "sc.tb2")
  )
  expect_equal(get_status(res), "succeeded")
})

test_that("refresh_table succeeds", {
  with_mock(
    `civis::get_database_id` = function(...) 32,
    `civis::get_table_id` = function(...) 5,
    `civis::tables_post_refresh` = function(...) "",
    `civis::tables_get` = function(...) list(refreshStatus = "current"),
    expect_equal(get_status(refresh_table("sea_creatures.whales")), "current")
  )
})

test_that("get_table_id returns table id", {
   with_mock(
    `civis::get_database_id` = function(...) 32,
    `civis::tables_list` = function(...) list(list(name = "whales", id = 5)),
    expect_equal(get_table_id("sea_creatures.whales"), 5)
  )
})

test_that("get_table_id returns error if not found", {
  msg <- paste0("Table sea_creatures.squid not found.")
  with_mock(
    `civis::get_database_id` = function(...) 32,
    `civis::tables_list` = function(...) list(list(name = "whales", id = 5)),
    expect_error(get_table_id("sea_creatures.squid"), msg)
  )
})

test_that("get_table_id returns error if not schema.tablename", {
  expect_error(get_table_id("asdf"), "table name must be 'schema.table'")
})