# GACRUN-TOC ----

test_that("No water defined, no default listed", {
  water <- water_df[1, ]

  expect_error(gac_toc(media_size = "8x30", ebct = 10)) # argument water is missing, with no default
  expect_error(gac_toc(water)) # water is not a defined water object
})

test_that("gacrun_toc returns error if inputs are misspelled or missing.", {
  water1 <- suppressWarnings(define_water(ph = 7.5, toc = 3.5))
  water2 <- suppressWarnings(define_water(temp = 25, tot_hard = 100, toc = 3.5)) # ph is not defined
  water3 <- suppressWarnings(define_water(ph = 7.5, temp = 25, tot_hard = 100)) # toc is not defined

  expect_error(gacrun_toc(water1, media_size = "11x40", model = "Zachman"))
  expect_error(gacrun_toc(water1, ebct = 15, model = "Zachman"))
  expect_error(gacrun_toc(water1, model = "Zachmann"))
  expect_error(gacrun_toc(water2, model = "Zachman"))
  expect_error(gacrun_toc(water3, model = "Zachman"))
})

test_that("gacrun_toc defaults to correct values.", {
  water <- suppressWarnings(define_water(ph = 7.5, toc = 3.5))

  plot1 <- gacrun_toc(water)
  plot2 <- gacrun_toc(water, ebct = 10, model = "Zachman", media_size = "12x40")

  expect_true(identical(plot1, plot2))
})

test_that("gacrun_toc works.", {
  water <- suppressWarnings(define_water(ph = 7.5, toc = 3.5))

  plot1 <- gacrun_toc(water, model = "WTP")
  plot2 <- gacrun_toc(water, model = "Zachman")
  plot3 <- gacrun_toc(water, ebct = 20, model = "WTP")
  plot4 <- gacrun_toc(water, model = "WTP", media_size = "8x30")
  plot5 <- gacrun_toc(water, model = "Zachman", media_size = "8x30")
  plot6 <- gacrun_toc(water, model = "WTP", bvs = c(2000, 50000, 500))

  expect_true(is.data.frame(plot1))
  expect_false(identical(plot1, plot2))
  expect_false(identical(plot1, plot3))
  expect_true(identical(plot1, plot4)) # media size isn't used in WTP model
  expect_false(identical(plot2, plot5)) # media size is used in Zachman model
  expect_false(nrow(plot4) == nrow(plot6)) # custom bed volume sequence is different length
})

################################################################################*
################################################################################*
# gacrun_toc helpers ----

test_that("gacrun_toc_df outputs are the same as base function, gacrun_toc", {
  testthat::skip_on_cran()
  water0 <- define_water(7.9, 20, 50,
    tds = 200,
    toc = 2, doc = 1.8, uv254 = 0.05
  )

  water1 <- define_water(8.5, 25, 80,
    tds = 100,
    toc = 3, doc = 2.8, uv254 = .08
  )

  water2 <- water0 %>%
    gacrun_toc(model = "WTP")
  water3 <- water1 %>%
    gacrun_toc(model = "WTP")

  water4 <- water_df %>%
    dplyr::slice_head(n = 2) %>%
    define_water_df() %>%
    gacrun_toc_df(model = "WTP", media_size = "12x40", ebct = 10)

  expect_equal(c(water2$bv, water3$bv), water4$defined_bv)
  expect_equal(c(water2$x_norm, water3$x_norm), water4$defined_x_norm)
})

# Test that output is a data frame with the correct number of columns
test_that("gacrun_toc_df output is data frame", {
  testthat::skip_on_cran()
  water0 <- suppressWarnings(water_df %>%
    define_water_df("raw") %>%
    mutate(
      model = "WTP",
      media_size = "12x40",
      ebct = 10,
      bvs = list(c(2000, 20000, 100))
    ))

  water1 <- water0 %>%
    gacrun_toc_df(input_water = "raw") %>%
    pluck_water(input_waters = c("raw"), parameter = c("doc"))

  expect_true(is.data.frame(water1))
  expect_equal(ncol(water0), ncol(water1) - 2)
  expect_equal(nrow(water1), 2172)
})

test_that("gacrun_toc_df handles optional bv argument correctly.", {
  testthat::skip_on_cran()
  water1 <- water_df %>%
    dplyr::slice(1) %>%
    define_water_df() %>%
    gacrun_toc_df(model = "WTP")

  water2 <- water_df %>%
    dplyr::slice(1) %>%
    define_water_df() %>%
    gacrun_toc_df(model = "WTP", bvs = c(2000, 50000, 500))

  expect_false(nrow(water1) == nrow(water2))
  expect_equal(nrow(water2), 97)
})