library(testthat)
library(TreatmentPatterns)
library(dplyr)
library(CDMConnector)

andromeda <- Andromeda::andromeda()

con <- DBI::dbConnect(duckdb::duckdb(), dbdir = eunomia_dir())

withr::defer({
  Andromeda::close(andromeda)
  DBI::dbDisconnect(con, shutdown = TRUE)
})

cohorts <- data.frame(
  cohortId = c(1, 2, 3),
  cohortName = c("Disease X", "Drug A", "Drug B"),
  type = c("target", "event", "event")
)

cohort_table <- tibble(
  cohort_definition_id = c(3, 2, 1),
  subject_id = c(1, 1, 1),
  cohort_start_date = as.Date(c("2014-10-10", "2014-11-07", "2014-10-10")),
  cohort_end_date = as.Date(c("2015-08-01", "2014-12-04", "2015-08-01"))
)

dplyr::copy_to(con, cohort_table, overwrite = TRUE)

localCdm <- cdmFromCon(
  con = con,
  cdmSchema = "main",
  writeSchema = "main",
  cohortTables = "cohort_table"
)

cdmInterface <- TreatmentPatterns:::CDMInterface$new(cdm = localCdm)

test_that("Method: new", {
  expect_true(R6::is.R6(
    TreatmentPatterns:::CDMInterface$new(cdm = localCdm)
  ))
})

test_that("Method: new - empty", {
  expect_error(
    TreatmentPatterns:::CDMInterface$new(),
    "Could not assert if CDMConnector or DatabaseConnector is being used.")
})

test_that("Method: fetchMetadata", {
  cdmInterface$fetchMetadata(andromeda)

  metadata <- andromeda$metadata %>% collect()

  expect_in(
    c("cdmSourceName", "cdmSourceAbbreviation", "cdmReleaseDate", "vocabularyVersion"),
    names(metadata)
  )

  expect_identical(metadata$rVersion, base::version$version.string)
  expect_identical(metadata$platform, base::version$platform)
  expect_identical(nrow(metadata), 1L)
  expect_identical(ncol(metadata), 8L)
})

test_that("Method: fetchCohortTable", {
  # Update CDM with new dummy data
  cdmInterface <- TreatmentPatterns:::CDMInterface$new(
    cdm = localCdm
  )
  
  # Viral Sinusitis
  cdmInterface$fetchCohortTable(
    cohorts = cohorts,
    cohortTableName = "cohort_table",
    andromeda = andromeda,
    andromedaTableName = "cohortTable",
    minEraDuration = 5
  )
  
  res <- andromeda$cohortTable

  expect_identical(ncol(res), 6L)
  expect_identical(res %>% collect() %>% nrow(), 3L)

  # Empty
  cdmInterface$fetchCohortTable(
    cohorts = data.frame(
      cohortId = numeric(),
      cohortName = character(),
      type = character()
    ),
    cohortTableName = "cohort_table",
    andromeda = andromeda,
    andromedaTableName = "cohortTable",
    minEraDuration = 5
  )
  
  res <- andromeda$cohortTable
  
  expect_identical(ncol(res), 6L)
  expect_identical(res %>% collect() %>% nrow(), 0L)
})