# Unit tests for "surveydata" class
#
# Author: Andrie
#------------------------------------------------------------------------------

if (interactive()) library(testthat)

{
  sdat <- data.frame(
    id = 1:4,
    Q1 = c("Yes", "No", "Yes", "Yes"),
    Q4_1 = c(1, 2, 1, 2),
    Q4_2 = c(3, 4, 4, 3),
    Q4_3 = c(5, 5, 6, 6),
    Q4_other = c(NA, NA, "some text", NA),
    Q10 = factor(c("Male", "Female", "Female", "Male")),
    crossbreak = c("A", "A", "B", "B"),
    crossbreak2 = c("D", "E", "D", "E"),
    weight = c(0.9, 1.1, 0.8, 1.2)
  )

  varlabels(sdat) <- setNames(
    c(
      "RespID",
      "Question 1",
      "Question 4: red", "Question 4: green", "Question 4: blue",
      "Question 4: Other",
      "Question 10",
      "crossbreak",
      "crossbreak2",
      "weight"
    ),
    names(sdat)
  )
}


#------------------------------------------------------------------------------



test_that("question_text, qTextCommon and qTextUnique work as expected", {
  s <- as.surveydata(sdat)
  expect_equal(question_text(s, "Q1"), "Question 1")
  expect_equal(question_text(s, "Q4"), c("Question 4: red", "Question 4: green", "Question 4: blue"))
  expect_equal(question_text(s, "Q10"), "Question 10")
  expect_equal(question_text(s, "Q99"), character(0))

  expect_equal(question_text_common(s, "Q4"), "Question 4")
  expect_equal(question_text_unique(s, "Q4"), c("red", "green", "blue"))

  expect_equal(questions(s), c(
    "id",
    "Q1",
    "Q4",
    "Q4_other",
    "Q10",
    "crossbreak",
    "crossbreak2",
    "weight"
  ))

  expect_equal(
    question_text(s),
    sapply(questions(s), question_text, x = s)
  )
})

#------------------------------------------------------------------------------

{
  sdat2 <- data.frame(
    id = 1:4,
    Q1 = c("Yes", "No", "Yes", "Yes"),
    Q4__1 = c(1, 2, 1, 2),
    Q4__2 = c(3, 4, 4, 3),
    Q4__3 = c(5, 5, 6, 6),
    Q4__ignore = c(NA, NA, "some text", NA),
    Q10 = factor(c("Male", "Female", "Female", "Male")),
    crossbreak = c("A", "A", "B", "B"),
    crossbreak2 = c("D", "E", "D", "E"),
    weight = c(0.9, 1.1, 0.8, 1.2),
    check.names = FALSE
  )

  varlabels(sdat2) <- c(
    "RespID",
    "Question 1",
    "Question 4: red", "Question 4: green", "Question 4: blue",
    "Question 4: Other",
    "Question 10",
    "crossbreak",
    "crossbreak2",
    "weight"
  )
}

# context("Questions 2")
test_that("question_text, qTextCommon and qTextUnique work as expected", {
  s2 <- as.surveydata(sdat2, sep = "__", exclude = "ignore", renameVarlabels = TRUE)
  expect_equal(question_text(s2, "Q1"), "Question 1")
  expect_equal(question_text(s2, "Q4"), c("Question 4: red", "Question 4: green", "Question 4: blue"))
  expect_equal(question_text(s2, "Q10"), "Question 10")
  expect_equal(question_text(s2, "Q99"), character(0))

  expect_equal(question_text_common(s2, "Q4"), "Question 4")
  expect_equal(question_text_unique(s2, "Q4"), c("red", "green", "blue"))

  expect_equal(questions(s2), c(
    "id",
    "Q1",
    "Q4",
    "Q4__ignore",
    "Q10",
    "crossbreak",
    "crossbreak2",
    "weight"
  ))
})


#------------------------------------------------------------------------------

# context("split_common_unique")
test_that("split_common_unique works as expected", {
  test <- c("Email (Please tell us)", "Phone (Please tell us)")
  exp <- list(common = "Please tell us", unique = c("Email", "Phone"))
  expect_equal(split_common_unique(test), exp)

  test <- c("What is your choice?: Email", "What is your choice?: Phone")
  exp <- list(common = "What is your choice?", unique = c("Email", "Phone"))
  expect_equal(split_common_unique(test), exp)

  test <- c("What is your choice?:Email", "What is your choice?:Phone")
  exp <- list(common = "What is your choice?", unique = c("Email", "Phone"))
  expect_equal(split_common_unique(test), exp)

  test <- c("Q3(001)Email", "Q3(001)Phone")
  exp <- list(common = "Q3", unique = c("Email", "Phone"))
  expect_equal(split_common_unique(test), exp)

  test <- c("Q3(001) Email", "Q3(001) Phone")
  exp <- list(common = "Q3", unique = c("Email", "Phone"))
  expect_equal(split_common_unique(test), exp)

  test <- c("Q3[001] Email", "Q3[001] Phone")
  exp <- list(common = "Q3", unique = c("Email", "Phone"))
  expect_equal(split_common_unique(test), exp)

  test <- c("Q3[01]Email", "Q3[01]Phone")
  exp <- list(common = "Q3", unique = c("Email", "Phone"))
  expect_equal(split_common_unique(test), exp)

  test <- c("[Email]What is your choice?", "[Phone]What is your choice?")
  exp <- list(common = "What is your choice?", unique = c("Email", "Phone"))
  expect_equal(split_common_unique(test), exp)

  test <- c("What is your choice? [Email]", "What is your choice? [Phone]")
  exp <- list(common = "What is your choice?", unique = c("Email", "Phone"))
  expect_equal(split_common_unique(test), exp)

  test <- c(
    "Question (answer 1 (with embedded parens))",
    "Question (answer 2 without embedded parens)"
  )
  exp <- list(common = "Question", unique = c(
    "answer 1 (with embedded parens)",
    "answer 2 without embedded parens"
  ))
  expect_equal(split_common_unique(test), exp)

  test <- c("Q_1", "Q_2")
  exp <- list(common = "Q_", unique = c("1", "2"))
  expect_equal(split_common_unique(test), exp)
})