pictoralist:

context("Integration test of baked in templates")

test_that("Baked in templates with single time points work with mtx data",{
  mtx_data <- read_data(spekex::get_data_path("mtx"))
  mtx_spek <- spekex::read_spek(spekex::get_spek_path("mtx"))

  templates <- load_templates()
  mtx_templates <- c(templates$ComparisonBarGraphHOR,
                     templates$ComparisonBarGraphVERT,
                     templates$EnhancedLeaderboard,
                     templates$Leaderboard,
                     templates$IUDGraph,
                     templates$TopPerformerGraph)

  results <- lapply(mtx_templates, FUN=function(t, recip, data, spek){t$run(recip, data, spek)},
                    recip = "E87746", data=mtx_data, spek=mtx_spek)
  is_ggplot <- sapply(results, function(x){"ggplot" %in% class(x)})
  expect_true(all(is_ggplot))
})

test_that("Baked in templates with single time points work with va data",{
  va_data <- read_data(spekex::get_data_path("va"))
  va_spek <- spekex::read_spek(spekex::get_spek_path("va"))

  templates <- load_templates()
  va_templates <- c(templates$SingleLineGraph)

  results <- lapply(va_templates, FUN=function(t, recip, data, spek){t$run(recip, data, spek)},
                    recip = "6559AA", data=va_data, spek=va_spek)
  is_ggplot <- sapply(results, function(x){"ggplot" %in% class(x)})
  expect_true(all(is_ggplot))
})

test_that("Data provided is used in Top Performer Template", {
  mtx_data <- read_data(spekex::get_data_path("mtx"))
  mtx_spek <- spekex::read_spek(spekex::get_spek_path("mtx"))

  templates <- load_templates()

  denom_colname <- 'total_scripts'
  numer_colname <- 'high_dose_scripts'

  recip_data <- filter(mtx_data, mtx_data$practice == "E87746")
  recip_data_zero <- filter(mtx_data, mtx_data$practice == "A81001")
  data_denom <- sum(recip_data[denom_colname])
  data_numer <- sum(recip_data[numer_colname])

  tpg_env <- templates$TopPerformerGraph
  result <- tpg_env$run("E87746", mtx_data, mtx_spek)
  result_zero <- tpg_env$run("A81001", mtx_data, mtx_spek)

  template_denom <- result$data$value[1]
  template_recip <- result$data$id[1]
  template_recip_zero <- result_zero$data$id[1]
  expect_true(template_denom == data_denom)
  expect_true(template_recip == "E87746")
  expect_true(template_recip_zero == "A81001")
})

test_that("Data provided is used in IUD Graph Template", {
  mtx_data <- read_data(spekex::get_data_path("mtx"))
  mtx_spek <- spekex::read_spek(spekex::get_spek_path("mtx"))

  templates <- load_templates()

  denom_colname <- 'total_scripts'
  numer_colname <- 'high_dose_scripts'
  recip_data <- filter(mtx_data, mtx_data$practice == "E84076")
  data_denom <- sum(recip_data[denom_colname])
  data_numer <- sum(recip_data[numer_colname])

  iud_env <- templates$IUDGraph
  result <- iud_env$run("E84076", mtx_data, mtx_spek)

  template_recip <- result$data$id[1]
  template_numer <- result$data$numer[1]
  template_denom <- result$data$denom[1]

  expect_true(template_recip == "E84076")
  expect_true(template_numer == data_numer)
  expect_true(template_denom == data_denom)
})

