|
1 |
| -# create SQL tables |
| 1 | +library(DBI) |
| 2 | +library(duckdb) |
| 3 | +library(dplyr) |
| 4 | +library(lubridate) |
| 5 | +library(testthat) |
| 6 | + |
| 7 | +# Create SQL tables |
2 | 8 | redcap_projects <- data.frame(
|
3 | 9 | project_id = 1:6,
|
4 | 10 | date_deleted = c(rep(NA, 5), format(Sys.time() - 86400, "%Y-%m-%d %H:%M:%S")),
|
5 |
| - log_event_table = c(rep('redcap_log_event1', 3), rep('redcap_log_event2', 3)) |
| 11 | + log_event_table = c(rep("redcap_log_event1", 3), rep("redcap_log_event2", 3)) |
6 | 12 | )
|
7 | 13 |
|
8 | 14 | redcap_log_event1 <- data.frame(
|
9 |
| - object_type = NA_character_, |
10 |
| - event = NA_character_, |
| 15 | + log_event_id = NA_integer_, |
11 | 16 | project_id = NA_integer_,
|
12 |
| - description = NA_character_ |
13 |
| -) |
14 |
| - |
15 |
| -redcap_log_event2 <- data.frame( |
16 |
| - object_type = NA_character_, |
| 17 | + ts = NA_character_, |
| 18 | + user = NA_character_, |
| 19 | + ip = NA_character_, |
| 20 | + page = NA_character_, |
17 | 21 | event = NA_character_,
|
18 |
| - project_id = NA_integer_, |
19 |
| - description = NA_character_ |
| 22 | + object_type = NA_character_, |
| 23 | + sql_log = NA_character_, |
| 24 | + pk = NA_character_, |
| 25 | + event_id = NA_character_, |
| 26 | + data_values = NA_character_, |
| 27 | + description = NA_character_, |
| 28 | + legacy = NA_integer_, |
| 29 | + change_reason = NA_character_ |
20 | 30 | )
|
21 | 31 |
|
22 |
| -# write SQL tables |
| 32 | +redcap_log_event2 <- redcap_log_event1 |
| 33 | + |
| 34 | +# Write SQL tables |
23 | 35 | conn <- DBI::dbConnect(duckdb::duckdb(), dbname = ":memory:")
|
24 | 36 | DBI::dbWriteTable(conn, "redcap_projects", redcap_projects)
|
25 | 37 | DBI::dbWriteTable(conn, "redcap_log_event1", redcap_log_event1)
|
26 | 38 | DBI::dbWriteTable(conn, "redcap_log_event2", redcap_log_event2)
|
27 | 39 |
|
28 |
| -# create comparison dfs |
29 |
| -expected_redcap_projects <- data.frame( |
30 |
| - project_id = 1:6, |
31 |
| - # convert to UTC to prevent test from failing due to timezone differences |
32 |
| - date_deleted = c(rep(as.Date(lubridate::with_tz(Sys.time(), "UTC")), 5), Sys.Date() - 1), |
33 |
| - log_event_table = c(rep('redcap_log_event1', 3), rep('redcap_log_event2', 3)) |
34 |
| -) |
| 40 | +current_ts <- format(Sys.time(), "%Y%m%d%H%M%S") |
| 41 | + |
| 42 | +# Expected results |
| 43 | +expected_redcap_projects <- redcap_projects |> |
| 44 | + mutate( |
| 45 | + date_deleted = if_else(is.na(date_deleted), as.character(as.Date(Sys.time())), date_deleted) |
| 46 | + ) |
35 | 47 |
|
36 | 48 | expected_redcap_log_event1 <- data.frame(
|
37 |
| - object_type = c(NA, rep("redcap_projects", 3)), |
38 |
| - event = c(NA, rep("MANAGE", 3)), |
| 49 | + log_event_id = NA_integer_, |
39 | 50 | project_id = c(NA, 1:3),
|
40 |
| - description = c(NA, rep("delete project", 3)) |
| 51 | + ts = c(NA, rep(current_ts, 3)), |
| 52 | + user = c(NA, rep("admin", 3)), |
| 53 | + ip = c(NA, rep(getip::getip("local"), 3)), |
| 54 | + page = c(NA, rep("rcc.billing::delete_abandoned_projects", 3)), |
| 55 | + event = c(NA, rep("MANAGE", 3)), |
| 56 | + object_type = c(NA, rep("redcap_projects", 3)), |
| 57 | + sql_log = c(NA, rep("UPDATE redcap_projects SET date_deleted = NOW() WHERE project_id IN (1,2,3,4,5)", 3)), |
| 58 | + pk = c(NA, as.character(1:3)), |
| 59 | + event_id = NA_character_, |
| 60 | + data_values = c(NA, sprintf("project_id = %d", 1:3)), |
| 61 | + description = c(NA, rep("Delete project", 3)), |
| 62 | + legacy = c(NA, rep(0, 3)), |
| 63 | + change_reason = NA_character_ |
41 | 64 | )
|
42 | 65 |
|
43 | 66 | expected_redcap_log_event2 <- data.frame(
|
44 |
| - object_type = c(NA, rep("redcap_projects", 2)), |
| 67 | + log_event_id = NA_integer_, |
| 68 | + project_id = c(NA, 4, 5), |
| 69 | + ts = c(NA, rep(current_ts, 2)), |
| 70 | + user = c(NA, rep("admin", 2)), |
| 71 | + ip = c(NA, rep(getip::getip("local"), 2)), |
| 72 | + page = c(NA, rep("rcc.billing::delete_abandoned_projects", 2)), |
45 | 73 | event = c(NA, rep("MANAGE", 2)),
|
46 |
| - project_id = c(NA, 4:5), |
47 |
| - description = c(NA, rep("delete project", 2)) |
| 74 | + object_type = c(NA, rep("redcap_projects", 2)), |
| 75 | + sql_log = c(NA, rep("UPDATE redcap_projects SET date_deleted = NOW() WHERE project_id IN (1,2,3,4,5)", 2)), |
| 76 | + pk = c(NA, "4", "5"), |
| 77 | + event_id = NA_character_, |
| 78 | + data_values = c(NA, sprintf("project_id = %d", 4:5)), |
| 79 | + description = c(NA, rep("Delete project", 2)), |
| 80 | + legacy = c(NA, rep(0, 2)), |
| 81 | + change_reason = NA_character_ |
48 | 82 | )
|
49 | 83 |
|
50 | 84 | expected_result <- data.frame(
|
51 | 85 | project_id = 1:8,
|
52 | 86 | status = c(rep("deleted", 5), "previously deleted", rep("does not exist", 2))
|
53 | 87 | )
|
54 | 88 |
|
55 |
| -# test function |
| 89 | +# Test function |
56 | 90 | project_ids <- 1:8
|
57 | 91 | deleted_projects <- delete_project(project_ids, conn)
|
58 | 92 |
|
59 |
| -testthat::test_that("delete_project deletes, updates and returns the correct project IDs", { |
60 |
| - expect_equal( |
61 |
| - DBI::dbGetQuery(conn, "select * from redcap_projects") |> |
62 |
| - # convert date_deleted to yyyy-mm-dd to allow comparison with expected_redcap_projects |
63 |
| - dplyr::mutate(date_deleted = as.Date(date_deleted)), |
64 |
| - expected_redcap_projects |
65 |
| - ) |
66 |
| - |
67 |
| - testthat::expect_equal( |
68 |
| - DBI::dbGetQuery(conn, "select * from redcap_log_event1"), |
69 |
| - expected_redcap_log_event1 |
70 |
| - ) |
71 |
| - testthat::expect_equal( |
72 |
| - DBI::dbGetQuery(conn, "select * from redcap_log_event2"), |
73 |
| - expected_redcap_log_event2 |
74 |
| - ) |
| 93 | +testthat::test_that("delete_project deletes, updates, and returns the correct project IDs and logs", { |
| 94 | + remove_seconds <- function(ts) { |
| 95 | + if (is.na(ts)) { |
| 96 | + return(NA) |
| 97 | + } |
| 98 | + substr(ts, 1, 12) |
| 99 | + } |
| 100 | + |
| 101 | + actual_redcap_log_event1 <- DBI::dbGetQuery(conn, "SELECT * FROM redcap_log_event1") |> |
| 102 | + select(-log_event_id) |> |
| 103 | + mutate( |
| 104 | + ts = sapply(ts, remove_seconds) |
| 105 | + ) |
| 106 | + |
| 107 | + expected_redcap_log_event1 <- expected_redcap_log_event1 |> |
| 108 | + select(-log_event_id) |> |
| 109 | + mutate( |
| 110 | + ts = sapply(ts, remove_seconds) |
| 111 | + ) |
| 112 | + |
| 113 | + testthat::expect_equal(actual_redcap_log_event1, expected_redcap_log_event1) |
| 114 | + |
| 115 | + actual_redcap_log_event2 <- DBI::dbGetQuery(conn, "SELECT * FROM redcap_log_event2") |> |
| 116 | + select(-log_event_id) |> |
| 117 | + mutate( |
| 118 | + ts = sapply(ts, remove_seconds) |
| 119 | + ) |
| 120 | + |
| 121 | + expected_redcap_log_event2 <- expected_redcap_log_event2 |> |
| 122 | + select(-log_event_id) |> |
| 123 | + mutate( |
| 124 | + ts = sapply(ts, remove_seconds) |
| 125 | + ) |
| 126 | + |
| 127 | + testthat::expect_equal(actual_redcap_log_event2, expected_redcap_log_event2) |
75 | 128 |
|
76 | 129 | testthat::expect_equal(deleted_projects$n, 5)
|
77 |
| - |
78 | 130 | testthat::expect_equal(deleted_projects$number_rows_logged, 5)
|
79 |
| - |
80 | 131 | testthat::expect_equal(deleted_projects$project_ids_deleted, 1:5)
|
81 |
| - |
82 | 132 | testthat::expect_equal(deleted_projects$data, expected_result)
|
83 |
| - |
84 | 133 | })
|
85 | 134 |
|
86 | 135 | DBI::dbDisconnect(conn, shutdown = TRUE)
|
0 commit comments