test_that("find_unique_key returns base_id when already unique", {
  data <- data.table::data.table(
    part_id = 1:100,
    wave = rep(1:2, 50),
    age = sample(20:60, 100, replace = TRUE)
  )
  expect_identical(find_unique_key(data, "part_id"), "part_id")
})

test_that("find_unique_key finds single column extension", {
  data <- data.table::data.table(
    part_id = rep(1:50, each = 2),
    wave = rep(1:2, 50),
    age = sample(20:60, 100, replace = TRUE)
  )
  result <- find_unique_key(data, "part_id")
  expect_identical(result, c("part_id", "wave"))
})

test_that("find_unique_key finds two-column extension", {
  # Create data where part_id + wave + studyDay is needed
  data <- data.table::rbindlist(list(
    data.table::data.table(
      part_id = rep(1:50, each = 2),
      wave = 1,
      studyDay = rep(1:2, 50)
    ),
    data.table::data.table(
      part_id = rep(25:74, each = 2),
      wave = 2,
      studyDay = rep(1:2, 50)
    )
  ))
  result <- find_unique_key(data, "part_id")
  # Should find wave + studyDay (both have 2 unique values)
  expect_length(result, 3)
  expect_true("part_id" %in% result)
  expect_identical(anyDuplicated(data, by = result), 0L)
})

test_that("find_unique_key returns NULL when no unique key exists", {
  # True many-to-many: no combination makes it unique
  data <- data.table::data.table(
    part_id = rep(1:10, each = 10),
    category = rep(rep(c("A", "B"), each = 5), 10)
  )
  expect_null(find_unique_key(data, "part_id"))
})

test_that("find_unique_key ignores internal columns", {
  data <- data.table::data.table(
    part_id = rep(1:50, each = 2),
    wave = rep(1:2, 50),
    ..main_id = 1:100,
    ..merge_id = 1:100
  )
  result <- find_unique_key(data, "part_id")
  expect_identical(result, c("part_id", "wave"))
})

test_that("load_survey handles longitudinal data with sday files", {
  skip_if_offline("zenodo.org")
  skip_on_cran()
  skip_on_ci()

  # Beraud France has longitudinal structure with wave and studyDay
  files <- suppressMessages(suppressWarnings(
    download_survey("10.5281/zenodo.3886590") # nolint
  ))
  survey <- suppressMessages(load_survey(files))

  # Check that sday columns are present (wave, studyDay, etc.)
  expect_true("wave" %in% names(survey$participants))
  expect_true("studyDay" %in% names(survey$participants))

  # Check that we have multiple rows per participant
  part_counts <- survey$participants[, .N, by = part_id]
  expect_true(any(part_counts$N > 1))

  # Check that observation_key is stored (without part_id, just the
  # distinguishing columns)
  expect_false(is.null(survey$observation_key))
  expect_false("part_id" %in% survey$observation_key)
  # Beraud France has wave and studyDay creating the longitudinal structure
  expect_true(all(c("wave", "studyDay") %in% survey$observation_key))
})