test_that("useful message if no tokens", {
  local_tokens()

  expect_snapshot(token_usage())
})

test_that("can retrieve and log tokens", {
  local_tokens()
  provider <- test_provider("testprovider", "test")

  log_tokens(provider, tokens(input = 1), dollars(0))
  expect_equal(the$tokens, tokens_row("testprovider", "test", 1, 0, 0, 0))

  log_tokens(provider, tokens(output = 1), dollars(0))
  expect_equal(the$tokens, tokens_row("testprovider", "test", 1, 1, 0, 0))

  log_tokens(provider, tokens(cached_input = 1), dollars(0))
  expect_equal(the$tokens, tokens_row("testprovider", "test", 1, 1, 1, 0))

  log_tokens(provider, tokens(), dollars(0))
  expect_equal(the$tokens, tokens_row("testprovider", "test", 1, 1, 1, 0))

  expect_snapshot(token_usage())

  log_tokens(provider, tokens(), dollars(NA_real_))
  expect_equal(
    the$tokens,
    tokens_row("testprovider", "test", 1, 1, 1, NA_real_)
  )
})

test_that("can compute price of tokens", {
  provider <- test_provider("OpenAI", "gpt-4o")

  expect_equal(get_token_cost(provider, tokens(input = 1e6)), dollars(2.5))
  expect_equal(get_token_cost(provider, tokens(output = 1e6)), dollars(10))
  expect_equal(
    get_token_cost(provider, tokens(cached_input = 1e6)),
    dollars(1.25)
  )
})

test_that("can compute price of tokens with a variant", {
  provider <- test_provider("OpenAI", "gpt-4o")

  expect_equal(
    get_token_cost(provider, tokens(input = 1e6), variant = "priority"),
    dollars(4.25)
  )

  # fals back to baseline if no match
  expect_equal(
    get_token_cost(provider, tokens(input = 1e6), variant = "tuesday-pm"),
    get_token_cost(provider, tokens(input = 1e6))
  )
})

test_that("price is NA if we don't have the data for it", {
  provider <- test_provider("ClosedAI", "gpt-4o")
  expect_equal(
    get_token_cost(provider, tokens(1, 1, 1)),
    dollars(NA_real_)
  )
})

test_that("token_usage() shows price if available", {
  local_tokens()
  provider <- test_provider("OpenAI", "gpt-4o")

  toks <- tokens(input = 1.5e6, output = 2e5, cached_input = 0)
  cost <- get_token_cost(provider, toks)
  log_tokens(provider, toks, cost)
  expect_snapshot(token_usage())
})

# ellmer_dollars --------------------------------------------------------------

test_that("price is formatted nicely", {
  expect_equal(format(dollars(NA)), "NA")
  expect_equal(format(dollars(0.0001)), "$0.00")
  expect_equal(format(dollars(c(10, 1))), c("$10.00", "$ 1.00"))
})

test_that("dollars looks good, including in data.frames", {
  price <- dollars(1.234567)
  expect_snapshot({
    price
    data.frame(price)
  })
})

test_that("dollars class survives basic transforms", {
  expect_equal(sum(dollars(c(1, 2))), dollars(3))

  d <- dollars(c(1, 2, 3))
  expect_equal(d[1:2], dollars(c(1, 2)))
  expect_equal(d[[1]], dollars(1))
})

# Helpers ---------------------------------------------------------------------

test_that("log_turns ignores non-assistant turns", {
  local_tokens()
  provider <- test_provider("testprovider", "test")

  turn1 <- UserTurn(contents = "text")
  turn2 <- AssistantTurn(
    contents = "Hello",
    tokens = c(8, 3, 2),
    cost = dollars(1)
  )

  log_turns(provider, list(turn1, turn2, NULL))
  expect_equal(the$tokens, tokens_row("testprovider", "test", 8, 3, 2, 1))
})

test_that("log_turns aggregates multiple turns", {
  local_tokens()
  provider <- test_provider("testprovider", "test")

  turn1 <- AssistantTurn(contents = "Hello", tokens = c(8, 3, 2))
  turn2 <- AssistantTurn(contents = "World", tokens = c(1, 1, 1))
  log_turns(provider, list(turn1, turn2))
  expect_equal(
    the$tokens,
    tokens_row("testprovider", "test", 9, 4, 3, NA_real_)
  )
})