dbpath <- rbedrock_example_world("example1.mcworld")
db <- bedrockdb(dbpath)

test_that("put_legacy_biomes_value() accepts integers", {
  set.seed(102032)
  put_legacy_biomes_value(1L, 32, 3, 0, db = db)
  dat <- get_legacy_biomes_value(32, 3, 0, return_names = FALSE, db = db)
  expect_equal(dat, matrix(1L, 16L, 16L))

  new_dat <- sample(c(1, 10, 20), 256, replace = TRUE)
  put_legacy_biomes_value(new_dat, 32, 3, 0, db = db)
  dat <- get_legacy_biomes_value(32, 3, 0, return_names = FALSE, db = db)
  expect_equal(dat, matrix(new_dat, 16L, 16L))
})

test_that("put_legacy_biomes_value() accepts strings", {
  set.seed(102031)
  put_legacy_biomes_value("plains", 32, 3, 0, db = db)
  dat <- get_legacy_biomes_value(32, 3, 0, db = db)
  expect_equal(dat, matrix("plains", 16L, 16L))

  new_dat <- sample(c("plains", "desert", "ocean"), 256, replace = TRUE)
  put_legacy_biomes_value(new_dat, 32, 3, 0, db = db)
  dat <- get_legacy_biomes_value(32, 3, 0, db = db)
  expect_equal(dat, matrix(new_dat, 16L, 16L))

  put_legacy_biomes_value("plain:fake_biome", 32, 3, 0, db = db)
  dat <- get_legacy_biomes_value(32, 3, 0, db = db)
  expect_equal(dat, matrix("ocean", 16L, 16L))
})

test_that("put_legacy_biomes_data() accepts integers", {
  keys <- c("chunk:37:5:0:45", "chunk:37:6:0:45")
  put_legacy_biomes_data(list(2L, 2L), keys, db = db)

  new_dat <- list(
    `chunk:37:5:0:45` = matrix(2L, 16, 16),
    `chunk:37:6:0:45` = matrix(2L, 16, 16)
  )

  dat <- get_legacy_biomes_data(keys, db = db, return_names = FALSE)
  expect_equal(dat, new_dat)
})

test_that("put_legacy_biomes_data() accepts strings", {
  keys <- c("chunk:37:5:0:45", "chunk:37:6:0:45")
  put_legacy_biomes_data(list("plains", "plains"), keys, db = db)

  new_dat <- list(
    `chunk:37:5:0:45` = matrix("plains", 16, 16),
    `chunk:37:6:0:45` = matrix("plains", 16, 16)
  )

  dat <- get_legacy_biomes_data(keys, db = db)
  expect_equal(dat, new_dat)
})

close(db)
unlink(dbpath, recursive = TRUE)