context("join_header()")

expect_purpose <-
  c(
    "0 - 6", "7 - 10", "7 - 10", "0 - 6", "7 - 10", "0 - 6", "7 - 10",
    "0 - 6", "7 - 10", "0 - 6", "7 - 10", "7 - 10", "0 - 6", "7 - 10",
    "0 - 6", "7 - 10", "0 - 6", "7 - 10", "0 - 6", "7 - 10", "0 - 6",
    "7 - 10", "0 - 6", "7 - 10", "0 - 6", "7 - 10", "0 - 6", "7 - 10",
    "0 - 6", "7 - 10", "7 - 10", "0 - 6", "7 - 10", "0 - 6", "7 - 10",
    "0 - 6", "7 - 10", "0 - 6", "7 - 10", "0 - 6", "7 - 10", "0 - 6",
    "7 - 10", "0 - 6", "7 - 10", "0 - 6", "7 - 10", "0 - 6", "7 - 10",
    "0 - 6", "7 - 10", "0 - 6", "7 - 10", "0 - 6", "7 - 10", "0 - 6",
    "7 - 10", "0 - 6", "7 - 10", "0 - 6", "7 - 10", "7 - 10", "0 - 6",
    "7 - 10", "0 - 6", "7 - 10", "0 - 6", "7 - 10", "0 - 6", "7 - 10",
    "7 - 10", "7 - 10"
  )

expect_sex <-
  c(
    "Female", "Female", "Male", "Female", "Female", "Male", "Male",
    "Female", "Female", "Male", "Male", "Female", "Male", "Male",
    "Female", "Female", "Male", "Male", "Female", "Female", "Male",
    "Male", "Female", "Female", "Male", "Male", "Female", "Female",
    "Male", "Male", "Female", "Male", "Male", "Female", "Female",
    "Male", "Male", "Female", "Female", "Male", "Male", "Female",
    "Female", "Male", "Male", "Female", "Female", "Male", "Male",
    "Female", "Female", "Male", "Male", "Female", "Female", "Male",
    "Male", "Female", "Female", "Male", "Male", "Female", "Female",
    "Female", "Male", "Male", "Female", "Female", "Male", "Male",
    "Female", "Male"
  )

expect_age <-
  c(
    "15 - 24", "15 - 24", "15 - 24", "25 - 44", "25 - 44", "25 - 44",
    "25 - 44", "45 - 64", "45 - 64", "45 - 64", "45 - 64", "65+",
    "65+", "65+", "15 - 24", "15 - 24", "15 - 24", "15 - 24", "25 - 44",
    "25 - 44", "25 - 44", "25 - 44", "45 - 64", "45 - 64", "45 - 64",
    "45 - 64", "65+", "65+", "65+", "65+", "15 - 24", "15 - 24",
    "15 - 24", "25 - 44", "25 - 44", "25 - 44", "25 - 44", "45 - 64",
    "45 - 64", "45 - 64", "45 - 64", "65+", "65+", "65+", "65+",
    "15 - 24", "15 - 24", "15 - 24", "15 - 24", "25 - 44", "25 - 44",
    "25 - 44", "25 - 44", "45 - 64", "45 - 64", "45 - 64", "45 - 64",
    "65+", "65+", "65+", "65+", "15 - 24", "25 - 44", "25 - 44",
    "25 - 44", "25 - 44", "45 - 64", "45 - 64", "45 - 64", "45 - 64",
    "65+", "65+"
  )

