yn <- c(NA, NA, "fizz", NA, "buzz", "fizz", NA, NA, "fizz", "buzz")
yp <- c(1, 2, "fizz", 4, "buzz", "fizz", 7, 8, "fizz", "buzz")
yd <- c(
  "bam", "bam", "fizz", "bam", "buzz", "fizz", "bam", "bam", "fizz", "buzz"
)

test_that("unpiped unpreserved switch_case()", {
  x <- switch_case(
    1:10,
    3 ~ "fizz",
    5 ~ "buzz",
    6 ~ "fizz",
    9 ~ "fizz",
    10 ~ "buzz",
    .preserve = FALSE
  )

  expect_equal(x, yn)

  x <- switch_case(
    1:10,
    3 ~ "fizz",
    5 ~ "buzz",
    6 ~ "fizz",
    9 ~ "fizz",
    10 ~ "buzz"
  )

  expect_equal(x, yn)
})

test_that("unpiped preserved switch_case()", {
  x <- switch_case(
    1:10,
    3 ~ "fizz",
    5 ~ "buzz",
    6 ~ "fizz",
    9 ~ "fizz",
    10 ~ "buzz",
    .preserve = TRUE
  )

  expect_equal(x, yp)
})

test_that("unpiped defaulted switch_case()", {
  x <- switch_case(
    1:10,
    3 ~ "fizz",
    5 ~ "buzz",
    6 ~ "fizz",
    9 ~ "fizz",
    10 ~ "buzz",
    .default = "bam"
  )

  expect_equal(x, yd)
})

test_that("unpiped vectored switch_case()", {
  x <- switch_case(
    1:10,
    c(3, 6, 9) ~ "fizz",
    c(5, 10) ~ "buzz"
  )

  expect_equal(x, yn)
})

test_that("piped unpreserved switch_case()", {
  x <- 1:10 %>%
    switch_case(
      3 ~ "fizz",
      5 ~ "buzz",
      6 ~ "fizz",
      9 ~ "fizz",
      10 ~ "buzz",
      .preserve = FALSE
    )

  expect_equal(x, yn)

  x <- 1:10 %>%
    switch_case(
      3 ~ "fizz",
      5 ~ "buzz",
      6 ~ "fizz",
      9 ~ "fizz",
      10 ~ "buzz"
    )

  expect_equal(x, yn)
})

test_that("piped preserved switch_case()", {
  x <- 1:10 %>%
    switch_case(
      3 ~ "fizz",
      5 ~ "buzz",
      6 ~ "fizz",
      9 ~ "fizz",
      10 ~ "buzz",
      .preserve = TRUE
    )

  expect_equal(x, yp)
})

test_that("piped defaulted switch_case()", {
  x <- 1:10 %>%
    switch_case(
      3 ~ "fizz",
      5 ~ "buzz",
      6 ~ "fizz",
      9 ~ "fizz",
      10 ~ "buzz",
      .default = "bam"
    )

  expect_equal(x, yd)
})

test_that("unpiped vectored switch_case()", {
  x <- 1:10 %>%
    switch_case(
      c(3, 6, 9) ~ "fizz",
      c(5, 10) ~ "buzz"
    )

  expect_equal(x, yn)
})

test_that("exhaustive switch_case()", {
  error <- expect_error(
    switch_case(
      1:10,
      c(3, 6, 9) ~ "fizz",
      c(5, 10) ~ "buzz",
      .exhaustive = TRUE
    ),
    "The following values were not matched: 1, 2, 4, 7, and 8.",
    fixed = TRUE
  )

  expect_equal(error$call[[1]], rlang::sym("switch_case"))

  expect_equal(
    switch_case(
      1:10,
      c(3, 6, 9) ~ "fizz",
      c(5, 10) ~ "buzz",
      1:10 ~ "bam",
      .exhaustive = TRUE
    ),
    yd
  )

  expect_equal(
    switch_case(
      1:10,
      c(3, 6, 9) ~ "fizz",
      c(5, 10) ~ "buzz",
      .preserve = TRUE,
      .exhaustive = TRUE
    ),
    c(1, 2, "fizz", 4, "buzz", "fizz", 7, 8, "fizz", "buzz")
  )
})