test_that("Data provided is used in ComparisonBarGraphHOR", {
  mtx_data <- read_data(spekex::get_data_path("mtx"))
  mtx_spek <- spekex::read_spek(spekex::get_spek_path("mtx"))

  templates <- load_templates()

  denom_colname <- 'total_quantity'
  numer_colname <- 'total_scripts'
  recipient <- "E84076"

  compHOR_env <- templates$ComparisonBarGraphHOR
  result <- compHOR_env$run(recipient, mtx_data, mtx_spek)

  top_performers <- mtx_data %>%
    group_by(practice) %>%
    summarise(total_scripts = sum(total_scripts), total_quantity = sum(total_quantity)) %>%
    mutate(percentage = round(total_scripts/total_quantity, digits=2)) %>%
    arrange(desc(total_scripts/total_quantity)) %>%
    select(practice, percentage) %>%
    head(14)

  # If recipient not in top 14, remove last elem and add recipient
  if(!(recipient %in% top_performers$practice)) {
    recip_data <- filter(mtx_data, mtx_data$practice == recipient)
    data_denom <- sum(recip_data[denom_colname])
    data_numer <- sum(recip_data[numer_colname])
    top_performers <- top_performers %>% head(13) %>%
      rbind(c(recipient, round(data_numer/data_denom, digits = 2)))
  }

  are_equal <- all(result$data$lengths == top_performers$percentage)
  expect_true(are_equal)
})

test_that("Data provided is used in ComparisonBarGraphVERT", {
  mtx_data <- read_data(spekex::get_data_path("mtx"))
  mtx_spek <- spekex::read_spek(spekex::get_spek_path("mtx"))

  templates <- load_templates()

  denom_colname <- 'total_quantity'
  numer_colname <- 'total_scripts'
  recipient <- "E84076"

  compVERT_env <- templates$ComparisonBarGraphVERT
  result <- compVERT_env$run(recipient, mtx_data, mtx_spek)

  top_performers <- mtx_data %>%
    group_by(practice) %>%
    summarise(total_scripts = sum(total_scripts), total_quantity = sum(total_quantity)) %>%
    mutate(percentage = round(total_scripts/total_quantity, digits=2)) %>%
    arrange(desc(total_scripts/total_quantity)) %>%
    select(practice, percentage) %>%
    head(14)

  # If recipient not in top 14, remove last elem and add recipient
  if(!(recipient %in% top_performers$practice)) {
    recip_data <- filter(mtx_data, mtx_data$practice == recipient)
    data_denom <- sum(recip_data[denom_colname])
    data_numer <- sum(recip_data[numer_colname])
    top_performers <- top_performers %>% head(13) %>%
      rbind(c(recipient, round(data_numer/data_denom, digits = 2)))
  }

  are_equal <- all(result$data$lengths == top_performers$percentage)
  expect_true(are_equal)
})

test_that("Data provided is used in EnhancedLeaderboard", {
  mtx_data <- read_data(spekex::get_data_path("mtx"))
  mtx_spek <- spekex::read_spek(spekex::get_spek_path("mtx"))

  templates <- load_templates()

  denom_colname <- 'total_quantity'
  numer_colname <- 'total_scripts'
  recipient <- "E84076"

  enh_env <- templates$EnhancedLeaderboard
  result <- enh_env$run(recipient, mtx_data, mtx_spek)

  top_performers <- mtx_data %>%
    group_by(practice) %>%
    summarise(total_scripts = sum(total_scripts), total_quantity = sum(total_quantity)) %>%
    mutate(percentage = floor(100*total_scripts/total_quantity)) %>%
    arrange(desc(total_scripts/total_quantity)) %>%
    select(practice, percentage, total_scripts, total_quantity) %>%
    head(7)

  numer_all_equal <- all(result$data$numer == top_performers$total_scripts)
  denom_all_equal <- all(result$data$denom == top_performers$total_quantity)

  expect_true(numer_all_equal)
  expect_true(denom_all_equal)
})

test_that("Data provided is used in Leaderboard", {
  mtx_data <- read_data(spekex::get_data_path("mtx"))
  mtx_spek <- spekex::read_spek(spekex::get_spek_path("mtx"))

  templates <- load_templates()

  denom_colname <- 'total_quantity'
  numer_colname <- 'total_scripts'
  recipient <- "E84076"

  lead_env <- templates$Leaderboard
  result <- lead_env$run(recipient, mtx_data, mtx_spek)

  top_performers <- mtx_data %>%
    group_by(practice) %>%
    summarise(total_scripts = sum(total_scripts), total_quantity = sum(total_quantity)) %>%
    mutate(percentage = floor(100*total_scripts/total_quantity)) %>%
    arrange(desc(total_scripts/total_quantity)) %>%
    select(practice, percentage, total_scripts, total_quantity) %>%
    head(7)

  numer_all_equal <- all(result$data$numer == top_performers$total_scripts)
  denom_all_equal <- all(result$data$denom == top_performers$total_quantity)

  expect_true(numer_all_equal)
  expect_true(denom_all_equal)
})

