Add a dummyPrestoConnection() function

prestodb · Jan 13, 2023 · 786b86a · 786b86a
1 parent 2883624
commit 786b86a
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 2 deletions.
diff --git a/NAMESPACE b/NAMESPACE
@@ -32,6 +32,7 @@ export(Presto)
 export(add_chunk)
 export(dbCreateTableAs)
 export(dbRenameTable)
+export(dummyPrestoConnection)
 export(kerberos_configs)
 export(presto_default)
 export(presto_has_default)

diff --git a/R/dbConnect.R b/R/dbConnect.R
@@ -135,3 +135,18 @@ setMethod(
     return(conn)
   }
 )
+
+#' A dummy PrestoConnection
+#' 
+#' @export
+#' @keywords internal
+#' @examples
+#' dummyPrestoConnection()
+dummyPrestoConnection <- function() {
+  methods::new("PrestoConnection",
+    session.timezone = Sys.timezone(),
+    output.timezone = Sys.timezone(),
+    request.config = httr::config(),
+    session = PrestoSession$new(list(), list())
+  )
+}
diff --git a/man/dummyPrestoConnection.Rd b/man/dummyPrestoConnection.Rd
diff --git a/tests/testthat/test-dbQuoteLiteral.R b/tests/testthat/test-dbQuoteLiteral.R
@@ -8,8 +8,8 @@ context("dbQuoteLiteral")
 
 source("utilities.R")
 
-test_that("dbQuoteLiteral works with live connection", {
-  conn <- presto_default()
+test_that("dbQuoteLiteral works", {
+  conn <- dummyPrestoConnection()
   expect_equal(dbQuoteLiteral(conn, DBI::SQL("foo")), DBI::SQL("foo"))
   expect_equal(
     dbQuoteLiteral(conn, factor(c("a", "b", NA_character_))),