test_that("default from other variable switch_case()", {
  input  <- dplyr::tibble(a = letters[1:5], b = 1:5)
  output <- dplyr::tibble(a = letters[1:5], b = 1:5, c = c(10, 2:5))

  x <- dplyr::tibble(a = letters[1:5], b = 1:5) %>%
    dplyr::mutate(
      c = switch_case(
        a,
        "a" ~ 10,
        .default = b
      )
    )

  expect_equal(x, output)

  output <- dplyr::tibble(a = letters[1:5], b = 1:5, c = 1:5)

  x <- dplyr::tibble(a = letters[1:5], b = 1:5) %>%
    dplyr::mutate(
      c = switch_case(
        a,
        "z" ~ 10,
        .default = b
      )
    )

  expect_equal(x, output)
})

test_that("preserve and default", {
  expect_warning(
    switch_case(1:10, 5 ~ "buzz", .preserve = TRUE, .default = "bam")
  )
})

test_that("fn_switch_case()", {
  data <- c(1, 2, 999, 888, 777)

  expect_equal(
    fn_switch_case(
      data,
      function(x) paste(rep(x, 3), collapse = ""),
      7 ~ "Not asked",
      8 ~ "Refused",
      9 ~ "Missing",
      .preserve = TRUE
    ),
    c("1", "2", "Missing", "Refused", "Not asked")
  )
})

test_that("fn_switch_case() exhaustive", {
  data <- c(1, 2, 999, 888, 777)

  error <- expect_error(
    fn_switch_case(
      data,
      function(x) strrep(x, 3),
      7 ~ "Not asked",
      8 ~ "Refused",
      9 ~ "Missing",
      .exhaustive = TRUE
    ),
    "The following values were not matched: 1 and 2.",
    fixed = TRUE
  )

  expect_equal(error$call[[1]], rlang::sym("fn_switch_case"))

  expect_equal(
    fn_switch_case(
      c(111, 222, 999, 888, 777),
      function(x) strrep(x, 3),
      1:2 ~ "Valid",
      7 ~ "Not asked",
      8 ~ "Refused",
      9 ~ "Missing",
      .exhaustive = TRUE
    ),
    c("Valid", "Valid", "Missing", "Refused", "Not asked")
  )
})

test_that("fn_switch_case() errors", {
  expect_error(fn_switch_case(1:10, function(x) x + 5))
})

test_that("warning for deprecated argument", {
  lifecycle::expect_deprecated(
    preserve <- switch_case(
      1:10,
      3 ~ "fizz",
      5 ~ "buzz",
      6 ~ "fizz",
      9 ~ "fizz",
      10 ~ "buzz",
      preserve = FALSE
    ),
    "The `preserve` argument of `switch_case()` is deprecated as of incase 0.3.3.",
    fixed = TRUE
  )

  expect_equal(
    preserve,
    switch_case(
      1:10,
      3 ~ "fizz",
      5 ~ "buzz",
      6 ~ "fizz",
      9 ~ "fizz",
      10 ~ "buzz",
      .preserve = FALSE
    )
  )

  lifecycle::expect_deprecated(
    default <- switch_case(
      1:10,
      3 ~ "fizz",
      5 ~ "buzz",
      6 ~ "fizz",
      9 ~ "fizz",
      10 ~ "buzz",
      default = "bam"
    ),
    "The `default` argument of `switch_case()` is deprecated as of incase 0.3.3.",
    fixed = TRUE
  )

  expect_equal(
    default,
    switch_case(
      1:10,
      3 ~ "fizz",
      5 ~ "buzz",
      6 ~ "fizz",
      9 ~ "fizz",
      10 ~ "buzz",
      .default = "bam"
    )
  )

  data <- c(1, 2, 999, 888, 777)

  lifecycle::expect_deprecated(
    preserve <- fn_switch_case(
      data,
      function(x) paste(rep(x, 3), collapse = ""),
      7 ~ "Not asked",
      8 ~ "Refused",
      9 ~ "Missing",
      preserve = TRUE
    ),
    "The `preserve` argument of `fn_switch_case()` is deprecated as of incase 0.3.3.",
    fixed = TRUE
  )

  expect_equal(
    preserve,
    fn_switch_case(
      data,
      function(x) paste(rep(x, 3), collapse = ""),
      7 ~ "Not asked",
      8 ~ "Refused",
      9 ~ "Missing",
      .preserve = TRUE
    )
  )
})