expect_education <-
  c(
    "Bachelor's degree", "Bachelor's degree", "Bachelor's degree",
    "Bachelor's degree", "Bachelor's degree", "Bachelor's degree",
    "Bachelor's degree", "Bachelor's degree", "Bachelor's degree",
    "Bachelor's degree", "Bachelor's degree", "Bachelor's degree",
    "Bachelor's degree", "Bachelor's degree", "Certificate", "Certificate",
    "Certificate", "Certificate", "Certificate", "Certificate", "Certificate",
    "Certificate", "Certificate", "Certificate", "Certificate", "Certificate",
    "Certificate", "Certificate", "Certificate", "Certificate", "Diploma",
    "Diploma", "Diploma", "Diploma", "Diploma", "Diploma", "Diploma",
    "Diploma", "Diploma", "Diploma", "Diploma", "Diploma", "Diploma",
    "Diploma", "Diploma", "No Qualification", "No Qualification",
    "No Qualification", "No Qualification", "No Qualification", "No Qualification",
    "No Qualification", "No Qualification", "No Qualification", "No Qualification",
    "No Qualification", "No Qualification", "No Qualification", "No Qualification",
    "No Qualification", "No Qualification", "Postgraduate qualification",
    "Postgraduate qualification", "Postgraduate qualification", "Postgraduate qualification",
    "Postgraduate qualification", "Postgraduate qualification", "Postgraduate qualification",
    "Postgraduate qualification", "Postgraduate qualification", "Postgraduate qualification",
    "Postgraduate qualification"
  )

expect_purpose_short <-
  c(
    "0 - 6", "0 - 6", "0 - 6", "7 - 10", "7 - 10", "7 - 10", "0 - 6",
    "0 - 6", "0 - 6", "7 - 10", "7 - 10", "7 - 10", "0 - 6", "0 - 6",
    "7 - 10", "7 - 10", "7 - 10", "0 - 6", "0 - 6", "0 - 6", "7 - 10",
    "7 - 10", "7 - 10", "0 - 6", "0 - 6", "7 - 10", "7 - 10", "7 - 10",
    "0 - 6", "0 - 6", "7 - 10", "7 - 10", "7 - 10", "0 - 6", "0 - 6",
    "0 - 6", "7 - 10", "7 - 10", "7 - 10", "0 - 6", "0 - 6", "0 - 6",
    "7 - 10", "7 - 10", "7 - 10", "0 - 6", "0 - 6", "0 - 6", "7 - 10",
    "7 - 10", "7 - 10", "0 - 6", "0 - 6", "7 - 10", "7 - 10"
  )

expect_sex_short <-
  c(
    "Female", "Female", "Female", "Female", "Female", "Female",
    "Female", "Female", "Female", "Female", "Female", "Female", "Female",
    "Female", "Female", "Female", "Female", "Female", "Female", "Female",
    "Female", "Female", "Female", "Female", "Female", "Female", "Female",
    "Female", "Male", "Male", "Male", "Male", "Male", "Male", "Male",
    "Male", "Male", "Male", "Male", "Male", "Male", "Male", "Male",
    "Male", "Male", "Male", "Male", "Male", "Male", "Male", "Male",
    "Male", "Male", "Male", "Male"
  )

expect_age_short <-
  c(
    "15 - 24", "25 - 44", "45 - 64", "15 - 24", "25 - 44", "45 - 64",
    "15 - 24", "25 - 44", "45 - 64", "15 - 24", "25 - 44", "45 - 64",
    "25 - 44", "45 - 64", "15 - 24", "25 - 44", "45 - 64", "15 - 24",
    "25 - 44", "45 - 64", "15 - 24", "25 - 44", "45 - 64", "25 - 44",
    "45 - 64", "15 - 24", "25 - 44", "45 - 64", "25 - 44", "45 - 64",
    "15 - 24", "25 - 44", "45 - 64", "15 - 24", "25 - 44", "45 - 64",
    "15 - 24", "25 - 44", "45 - 64", "15 - 24", "25 - 44", "45 - 64",
    "15 - 24", "25 - 44", "45 - 64", "15 - 24", "25 - 44", "45 - 64",
    "15 - 24", "25 - 44", "45 - 64", "25 - 44", "45 - 64", "25 - 44",
    "45 - 64"
  )

