testthat::context("attr labs")
library(labelled)
library(ggplot2)
library(ggeasy)

expect_eqNe <- function(...) expect_equal(..., check.environment = FALSE)

expect_doppelganger <- function(title, fig, ...) {
  testthat::skip_if_not_installed("vdiffr")
  vdiffr::expect_doppelganger(title, fig, ...)
}

iris_labs <- iris

lbl <- c(
  'Sepal Length',
  'Sepal Width',
  'Petal Length',
  'Petal Width',
  'Flower Species'
)

labelled::var_label(iris_labs) <- split(lbl, names(iris_labs))

p <- ggplot2::ggplot(
  iris_labs,
  ggplot2::aes(x = Sepal.Length, y = Sepal.Width)
) +
  ggplot2::geom_line(ggplot2::aes(colour = Species))

test_that("easy_labs uses column attrib", {
  skip_if(utils::packageVersion("ggplot2") >= package_version("3.5.2.9001"))
  hard_res <- p +
    ggplot2::labs(
      x = 'Sepal Length',
      y = 'Sepal Width',
      colour = 'Flower Species'
    )
  easy_res <- p + easy_labs()

  expect_mapequal(get_labs(easy_res), get_labs(hard_res))
  expect_message(
    p + easy_labs(teach = TRUE),
    regexp = "colour = 'Flower Species'"
  )

  easy_res_test_teach <- p + easy_labs()
  teach_message <- capture_messages({
    p + easy_labs(teach = TRUE)
  })[2]
  teach_res <- rlang::eval_bare(rlang::parse_expr(teach_message))
  expect_eqNe(easy_res_test_teach, p + teach_res)

  expect_doppelganger("labels_attrib", easy_res)
})

testthat::test_that("regular labs overides easy_labs ", {
  skip_if(utils::packageVersion("ggplot2") >= package_version("3.5.2.9001"))
  hard_res <- p +
    ggplot2::labs(x = 'x axis', y = 'Sepal Width', colour = 'Flower Species')
  easy_res <- p + easy_labs(x = 'x axis')

  expect_mapequal(get_labs(easy_res), get_labs(hard_res))
  expect_doppelganger("labels_manual", easy_res)
})

testthat::test_that("regular labs pass new labels through easy_labs ", {
  skip_if(utils::packageVersion("ggplot2") >= package_version("3.5.2.9001"))
  hard_res <- p +
    ggplot2::labs(
      x = 'Sepal Length',
      y = 'Sepal Width',
      colour = 'Flower Species',
      title = 'my title'
    )
  easy_res <- p + easy_labs(title = 'my title')

  expect_eqNe(
    easy_res$labels[sort(names(easy_res$labels))],
    hard_res$labels[sort(names(hard_res$labels))]
  )
  expect_doppelganger("labels_mytitle", easy_res)
})

test_that("column name used when no column attrib present", {
  skip_if(utils::packageVersion("ggplot2") >= package_version("3.5.2.9001"))
  labelled::var_label(iris_labs$Sepal.Length) <- NULL

  p <- ggplot2::ggplot(
    iris_labs,
    ggplot2::aes(x = Sepal.Length, y = Sepal.Width)
  ) +
    ggplot2::geom_line(ggplot2::aes(colour = Species))

  hard_res <- p + ggplot2::labs(y = 'Sepal Width', colour = 'Flower Species')
  easy_res <- p + ggeasy::easy_labs()

  expect_mapequal(get_labs(easy_res), get_labs(hard_res))
  expect_doppelganger("labels_y_col", easy_res)
})