test_that("Data provided is used in baked in SingleLineGraph", {
  va_data <- read_data(spekex::get_data_path("va"))
  va_spek <- spekex::read_spek(spekex::get_spek_path("va"))

  templates <- load_templates()

  numer_colname <- 'documented'
  denom_colname <- 'total'
  recipient <- "6559AA"

  lead_env <- templates$SingleLineGraph
  result <- lead_env$run(recipient, va_data, va_spek)

  performer <- va_data %>%
    filter(sta6a == recipient) %>%
    select(sta6a, report_month, documented, total)

  dates <- performer$report_month
  template_dates <- result$data$dates
  all_equal <- all(dates == template_dates)

  expect_true(all_equal)
})

test_that("Data provided is used in baked in ComparisonLineGraph", {
  va_data <- read_data(spekex::get_data_path("va"))
  va_spek <- spekex::read_spek(spekex::get_spek_path("va"))

  templates <- load_templates()

  numer_colname <- 'documented'
  denom_colname <- 'total'
  recipient <- "6559AA"

  cmp_lne_env <- templates$ComparisonLineGraph
  result <- cmp_lne_env$run(recipient, va_data, va_spek)

  ids <- c("4429AA", "5569AA", "5689AB", "6559AA")

  ids_used_in_template <- as.character(unique(result$data$id))
  ids_used_in_test <- c(ids[1], ids[2], ids[3], ids[4])
  all_equal <- all(ids_used_in_template == ids_used_in_test)

  expect_true(all_equal)
})

test_that("Data provided is used in baked in PairedBarGraph", {
  va_data <- read_data(spekex::get_data_path("va"))
  va_spek <- spekex::read_spek(spekex::get_spek_path("va"))

  templates <- load_templates()

  numer_colname <- 'documented'
  denom_colname <- 'total'
  recipient <- "6559AA"

  paired_env <- templates$PairedBarGraph
  result <- paired_env$run(recipient, va_data, va_spek)
  test_dates <- c("2018-02-01", "2018-03-01", "2018-04-01", "2018-05-01")
  all_equal <- all(test_dates == unique(result$data$date))

  expect_true(all_equal)
})

test_that("Data provided is used in baked in PairedBarGraphHOR", {
  skip("No data available for testing")
  va_data <- read_data(spekex::get_data_path("va"))
  va_spek <- spekex::read_spek(spekex::get_spek_path("va"))

  templates <- load_templates()

  numer_colname <- 'documented'
  denom_colname <- 'total'
  recipient <- "6559AA"

  paired_HOR_env <- templates$PairedBarGraphHOR
  result <- paired_HOR_env$run(recipient, va_data, va_spek)
  test_dates <- c("2018-02-01", "2018-03-01", "2018-04-01", "2018-05-01")
  all_equal <- all(test_dates == unique(result$data$date))

  expect_true(all_equal)
})

test_that("Data provided is used in baked in SingleBarGraph", {
  va_data <- read_data(spekex::get_data_path("va"))
  va_spek <- spekex::read_spek(spekex::get_spek_path("va"))

  templates <- load_templates()

  numer_colname <- 'documented'
  denom_colname <- 'total'
  recipient <- "6559AA"

  bar_env <- templates$SingleBarTemplate
  result <- bar_env$run(recipient, va_data, va_spek)
  test_dates <- c("2018-02-01", "2018-03-01", "2018-04-01", "2018-05-01")
  all_equal <- all(test_dates == unique(result$data$dates))

  expect_true(all_equal)
})