tests/testthat/test-redshift-queries.R

In Financial-Times/ftRtools: FT R Data Science Tools

context("Running queries")

test_that("select queries run on int", {
  query_result <- redshift_query("simple_query.sql",
                                 "int",
                                 Sys.getenv("redshift_username"),
                                 Sys.getenv("redshift_password"))
  expect_equal(is.data.frame(query_result), TRUE)
  expect_gt(nrow(query_result), 0)
})

test_that("select queries run on prod", {
  query_result <- redshift_query("simple_query.sql",
                                 "prod",
                                 Sys.getenv("prod_username"),
                                 Sys.getenv("prod_password"))
  expect_equal(is.data.frame(query_result), TRUE)
  expect_gt(nrow(query_result), 0)
})

test_that("column names are preserved", {
  query_result <- redshift_query("simple_query.sql",
                                 "int",
                                 Sys.getenv("redshift_username"),
                                 Sys.getenv("redshift_password"))
  expect_equal(names(query_result), c("tabs"))
})

test_that("query as string runs on int", {
  q <- readr::read_file(here::here("tests","testthat","simple_query.sql"))
  query_result <- redshift_query(q,
                                 "int",
                                 Sys.getenv("redshift_username"),
                                 Sys.getenv("redshift_password"),
                                 from_file = FALSE)
  expect_equal(is.data.frame(query_result), TRUE)
  expect_gt(nrow(query_result), 0)
})