test_that("check cdm object get created", {
  indexCohort <- dplyr::tibble(
    cohort_definition_id = c(1, 1, 2, 2, 1, 3, 3, 3, 1, 3),
    subject_id = c(1, 4, 2, 3, 5, 5, 4, 3, 3, 1),
    cohort_start_date = as.Date(
      c(
        "2020-04-01",
        "2021-06-01",
        "2022-05-22",
        "2010-01-01",
        "2019-08-01",
        "2019-04-07",
        "2021-01-01",
        "2008-02-02",
        "2009-09-09",
        "2021-01-01"
      )
    ),
    cohort_end_date = as.Date(
      c(
        "2021-04-01",
        "2022-08-01",
        "2023-05-23",
        "2011-03-01",
        "2021-04-01",
        "2021-05-30",
        "2023-02-02",
        "2014-12-03",
        "2010-11-01",
        "2022-01-01"
      )
    )
  )

  markerCohort <- dplyr::tibble(
    cohort_definition_id = c(1, 1, 2, 2, 2, 3, 3, 3, 3, 3, 2),
    subject_id = c(1, 3, 4, 2, 5, 1, 2, 3, 4, 5, 1),
    cohort_start_date = as.Date(
      c(
        "2020-12-30",
        "2010-01-01",
        "2021-05-25",
        "2022-05-31",
        "2020-05-25",
        "2019-05-25",
        "2022-05-25",
        "2010-09-30",
        "2022-05-25",
        "2020-02-29",
        "2021-01-01"
      )
    ),
    cohort_end_date = cohort_start_date
  )

  expect_no_warning(omock::emptyCdmReference(cdmName = "mock") |>
    omock::mockCdmFromTables(tables = list(index_cohort = indexCohort, marker_cohort = markerCohort)))

  cdm <- omock::emptyCdmReference(cdmName = "mock") |>
    omock::mockCdmFromTables(tables = list(index_cohort = indexCohort, marker_cohort = markerCohort))

  expect_no_error(cdm$marker_cohort)
  expect_no_error(cdm$index_cohort)


  expect_true(attributes(cdm$marker_cohort)$cohort_set |> dplyr::tally() > 0)
  expect_true(attributes(cdm$index_cohort)$cohort_set |> dplyr::tally() > 0)

  tables <- list(
    observation_period = dplyr::tibble(
      observation_period_id = as.integer(1:8),
      person_id = c(1, 1, 1, 2, 2, 3, 3, 4) |> as.integer(),
      observation_period_start_date = as.Date(
        c(
          "2020-03-01",
          "2020-03-25",
          "2020-04-25",
          "2020-08-10",
          "2020-03-10",
          "2020-03-01",
          "2020-04-10",
          "2020-03-10"
        )
      ),
      observation_period_end_date = as.Date(
        c(
          "2020-03-20",
          "2020-03-30",
          "2020-08-15",
          "2020-12-31",
          "2020-03-27",
          "2020-03-09",
          "2020-05-08",
          "2020-12-10"
        )
      )
    )
  )


  cdm <- omock::mockCdmFromTables(tables = tables)

  expect_true(
    all(
      cdm$observation_period |> dplyr::select(person_id) ==
        tables$observation_period |> dplyr::select(person_id)
    )
  )

  expect_true(
    all(
      cdm$observation_period |> dplyr::select(observation_period_start_date) ==
        tables$observation_period |> dplyr::select(observation_period_start_date)
    )
  )

  expect_true(cdm$person |> nrow() == 4)
})

dd <- omock::mockCdmFromTables(tables = list(
  observation_period = dplyr::tibble(
    observation_period_id = as.integer(1:8),
    person_id = c(1, 1, 1, 2, 2, 3, 3, 4) |> as.integer(),
    observation_period_start_date = as.Date(
      c(
        "2020-03-01",
        "2020-03-25",
        "2020-04-25",
        "2020-08-10",
        "2020-03-10",
        "2020-03-01",
        "2020-04-10",
        "2020-03-10"
      )
    ),
    observation_period_end_date = as.Date(
      c(
        "2020-03-20",
        "2020-03-30",
        "2020-08-15",
        "2020-12-31",
        "2020-03-27",
        "2020-03-09",
        "2020-05-08",
        "2020-12-10"
      )
    )
  )
))


