test_that("gizmo_barcap trains correctly", {

  scale <- scale_colour_viridis_c(na.value = "#FF0000")
  scale$train(c(10, 30))

  guide <- gizmo_barcap(key = key_sequence(11))

  # No out-of-bounds
  params <- guide$train(guide$params, scale, "colour")
  expect_equal(params$limits, c(10, 30))
  expect_equal(params$key$.value, seq(10, 30, length.out = 11))
  expect_equal(params$key$colour[c(1, 11)], c("#440154", "#FDE725"))

  # Upper out-of-bounds
  scale <- scale_colour_viridis_c(limits = c(10, 30), na.value = "#FF0000")
  scale$train(c(10, 40))

  params <- guide$train(guide$params, scale, "colour")
  expect_equal(params$limits, c(10, 30.0391), tolerance = 1e-3)
  expect_equal(nrow(params$key), 12)
  expect_equal(params$key$colour[c(1, 12)], c("#440154", "#FF0000"))

  # Lower out-of-bounds
  scale <- scale_colour_viridis_c(limits = c(10, 30), na.value = "#FF0000")
  scale$train(c(0, 30))

  params <- guide$train(guide$params, scale, "colour")
  expect_equal(params$limits, c(9.98, 30), tolerance = 1e-3)
  expect_equal(nrow(params$key), 12)
  expect_equal(params$key$colour[c(1, 12)], c("#FF0000", "#FDE725"))

  # Both out of bounds
  scale <- scale_colour_viridis_c(limits = c(10, 30), na.value = "#FF0000")
  scale$train(c(0, 40))

  params <- guide$train(guide$params, scale, "colour")
  expect_equal(params$limits, c(9.98, 30.02), tolerance = 1e-3)
  expect_equal(nrow(params$key), 13)
  expect_equal(params$key$colour[c(1, 13)], c("#FF0000", "#FF0000"))

  # Can ignore out of bounds by setting upper/lower to FALSE
  params <- guide$params
  params$show <- c(FALSE, FALSE)
  params <- guide$train(params, scale, "colour")
  expect_equal(params$limits, c(10, 30))
  expect_equal(nrow(params$key), 11)
  expect_equal(params$key$colour[c(1, 11)], c("#440154", "#FDE725"))

  # Can include caps even when all in bounds
  scale <- scale_colour_viridis_c(na.value = "#FF0000")
  scale$train(c(10, 30))

  params <- guide$params
  params$show <- c(TRUE, TRUE)
  params <- guide$train(params, scale, "colour")
  expect_equal(params$limits, c(9.98, 30.02), tolerance = 1e-3)
  expect_equal(nrow(params$key), 13)
  expect_equal(params$key$colour[c(1, 13)], c("#FF0000", "#FF0000"))

  # Can also squish into limits
  params <- guide$params
  params$show <- c(TRUE, TRUE)
  params$oob <- oob_squish
  params <- guide$train(params, scale, "colour")
  expect_equal(params$limits, c(10, 30))
  expect_equal(nrow(params$key), 13)
  expect_equal(params$key$colour[c(1, 13)], c("#440154", "#FDE725"))
})

test_that("gizmo_barcap shows caps correctly", {
  p <- ggplot(mpg, aes(displ, hwy, colour = cty)) + geom_point() +
    scale_colour_viridis_c(
      oob = oob_squish,
      guide = gizmo_barcap(show = TRUE)
    ) +
    theme(legend.frame = element_rect(colour = "black"),
          legend.key.size = unit(1, "cm"))
  b <- ggplot_build(p)

  theme <- ggplot2:::plot_theme(b$plot)
  theme$legend.key.width  <- theme$legend.key.size
  theme$legend.key.height <- theme$legend.key.size

  guide <- b$plot$guides
  old_params <- guide$params[[1]]
  guide <- guide$guides[[1]]

  draw_guide <- function(guide, params, theme) {
    hor <- guide$draw(theme, position = "right", direction = "horizontal", params)
    ver <- guide$draw(theme, position = "right", direction = "vertical", params)
    gt <- gtable(width = unit(c(1, 1), "null"), heights = unit(1, "null"))
    gtable_add_grob(gt, list(hor, ver), t = 1, l = 1:2)
  }

  params <- old_params
  gt <- draw_guide(guide, params, theme)
  vdiffr::expect_doppelganger("triangle-cap", gt)

  params$size <- unit(2, "cm")
  gt <- draw_guide(guide, params, theme)
  vdiffr::expect_doppelganger("triangle-cap-long", gt)

  params$size <- NULL
  params$shape <- cap_round(n = 30)
  gt <- draw_guide(guide, params, theme)
  vdiffr::expect_doppelganger("round-cap", gt)

  params$size <- unit(2, "cm")
  params$shape <- cap_round(n = 30)
  gt <- draw_guide(guide, params, theme)
  vdiffr::expect_doppelganger("round-cap-long", gt)

  params$size <- NULL
  params$shape <- cap_arch(n = 30)
  gt <- draw_guide(guide, params, theme)
  vdiffr::expect_doppelganger("arch-cap", gt)

  params$size <- unit(2, "cm")
  params$shape <- cap_arch(n = 30)
  gt <- draw_guide(guide, params, theme)
  vdiffr::expect_doppelganger("arch-cap-long", gt)

  params$size <- NULL
  params$shape <- cap_ogee(n = 30)
  gt <- draw_guide(guide, params, theme)
  vdiffr::expect_doppelganger("ogee-cap", gt)

  params$size <- unit(2, "cm")
  params$shape <- cap_ogee(n = 30)
  gt <- draw_guide(guide, params, theme)
  vdiffr::expect_doppelganger("ogee-cap-long", gt)

})