tests(pkg-r): Test that cleanup is correctly set in querychat_app()

Author: gadenbuie

pkg-r/R/QueryChat.R

@@ -680,7 +680,7 @@ querychat_app <- function(
     cleanup <-
       is.data.frame(data_source) &&
       !in_shiny_session() &&
-      interactive()
+      is_interactive()
   }
 
   qc <- QueryChat$new(

pkg-r/tests/testthat/test-QueryChat.R

@@ -306,3 +306,39 @@ describe("normalize_data_source()", {
     })
   })
 })
+
+test_that("querychat_app() only cleans up data frame sources on exit", {
+  local_mocked_r6_class(
+    QueryChat,
+    public = list(
+      initialize = function(..., cleanup) {
+        # have to use an option because the code is evaluated in a far-away env
+        options(.test_cleanup = cleanup)
+      },
+      app = function(...) {}
+    )
+  )
+  withr::local_options(rlang_interactive = TRUE)
+
+  withr::with_options(list(.test_cleanup = NULL), {
+    test_df <- new_test_df()
+    querychat_app(test_df)
+    cleanup_result <- getOption(".test_cleanup")
+    expect_true(cleanup_result)
+  })
+
+  withr::with_options(list(.test_cleanup = NULL), {
+    test_ds <- local_data_frame_source(new_test_df())
+    querychat_app(test_ds)
+    cleanup_result <- getOption(".test_cleanup")
+    expect_false(cleanup_result)
+  })
+
+  withr::with_options(list(.test_cleanup = NULL), {
+    con <- local_sqlite_connection(new_test_df())
+    test_ds <- DBISource$new(con$conn, "test_table")
+    querychat_app(test_ds)
+    cleanup_result <- getOption(".test_cleanup")
+    expect_false(cleanup_result)
+  })
+})