test: Fix snapshot tests

duckdb · Feb 24, 2024 · 1d54154 · 1d54154
1 parent 098d4c7
commit 1d54154
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/testthat/test_backend-dbplyr__duckdb_connection.R b/tests/testthat/test_backend-dbplyr__duckdb_connection.R
@@ -233,6 +233,8 @@ test_that("snapshots of dbplyr generic scalar translation", {
   local_edition(3)
   translate <- function(...) dbplyr::translate_sql(..., con = con)
 
+  rlang::local_options(cli.num_colors = 1)
+
   expect_snapshot({
     translate(as.character(1))
     translate(as.character(1L))
@@ -263,6 +265,8 @@ test_that("snapshots of duckdb custom scalars translations", {
   local_edition(3)
   translate <- function(...) dbplyr::translate_sql(..., con = con)
 
+  rlang::local_options(cli.num_colors = 1)
+
   expect_snapshot({
     #  translate(as(1,"CHARACTER"))        # Not implemented
     translate(as.raw(10))
@@ -304,6 +308,8 @@ test_that("snapshot tests for pasting translate", {
   local_edition(3)
   translate <- function(...) dbplyr::translate_sql(..., con = con)
 
+  rlang::local_options(cli.num_colors = 1)
+
   expect_snapshot({
     translate(paste("hi", "bye"))
     translate(paste("hi", "bye", sep = "-"))
@@ -327,6 +333,8 @@ test_that("snapshots for custom lubridate functions translated correctly", {
   local_edition(3)
   translate <- function(...) dbplyr::translate_sql(..., con = con)
 
+  rlang::local_options(cli.num_colors = 1)
+
   expect_snapshot({
     translate(yday(x))
     translate(quarter(x))
@@ -371,6 +379,8 @@ test_that("snapshots for custom stringr functions translated correctly", {
   local_edition(3)
   translate <- function(...) dbplyr::translate_sql(..., con = con)
 
+  rlang::local_options(cli.num_colors = 1)
+
   expect_snapshot({
     translate(str_c(x, y))
     #  translate(str_c(x, collapse = ""))  # Error
@@ -399,6 +409,8 @@ test_that("snapshots datetime escaping working as in DBI", {
   local_edition(3)
   escape <- function(...) dbplyr::escape(..., con = con)
 
+  rlang::local_options(cli.num_colors = 1)
+
   expect_snapshot({
     test_date <- as.Date("2020-01-01")
     escape(test_date)
@@ -422,6 +434,8 @@ test_that("two variable aggregates are translated correctly", {
   local_edition(3)
   translate <- function(...) dbplyr::translate_sql(..., con = con)
 
+  rlang::local_options(cli.num_colors = 1)
+
   expect_snapshot({
     translate(cor(x, y), window = FALSE)
     translate(cor(x, y), window = TRUE)
@@ -436,6 +450,8 @@ test_that("these should give errors", {
   local_edition(3)
   translate <- function(...) dbplyr::translate_sql(..., con = con)
 
+  rlang::local_options(cli.num_colors = 1)
+
   expect_snapshot(error = TRUE, {
     translate(grepl("dummy", txt, perl = TRUE)) # Expected error
     #    translate(paste0(x, collapse = ""), window = FALSE) # Skip because of changing rlang_error (sql_paste())