tests/testthat/test-auth.R

In virustotal: R Client for the 'VirusTotal' API

test_that("set_key validates input correctly", {
  expect_error(set_key(), class = "virustotal_validation_error")

  expect_error(set_key(NULL), class = "virustotal_validation_error")
  expect_error(set_key(123), class = "virustotal_validation_error")
  expect_error(set_key(character(0)), class = "virustotal_validation_error")

  expect_error(set_key("short"), class = "virustotal_validation_error")

  expect_error(
    set_key("your_api_key_here"),
    class = "virustotal_validation_error"
  )
  expect_error(
    set_key("api_key_here"),
    class = "virustotal_validation_error"
  )
})

test_that("set_key handles whitespace correctly", {
  old_key <- Sys.getenv("VirustotalToken")

  expect_warning(
    set_key("  valid_32_character_api_key_1234567  "),
    "Removed leading/trailing whitespace"
  )
  expect_equal(
    Sys.getenv("VirustotalToken"),
    "valid_32_character_api_key_1234567"
  )

  if (old_key != "") {
    Sys.setenv(VirustotalToken = old_key)
  } else {
    Sys.unsetenv("VirustotalToken")
  }
})

test_that("set_key sets environment variable correctly", {
  old_key <- Sys.getenv("VirustotalToken")

  expect_message(
    result <- set_key("valid_32_character_api_key_1234567890"),
    "VirusTotal API key successfully set"
  )
  expect_true(result)
  expect_equal(
    Sys.getenv("VirustotalToken"),
    "valid_32_character_api_key_1234567890"
  )

  if (old_key != "") {
    Sys.setenv(VirustotalToken = old_key)
  } else {
    Sys.unsetenv("VirustotalToken")
  }
})

test_that("API functions require API key", {
  old_key <- Sys.getenv("VirustotalToken")
  Sys.unsetenv("VirustotalToken")

  expect_error(file_report("dummy_hash"), class = "virustotal_auth_error")
  expect_error(ip_report("8.8.8.8"), class = "virustotal_auth_error")
  expect_error(domain_report("example.com"), class = "virustotal_auth_error")

  if (old_key != "") {
    Sys.setenv(VirustotalToken = old_key)
  }
})