test_that("Modify water errors when arguments are wrong", {
  water0 <- define_water(ph = 7, temp = 25, alk = 100, 0, 0, 0, 0, 0, 0, tds = 100, toc = 5, doc = 4.8, uv254 = .1)

  expect_error(modify_water(water0, "br", 50))
  expect_error(modify_water(water0, "br", 50, "cm-1"))
  expect_error(modify_water(water0, "br", "50", "ug/L"))
  expect_error(modify_water(water0, "bromide", 50, "ug/L"))
  expect_error(modify_water(water_df, "br", 50, "ug/L"))
})


test_that("Modify water works", {
  water0 <- define_water(ph = 7, temp = 25, alk = 100, 0, 0, 0, 0, 0, 0, tds = 100, toc = 5, doc = 4.8, uv254 = .1)

  water1 <- modify_water(water0, "br", 50, "ug/L")
  water2 <- modify_water(water0, "na", 50, "mg/L")

  expect_s4_class(water1, "water")
  expect_equal(water1@br, convert_units(50, "br", "ug/L"))
  expect_equal(water0@tds, water1@tds)
  expect_equal(water2@na, convert_units(50, "na", "mg/L"))
  expect_equal(water0@ph, water2@ph)
})

test_that("Modify water works with multiple inputs", {
  water0 <- define_water(ph = 7, temp = 25, alk = 100, 0, 0, 0, 0, 0, 0, tds = 100, toc = 5, doc = 4.8, uv254 = .1)

  water1 <- modify_water(water0, slot = c("br", "na"), value = c(50, 50), units = c("ug/L", "mg/L"))

  expect_s4_class(water1, "water")
  expect_equal(water1@br, convert_units(50, "br", "ug/L"))
  expect_equal(water1@na, convert_units(50, "na", "mg/L"))
  expect_equal(water0@ph, water1@ph)
})

test_that("Modify water df takes and returns correct argument types and classes.", {
  testthat::skip_on_cran()
  water0 <- water_df %>%
    define_water_df("test") %>%
    mutate(bromide = 50)

  water1 <- modify_water_df(water0, "test", "modified", "br", bromide, "ug/L")
  water2 <- modify_water_df(water0, "test", "modified", "br", 50, "ug/L")

  expect_error(modify_water_df(water_df, "br", 50, "ug/L"))
  expect_error(modify_water_df(water0, 50, "ug/L"))
  expect_equal(water1$modified[[1]]@br, water2$modified[[1]]@br)
  expect_s4_class(water1$modified[[1]], "water")
})

test_that("Modify water df works with multiple inputs.", {
  testthat::skip_on_cran()

  water1 <- water_df %>%
    define_water_df() %>%
    mutate(
      bromide = 50,
      sodium = 60
    ) %>%
    modify_water_df(output_water = "mod1", slot = "br", value = bromide, units = "ug/L") %>%
    modify_water_df("mod1", "mod2", slot = "na", value = sodium, units = "mg/L")

  water2 <- water_df %>%
    define_water_df() %>%
    modify_water_df(
      slot = c("br", "na"),
      value = c(50, 60),
      units = c("ug/L", "mg/L")
    )

  expect_s4_class(water2$modified[[1]], "water")
  expect_equal(water1$modified, water2$mod2) # can input to function or as col
})