expect_education_short <-
  c(
    "Bachelor's degree", "Bachelor's degree", "Bachelor's degree",
    "Bachelor's degree", "Bachelor's degree", "Bachelor's degree",
    "Certificate", "Certificate", "Certificate", "Certificate", "Certificate",
    "Certificate", "Diploma", "Diploma", "Diploma", "Diploma", "Diploma",
    "No Qualification", "No Qualification", "No Qualification", "No Qualification",
    "No Qualification", "No Qualification", "Postgraduate qualification",
    "Postgraduate qualification", "Postgraduate qualification", "Postgraduate qualification",
    "Postgraduate qualification", "Bachelor's degree", "Bachelor's degree",
    "Bachelor's degree", "Bachelor's degree", "Bachelor's degree",
    "Certificate", "Certificate", "Certificate", "Certificate", "Certificate",
    "Certificate", "Diploma", "Diploma", "Diploma", "Diploma", "Diploma",
    "Diploma", "No Qualification", "No Qualification", "No Qualification",
    "No Qualification", "No Qualification", "No Qualification", "Postgraduate qualification",
    "Postgraduate qualification", "Postgraduate qualification", "Postgraduate qualification"
  )

test_that("Compass directions \"up\", \"up-left\", \"left\", and \"left-up\" work", {
  cells <- as_cells(purpose$`up-left left-up`)
  col_headers <-
    cells %>%
    dplyr::filter(row <= 2, !is.na(chr)) %>%
    dplyr::select(row, col, header = chr) %>%
    split(.$row)
  row_headers <-
    cells %>%
    dplyr::filter(col <= 2, !is.na(chr)) %>% # Select all rows of headers at once
    dplyr::select(row, col, header = chr) %>%
    split(.$col) # Return each row of headers in its own element of a list
  data_cells <-
    cells %>%
    dplyr::filter(row >= 3, col >= 3, !is.na(chr)) %>%
    dplyr::mutate(value = as.integer(chr)) %>%
    dplyr::select(row, col, value) %>%
    enhead(col_headers[[1]], "up-left") %>%
    enhead(col_headers[[2]], "up") %>%
    enhead(row_headers[[1]], "left-up") %>%
    enhead(row_headers[[2]], "left") %>%
    dplyr::arrange(row, col)
  expect_equal(data_cells[[4]], expect_sex)
  expect_equal(data_cells[[5]], expect_purpose)
  expect_equal(data_cells[[6]], expect_education)
  expect_equal(data_cells[[7]], expect_age)
})

test_that("Compass directions \"up-right\" and \"left-down\" work", {
  cells <- as_cells(purpose$`up-right left-down`)
  row_headers <-
    cells %>%
    dplyr::filter(col <= 2, !is.na(chr)) %>%
    dplyr::select(row, col, header = chr) %>%
    split(.$col)
  col_headers <-
    cells %>%
    dplyr::filter(row <= 2, !is.na(chr)) %>%
    dplyr::select(row, col, header = chr) %>%
    split(.$row)
  data_cells <-
    cells %>%
    dplyr::filter(row >= 3, col >= 3, !is.na(chr)) %>%
    dplyr::mutate(value = as.integer(chr)) %>%
    dplyr::select(row, col, value) %>%
    enhead(col_headers[[1]], "up-right") %>%
    enhead(row_headers[[1]], "left-down") %>%
    dplyr::arrange(row, col)
  expect_equal(data_cells[[5]], expect_education)
  expect_equal(data_cells[[4]], expect_sex)
})

test_that("Compass directions \"down\", \"down-right\" , \"right\" and \"right-down\" work", {
  cells <- as_cells(purpose$`right-down down-right`)
  row_headers <-
    cells %>%
    dplyr::filter(col >= 5, !is.na(chr)) %>%
    dplyr::select(row, col, header = chr) %>%
    split(.$col)
  col_headers <-
    cells %>%
    dplyr::filter(row >= 21, !is.na(chr)) %>%
    dplyr::select(row, col, header = chr) %>%
    split(.$row)
  data_cells <-
    cells %>%
    dplyr::filter(row <= 20, col <= 4, !is.na(chr)) %>%
    dplyr::mutate(value = as.integer(chr)) %>%
    dplyr::select(row, col, value) %>%
    enhead(col_headers[[2]], "down-right") %>%
    enhead(col_headers[[1]], "down") %>%
    enhead(row_headers[[2]], "right-down") %>%
    enhead(row_headers[[1]], "right") %>%
    dplyr::arrange(row, col)
  expect_equal(data_cells[[4]], expect_sex)
  expect_equal(data_cells[[5]], expect_purpose)
  expect_equal(data_cells[[6]], expect_education)
  expect_equal(data_cells[[7]], expect_age)
})

