85 lines
3.1 KiB
R
85 lines
3.1 KiB
R
|
library(DatabaseConnector)
|
||
|
|
|
||
|
|
connectionDetails <- createConnectionDetails(
|
||
|
|
dbms = "postgresql",
|
||
|
|
user = Sys.getenv("CDMDDLBASE_POSTGRESQL_USER"),
|
||
|
|
password = Sys.getenv("CDMDDLBASE_POSTGRESQL_PASSWORD"),
|
||
|
|
server = Sys.getenv("CDMDDLBASE_POSTGRESQL_SERVER"),
|
||
|
|
pathToDriver = file.path(Sys.getenv("HOME"), "drivers")
|
||
|
|
)
|
||
|
|
|
||
|
|
# Helper functions used in tests
|
||
|
|
.listTablesInSchema <- function(connectionDetails, schema) {
|
||
|
|
con <- DatabaseConnector::connect(connectionDetails)
|
||
|
|
tables <- DBI::dbListObjects(con, prefix = "cdmddlbase")
|
||
|
|
DatabaseConnector::disconnect(con)
|
||
|
|
tables <- subset(tables, is_prefix == FALSE)
|
||
|
|
tables <- subset(tables, grepl("table", table))$table
|
||
|
|
tables <- vapply(tables, function(.) .@name, FUN.VALUE = character(1))
|
||
|
|
return(tables)
|
||
|
|
}
|
||
|
|
|
||
|
|
.dropAllTablesFromSchema <- function(connectionDetails, schema) {
|
||
|
|
tables <- .listTablesInSchema(connectionDetails, schema)
|
||
|
|
|
||
|
|
con <- DatabaseConnector::connect(connectionDetails)
|
||
|
|
for(table in tables) {
|
||
|
|
DBI::dbRemoveTable(con, name = DBI::SQL(paste(schema, table, sep = ".")))
|
||
|
|
}
|
||
|
|
DatabaseConnector::disconnect(con)
|
||
|
|
}
|
||
|
|
|
||
|
|
.removeConstraintsPostgresql <- function(connectionDetails, schema) {
|
||
|
|
# the order of removal of constraints matters!
|
||
|
|
con <- DatabaseConnector::connect(connectionDetails)
|
||
|
|
constraints <- DBI::dbGetQuery(con,
|
||
|
|
"SELECT con.conname, rel.relname as relname
|
||
|
|
FROM pg_catalog.pg_constraint con
|
||
|
|
INNER JOIN pg_catalog.pg_class rel
|
||
|
|
ON rel.oid = con.conrelid
|
||
|
|
INNER JOIN pg_catalog.pg_namespace nsp
|
||
|
|
ON nsp.oid = connamespace
|
||
|
|
WHERE nsp.nspname = 'cdmddlbase';")
|
||
|
|
|
||
|
|
|
||
|
|
constraints <- dplyr::mutate(constraints, sql = paste0("alter table ", schema, ".", relname, " drop constraint if exists ", conname, ";\n" ))
|
||
|
|
|
||
|
|
sql <- paste(rev(constraints$sql), collapse = "")
|
||
|
|
executeSql(con, sql)
|
||
|
|
|
||
|
|
disconnect(con)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
test_that("Database can be connected to", {
|
||
|
|
expect_error(con <- connect(connectionDetails), NA)
|
||
|
|
disconnect(con)
|
||
|
|
})
|
||
|
|
|
||
|
|
test_that("Execute DDL on Postgres", {
|
||
|
|
# make sure the schema is cleared out
|
||
|
|
cdmDatabaseSchema <- Sys.getenv("CDMDDLBASE_POSTGRESQL_SCHEMA")
|
||
|
|
cdmVersion <- "5.4"
|
||
|
|
# .removeConstraintsPostgresql(connectionDetails, cdmDatabaseSchema)
|
||
|
|
.dropAllTablesFromSchema(connectionDetails, cdmDatabaseSchema)
|
||
|
|
cat(paste("Connecting to schema", cdmDatabaseSchema, "\n"))
|
||
|
|
executeDdl(connectionDetails,
|
||
|
|
cdmVersion = cdmVersion,
|
||
|
|
cdmDatabaseSchema = cdmDatabaseSchema,
|
||
|
|
executeDdl = TRUE,
|
||
|
|
executePrimaryKey = FALSE,
|
||
|
|
executeForeignKey = FALSE)
|
||
|
|
|
||
|
|
tables <- .listTablesInSchema(connectionDetails, schema = cdmDatabaseSchema)
|
||
|
|
|
||
|
|
cdmTableCsvLoc <- system.file(file.path("csv", paste0("OMOP_CDMv", cdmVersion, "_Table_Level.csv")), package = "CommonDataModel", mustWork = TRUE)
|
||
|
|
tableSpecs <- read.csv(cdmTableCsvLoc, stringsAsFactors = FALSE)$cdmTableName
|
||
|
|
|
||
|
|
# check that the tables in the database match the tables in the specification
|
||
|
|
expect_equal(sort(tables), sort(tableSpecs))
|
||
|
|
|
||
|
|
# clear schema
|
||
|
|
# .removeConstraintsPostgresql(connectionDetails, cdmDatabaseSchema)
|
||
|
|
.dropAllTablesFromSchema(connectionDetails, cdmDatabaseSchema)
|
||
|
|
})
|