improve with_db_user example reproducibilty

getwilds · Feb 4, 2025 · 915b0ea · 915b0ea
1 parent c06d138
commit 915b0ea
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/R/with.R b/R/with.R
@@ -20,7 +20,8 @@
 #' library(RPostgres)
 #' con <- dbConnect(Postgres())
 #'
-#' rls_current_user(con)
+#' current1 <- rls_current_user(con)
+#' current1
 #'
 #' dbCreateTable(con, "cars", mtcars)
 #' rows_append(
@@ -30,12 +31,21 @@
 #' )
 #'
 #' dbExecute(con, "CREATE ROLE analysts")
+#' dbExecute(con, "GRANT SELECT ON TABLE cars TO analysts")
 #'
 #' with_db_user(
 #'  con = con,
 #'  user = "analysts",
 #'  code = dbGetQuery(con, "select * from cars")
 #' )
+#' # current user should be the same as above
+#' current2 <- rls_current_user(con)
+#' identical(current1, current2)
+#' 
+#' # cleanup
+#' dbRemoveTable(con, "cars")
+#' dbExecute(con, "DROP ROLE analysts")
+#' dbDisconnect(con)
 with_db_user <- function(con, user, code) {
 	original_user <- rls_current_user(con)
 	defer(dbExecute(con,

diff --git a/man/with_db_user.Rd b/man/with_db_user.Rd