test_that("Compass directions \"down-left\" and \"right-up\" work", {
  cells <- as_cells(purpose$`right-up down-left`)
  row_headers <-
    cells %>%
    dplyr::filter(col >= 5, !is.na(chr)) %>%
    dplyr::select(row, col, header = chr) %>%
    split(.$col)
  col_headers <-
    cells %>%
    dplyr::filter(row >= 21, !is.na(chr)) %>%
    dplyr::select(row, col, header = chr) %>%
    split(.$row)
  data_cells <-
    cells %>%
    dplyr::filter(row <= 20, col <= 4, !is.na(chr)) %>%
    dplyr::mutate(value = as.integer(chr)) %>%
    dplyr::select(row, col, value) %>%
    enhead(col_headers[[2]], "down-left") %>%
    enhead(row_headers[[2]], "right-up") %>%
    dplyr::arrange(row, col)
  expect_equal(data_cells[[5]], expect_education)
  expect_equal(data_cells[[4]], expect_sex)
})

test_that("Compass directions \"up-ish\" and \"left-ish\" work", {
  cells <- as_cells(purpose$`up-ish left-ish`)
  row_headers <-
    cells %>%
    dplyr::filter(col <= 2, !is.na(chr)) %>%
    dplyr::select(row, col, header = chr) %>%
    split(.$col)
  col_headers <-
    cells %>%
    dplyr::filter(row <= 2, !is.na(chr)) %>%
    dplyr::select(row, col, header = chr) %>%
    split(.$row)
  data_cells <-
    cells %>%
    dplyr::filter(row >= 3, col >= 3, !is.na(chr)) %>%
    dplyr::mutate(value = as.integer(chr)) %>%
    dplyr::select(row, col, value) %>%
    enhead(col_headers[[1]], "up-ish", drop = FALSE) %>%
    enhead(row_headers[[1]], "left-ish", drop = FALSE) %>%
    enhead(col_headers[[2]], "up-ish", drop = FALSE) %>%
    enhead(row_headers[[2]], "left-ish", drop = FALSE) %>%
    dplyr::arrange(row, col)
  expect_equal(data_cells[[5]], expect_sex_short)
  expect_equal(data_cells[[4]], expect_purpose_short)
})

test_that("Compass directions \"down-ish\" and \"right-ish\" work", {
  cells <- as_cells(purpose$`right-ish down-ish`)
  row_headers <-
    cells %>%
    dplyr::filter(col >= 7, !is.na(chr)) %>%
    dplyr::select(row, col, header = chr) %>%
    split(.$col)
  col_headers <-
    cells %>%
    dplyr::filter(row >= 11, !is.na(chr)) %>%
    dplyr::select(row, col, header = chr) %>%
    split(.$row)
  data_cells <-
    cells %>%
    dplyr::filter(row <= 10, col <= 6, !is.na(chr)) %>%
    dplyr::mutate(value = as.integer(chr)) %>%
    dplyr::select(row, col, value) %>%
    enhead(col_headers[[2]], "down-ish") %>%
    enhead(row_headers[[2]], "right-ish") %>%
    dplyr::arrange(row, col)
  expect_equal(data_cells[[5]], expect_sex_short)
  expect_equal(data_cells[[4]], expect_purpose_short)
})