test_that("check NA", {
  expect_error(omock::mockCdmFromTables(
    tables = list(
      visit_occurrence = dplyr::tibble(
        person_id = c(1L, 1L, 2L),
        visit_start_date = as.Date("2020-01-01") + c(0L, 29L, 70L),
        visit_end_date = as.Date("2020-01-01") + c(30L, 45L, 89L),
      ),
      condition_occurrence = dplyr::tibble(
        person_id = c(1L, 2L, 2L),
        condition_start_date = as.Date("2020-01-01") + c(50L, 51L, 89L),
        condition_end_date = as.Date("2020-01-01") + c(NA, 77L, 90L)
      )
    )
  ))


  expect_error(omock::mockCdmFromTables(
    tables = list(
      visit_occurrence = dplyr::tibble(
        person_id = c(1L, 1L, 2L),
        visit_start_date = as.Date("2020-01-01") + c(0L, 29L, 70L),
        visit_end_date = as.Date("2020-01-01") + c(30L, 45L, 89L),
      ),
      condition_occurrence = dplyr::tibble(
        person_id = c(1L, 2L, 2L),
        condition_start_date = as.Date("2020-01-01") + c(50L, 51L, 89L),
        condition_end_date = as.Date("2020-01-01") + c(NA, 77L, 90L)
      )
    )
  ))

  expect_no_warning(omock::mockCdmFromTables(
    tables = list(
      visit_occurrence = dplyr::tibble(
        person_id = c(1L, 1L, 2L),
        visit_start_date = as.Date("2020-01-01") + c(0L, 29L, 70L),
        visit_end_date = as.Date("2020-01-01") + c(30L, 45L, 89L),
      ),
      condition_occurrence = dplyr::tibble(
        person_id = c(1L, 2L, 2L),
        condition_start_date = as.Date("2020-01-01") + c(50L, 51L, 89L),
        condition_end_date = as.Date("2020-01-01") + c(1L, 77L, 90L)
      )
    )
  ))

  expect_error(omock::mockCdmFromTables(
    tables = list(
      visit_occurrence = dplyr::tibble(
        person_id = c(1L, 1L, 2L),
        visit_start_date = as.Date("2020-01-01") + c(0L, 29L, 70L),
        visit_end_date = as.Date("2025-01-01") + c(30L, 45L, 89L),
      ),
      condition_occurrence = dplyr::tibble(
        person_id = c(1L, 2L, 2L),
        condition_start_date = as.Date("2020-01-01") + c(50L, 51L, 89L),
        condition_end_date = as.Date("2020-01-01") + c(1L, 77L, 90L)
      )
    )
  ))

  expect_no_warning(omock::mockCdmFromTables(tables = list(
    observation_period = dplyr::tibble(
      "observation_period_id" = c(1L, 2L, 3L),
      "person_id" = c(1L, 2L, 3L),
      "observation_period_start_date" = as.Date("2000-01-01"),
      "observation_period_end_date" = as.Date("2024-01-01"),
      "period_type_concept_id" = 1L
    ),
    cohort = dplyr::tibble(
      "cohort_definition_id" = 1L,
      "subject_id" = c(1L, 1L, 2L, 3L),
      "cohort_start_date" = as.Date(c(
        "2020-01-01", "2020-01-12", "2021-01-01", "2022-01-01"
      )),
      "cohort_end_date" = as.Date(c(
        "2020-01-10", "2020-01-15", "2021-01-01", "2022-01-01"
      ))
    )
  )))

  cdm <- omock::mockCdmFromTables(tables = list(
    cohort = dplyr::tibble(
      "cohort_definition_id" = 1L,
      "subject_id" = 1L,
      "cohort_start_date" = as.Date("2020-01-01"),
      "cohort_end_date" = as.Date("2020-01-01")
    ),
    concept = dplyr::tibble(
      "concept_id" = 1L,
      "concept_name" = "my concept",
      "domain_id" = "drug",
      "vocabulary_id" = NA_integer_,
      "concept_class_id" = NA_integer_,
      "concept_code" = NA_integer_,
      "valid_start_date" = as.Date("1900-01-01"),
      "valid_end_date" = as.Date("2100-01-01")
    ),
    drug_exposure = dplyr::tibble(
      "drug_exposure_id" = c(1L, 2L),
      "person_id" = 1L,
      "drug_concept_id" = 1L,
      "drug_exposure_start_date" = as.Date(c("2020-01-01", "2021-01-01")),
      "drug_exposure_end_date" = as.Date(c("2020-01-15", "2021-01-15")),
      "drug_type_concept_id" = 1L,
      verbatim_end_date = drug_exposure_end_date
    ),
    observation_period = dplyr::tibble(
      "observation_period_id" = c(1L, 2L),
      "person_id" = 1L,
      "period_type_concept_id" = 1L,
      "observation_period_start_date" = as.Date(c("2020-01-01", "2021-01-01")),
      "observation_period_end_date" = as.Date(c("2020-06-01", "2021-06-01"))
    )
  ))

  expect_true(all(
    cdm$observation_period |> dplyr::pull("observation_period_start_date") %in%
      c("2020-01-01", "2021-01-01")
  ))

  expect_error(omock::mockCdmFromTables(tables = list("person" = dplyr::tibble(
    "person_id" = c(1L, 2L, 3L)
  ))))

  # cdm with just person table
  expect_no_error(cdm <- omock::mockCdmFromTables(tables = list(
    person = dplyr::tibble(
      person_id = 1:4,
      gender_concept_id = c(8532L, 8507L, 8507L, 8507L),
      year_of_birth = c(1997L, 1963L, 1986L, 1978L),
      month_of_birth = c(8L, 1L, 3L, 11L),
      day_of_birth = c(22L, 27L, 10L, 8L),
      race_concept_id = NA_integer_,
      ethnicity_concept_id = NA_integer_
    )
  )))
})