tests/testthat/test-helper-class.R

In chores: A Collection of Large Language Model Assistants

test_that("can find the previous helper", {
  skip_if(identical(Sys.getenv("ANTHROPIC_API_KEY"), ""))
  skip_if_not_installed("withr")
  withr::local_options(
    .chores_chat = ellmer::chat_claude(model = "claude-3-7-sonnet-20250219")
  )

  cli_helper <- .init_helper("cli")
  expect_no_error(response <- cli_helper$chat("stop(\"Error message here\")"))
})

test_that("chores_chat effectively integrates system prompt", {
  skip_if(identical(Sys.getenv("ANTHROPIC_API_KEY"), ""))
  skip_if_not_installed("withr")
  withr::local_options(
    .chores_chat = ellmer::chat_claude(model = "claude-3-7-sonnet-20250219")
  )

  cli_helper <- .init_helper("cli")
  response <- cli_helper$chat("stop(\"Error message here\")")
  expect_true(grepl("cli_abort", response))
  expect_true(grepl("Error message here", response))

  testthat_helper <- .init_helper("testthat")
  response <- testthat_helper$chat("expect_error(beep_bop_boop())")
  expect_true(grepl("expect_snapshot", response))
  expect_true(grepl("beep_bop_boop", response))
})

test_that("fetch_chores_chat returns early with no option set", {
  skip_if(identical(Sys.getenv("ANTHROPIC_API_KEY"), ""))
  withr::local_options(.chores_chat = NULL)
  skip_if_not_installed("withr")

  expect_snapshot(.res <- fetch_chores_chat())
  expect_null(.res)
})

test_that("fetch_chores_chat returns early with bad config", {
  withr::local_options(.chores_chat = "chat")
  expect_snapshot(.res <- fetch_chores_chat())
  expect_null(.res)

  mock_chat <- structure(list(), class = "Chat")
  expect_identical(fetch_chores_chat(mock_chat), mock_chat)
})

test_that("get_chores_chat prefers chores.chat over .chores_chat", {
  mock_chat_new <- structure(list(name = "new"), class = "Chat")
  mock_chat_old <- structure(list(name = "old"), class = "Chat")

  withr::local_options(chores.chat = NULL, .chores_chat = NULL)
  expect_null(get_chores_chat())

  withr::local_options(chores.chat = NULL, .chores_chat = mock_chat_old)
  expect_identical(get_chores_chat(), mock_chat_old)

  withr::local_options(chores.chat = mock_chat_new, .chores_chat = NULL)
  expect_identical(get_chores_chat(), mock_chat_new)

  withr::local_options(
    chores.chat = mock_chat_new,
    .chores_chat = mock_chat_old
  )
  expect_identical(get_chores_chat(), mock_chat_new)
})