From e28df5f9209d146b75c9b630e66ab6d1cf404f32 Mon Sep 17 00:00:00 2001
From: Anthony Sena <asena5@its.jnj.com>
Date: Fri, 14 Jun 2024 16:44:12 -0400
Subject: [PATCH] Renable upload tests

---
 tests/testthat/test-ResultsDataModel.R | 111 ++++++++++++-------------
 1 file changed, 55 insertions(+), 56 deletions(-)

diff --git a/tests/testthat/test-ResultsDataModel.R b/tests/testthat/test-ResultsDataModel.R
index a77fbd6..5f8564a 100644
--- a/tests/testthat/test-ResultsDataModel.R
+++ b/tests/testthat/test-ResultsDataModel.R
@@ -64,7 +64,6 @@ testCreateSchema <- function(connectionDetails, resultsDatabaseSchema) {
       databaseSchema = resultsDatabaseSchema,
     )
   )
-  print('got here')
   specifications <- getResultsDataModelSpecifications()
   for (tableName in unique(specifications$tableName)) {
     expect_true(DatabaseConnector::existsTable(connection = connection,
@@ -79,62 +78,62 @@ testCreateSchema <- function(connectionDetails, resultsDatabaseSchema) {
 }
 
 test_that("Create schema", {
-  # testCreateSchema(connectionDetails = postgresConnectionDetails,
-  #                 resultsDatabaseSchema = postgresResultsDatabaseSchema)
+  testCreateSchema(connectionDetails = postgresConnectionDetails,
+                  resultsDatabaseSchema = postgresResultsDatabaseSchema)
   testCreateSchema(connectionDetails = sqliteConnectionDetails,
                    resultsDatabaseSchema = sqliteResultsDatabaseSchema)
 })
 
-# testUploadResults <- function(connectionDetails, resultsDatabaseSchema, resultsFolder) {
-#   uploadResults(
-#     connectionDetails = connectionDetails,
-#     schema = resultsDatabaseSchema,
-#     resultsFolder = resultsFolder,
-#     purgeSiteDataBeforeUploading = FALSE
-#   )
-# 
-#   # Check if there's data:
-#   connection <- DatabaseConnector::connect(connectionDetails)
-#   on.exit(DatabaseConnector::disconnect(connection))
-# 
-#   specifications <- getResultsDataModelSpecifications()
-#   for (tableName in unique(specifications$tableName)) {
-#     primaryKey <- specifications %>%
-#       dplyr::filter(tableName == !!tableName &
-#                       primaryKey == "Yes") %>%
-#       dplyr::select(columnName) %>%
-#       dplyr::pull()
-# 
-#     if ("database_id" %in% primaryKey) {
-#       sql <- "SELECT COUNT(*) FROM @database_schema.@table_name WHERE database_id = '@database_id';"
-#       databaseIdCount <- DatabaseConnector::renderTranslateQuerySql(
-#         connection = connection,
-#         sql = sql,
-#         database_schema = resultsDatabaseSchema,
-#         table_name = tableName,
-#         database_id = "Eunomia"
-#       )[, 1]
-#       expect_true(databaseIdCount >= 0)
-#     }
-#   }
-# }
-# 
-# test_that("Results upload", {
-#   unzipFolder <- tempfile("unzipTempFolder", tmpdir = tempdir())
-#   dir.create(path = unzipFolder, recursive = TRUE)
-#   on.exit(unlink(unzipFolder, recursive = TRUE), add = TRUE)
-# 
-#   zip::unzip(
-#     zipfile = system.file(
-#       "testdata/Results_Eunomia.zip",
-#       package = "CohortGenerator"
-#     ),
-#     exdir = unzipFolder
-#   )
-#   #testUploadResults(connectionDetails = postgresConnectionDetails,
-#   #                  resultsDatabaseSchema = postgresResultsDatabaseSchema,
-#   #                  resultsFolder = unzipFolder)
-#   # testUploadResults(connectionDetails = sqliteConnectionDetails,
-#   #                   resultsDatabaseSchema = sqliteResultsDatabaseSchema,
-#   #                   resultsFolder = unzipFolder)
-# })
\ No newline at end of file
+testUploadResults <- function(connectionDetails, resultsDatabaseSchema, resultsFolder) {
+  uploadResults(
+    connectionDetails = connectionDetails,
+    schema = resultsDatabaseSchema,
+    resultsFolder = resultsFolder,
+    purgeSiteDataBeforeUploading = FALSE
+  )
+
+  # Check if there's data:
+  connection <- DatabaseConnector::connect(connectionDetails)
+  on.exit(DatabaseConnector::disconnect(connection))
+
+  specifications <- getResultsDataModelSpecifications()
+  for (tableName in unique(specifications$tableName)) {
+    primaryKey <- specifications %>%
+      dplyr::filter(tableName == !!tableName &
+                      primaryKey == "Yes") %>%
+      dplyr::select(columnName) %>%
+      dplyr::pull()
+
+    if ("database_id" %in% primaryKey) {
+      sql <- "SELECT COUNT(*) FROM @database_schema.@table_name WHERE database_id = '@database_id';"
+      databaseIdCount <- DatabaseConnector::renderTranslateQuerySql(
+        connection = connection,
+        sql = sql,
+        database_schema = resultsDatabaseSchema,
+        table_name = tableName,
+        database_id = "Eunomia"
+      )[, 1]
+      expect_true(databaseIdCount >= 0)
+    }
+  }
+}
+
+test_that("Results upload", {
+  unzipFolder <- tempfile("unzipTempFolder", tmpdir = tempdir())
+  dir.create(path = unzipFolder, recursive = TRUE)
+  on.exit(unlink(unzipFolder, recursive = TRUE), add = TRUE)
+
+  zip::unzip(
+    zipfile = system.file(
+      "testdata/Results_Eunomia.zip",
+      package = "CohortGenerator"
+    ),
+    exdir = unzipFolder
+  )
+  #testUploadResults(connectionDetails = postgresConnectionDetails,
+  #                  resultsDatabaseSchema = postgresResultsDatabaseSchema,
+  #                  resultsFolder = unzipFolder)
+  # testUploadResults(connectionDetails = sqliteConnectionDetails,
+  #                   resultsDatabaseSchema = sqliteResultsDatabaseSchema,
+  #                   resultsFolder = unzipFolder)
+})
\ No newline at end of file