tests/testthat/test-experiments.R

In brickster: R Toolkit for 'Databricks'

test_that("experiments API - don't perform", {

  withr::local_envvar(c(
    "DATABRICKS_HOST" = "http://mock_host",
    "DATABRICKS_TOKEN" = "mock_token"
  ))

  resp_list <- db_experiments_list(max_results = 1, perform_request = FALSE)
  expect_s3_class(resp_list, "httr2_request")

  resp_get_by_name <- db_experiments_get(
    name = "some name",
    perform_request = FALSE
  )
  expect_s3_class(resp_get_by_name, "httr2_request")

  resp_get_by_id <- db_experiments_get(
    id = "some id",
    perform_request = FALSE
  )
  expect_s3_class(resp_get_by_id, "httr2_request")


})

skip_on_cran()
skip_unless_authenticated()
skip_unless_aws_workspace()

test_that("experiments API", {

  resp_list <- db_experiments_list(max_results = 1)
  expect_type(resp_list, "list")
  expect_named(resp_list[[1]])

  resp_get_by_name <- db_experiments_get(name = resp_list[[1]]$name)
  expect_equal(resp_list[[1]]$name, resp_get_by_name$name)

  resp_get_by_id <- db_experiments_get(id = resp_list[[1]]$experiment_id)
  expect_equal(resp_list[[1]]$experiment_id, resp_get_by_name$experiment_id)

  # don't allow both
  expect_error(db_experiments_get(
    id = resp_list[[1]]$experiment_id,
    name = resp_list[[1]]$name
  ))

})