test_that("enhead() works", {
  cells <- as_cells(purpose$`up-left left-up`)
  col_headers <-
    cells %>%
    dplyr::filter(row <= 2, !is.na(chr)) %>%
    dplyr::select(row, col, header = chr) %>%
    split(.$row)
  data_cells <-
    cells %>%
    dplyr::filter(row >= 3, col >= 3, !is.na(chr)) %>%
    dplyr::mutate(value = as.integer(chr)) %>%
    dplyr::select(row, col, value)
  multirow_header <-
    cells %>%
    dplyr::filter(row %in% 1:2, !is.na(chr)) %>%
    dplyr::select(row, col, header = chr)
  expect_error(
    enhead(data_cells, col_headers[[1]], "NORTH"),
		"The direction \"NORTH\" is not recognised.  See \\?directions."
  )
  expect_error(
    enhead(data_cells, multirow_header, "left"),
    "Multiple lines of headers are not supported in this way*"
  )
})

test_that("the `drop` argument works", {
  spreadsheet <- system.file("extdata/purpose.xlsx", package = "unpivotr")
  cells <- tidyxl::xlsx_cells(spreadsheet, "up-ish left-ish border")
  formatting <- tidyxl::xlsx_formats(spreadsheet)
  left_borders <- which(!is.na(formatting$local$border$left$style))
  top_borders <- which(!is.na(formatting$local$border$top$style))
  left_border_cells <-
    dplyr::filter(cells, row == 2, local_format_id %in% left_borders) %>%
    dplyr::select(row, col)
  top_border_cells <-
    dplyr::filter(cells, col == 2, local_format_id %in% top_borders) %>%
    dplyr::select(row, col)
  sex <-
    dplyr::filter(cells, col == 2, !is_blank) %>%
    dplyr::select(row, col, sex = character)
  qualification <-
    dplyr::filter(cells, col == 3, !is_blank) %>%
    dplyr::select(row, col, qualification = character)
  satisfaction <-
    dplyr::filter(cells, row == 2, !is_blank) %>%
    dplyr::select(row, col, satisfaction = character)
  age <-
    dplyr::filter(cells, row == 3, !is_blank) %>%
    dplyr::select(row, col, age = character)
  data_cells <-
    dplyr::filter(cells, row >= 4, col >= 4, !is_blank) %>%
    dplyr::transmute(row, col, value = as.integer(numeric))
  expect_equal(nrow(enhead(data_cells, satisfaction, "up")), 20)
  expect_equal(nrow(enhead(data_cells, satisfaction, "up", drop = FALSE)), 55)
  expect_equal(nrow(enhead(data_cells, sex, "left")), 12)
  expect_equal(nrow(enhead(data_cells, sex, "left", drop = FALSE)), 55)
  expect_equal(nrow(enhead(data_cells, satisfaction, "up-left")), 39)
  expect_equal(nrow(enhead(data_cells, satisfaction, "up-left", drop = FALSE)), 55)
  expect_equal(nrow(enhead(data_cells, sex, "left-up")), 49)
  expect_equal(nrow(enhead(data_cells, sex, "left-up", drop = FALSE)), 55)
})

test_that("enhead() stops on non-distinct cells", {
  cells <- as_cells(purpose$`up-left left-up`)
  col_headers <-
    cells %>%
    dplyr::filter(row <= 2, !is.na(chr)) %>%
    dplyr::select(row, col, header = chr) %>%
    split(.$row)
  row_headers <-
    cells %>%
    dplyr::filter(col <= 2, !is.na(chr)) %>% # Select all rows of headers at once
    dplyr::select(row, col, header = chr) %>%
    split(.$col) # Return each row of headers in its own element of a list
  data_cells <-
    cells %>%
    dplyr::filter(row >= 3, col >= 3, !is.na(chr)) %>%
    dplyr::mutate(value = as.integer(chr)) %>%
    dplyr::select(row, col, value)
  expect_error(
    enhead(dplyr::bind_rows(cells, cells), col_headers[[1]], "up-left"),
    "Row and column numbers must be distinct.\n  Perhaps you meant to use a single sheet.",
    fixed = TRUE
  )
})