tests/testthat/test-unity-catalog.R

In zacdav-db/brickster: R Toolkit for 'Databricks'

test_that("Unity Catalog API - don't perform", {

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

  resp_summary <- db_uc_metastore_summary(perform_request = F)
  expect_s3_class(resp_summary, "httr2_request")

  resp_sc_list <- db_uc_storage_creds_list(perform_request = F)
  expect_s3_class(resp_sc_list, "httr2_request")

  resp_sc_get <- db_uc_storage_creds_get(
    name = "some_name",
    perform_request = F
  )
  expect_s3_class(resp_sc_get, "httr2_request")

  resp_el_list <- db_uc_external_loc_list(
    perform_request = F
  )
  expect_s3_class(resp_el_list, "httr2_request")

  resp_el_get <- db_uc_external_loc_get(
    name = "some_name",
    perform_request = F
  )
  expect_s3_class(resp_el_get, "httr2_request")

  resp_catalog_list <- db_uc_catalogs_list(perform_request = F)
  expect_s3_class(resp_catalog_list, "httr2_request")

  resp_catalog_get <- db_uc_catalogs_get(
    catalog = "some_catalog",
    perform_request = F
  )
  expect_s3_class(resp_catalog_get, "httr2_request")

  resp_schema_list <- db_uc_schemas_list(
    catalog = "some_catalog",
    perform_request = F
  )
  expect_s3_class(resp_schema_list, "httr2_request")

  resp_schema_get <- db_uc_schemas_get(
    catalog = "some_catalog",
    schema = "some_schema",
    perform_request = F
  )
  expect_s3_class(resp_schema_get, "httr2_request")

  resp_models_get <- db_uc_models_get(
    catalog = "some_catalog",
    schema = "some_schema",
    model = "some_model",
    perform_request = F
  )
  expect_s3_class(resp_models_get, "httr2_request")

  resp_models_list <- db_uc_models_list(
    catalog = "some_catalog",
    schema = "some_schema",
    perform_request = F
  )
  expect_s3_class(resp_models_list, "httr2_request")

  resp_funcs_get <- db_uc_funcs_get(
    catalog = "some_catalog",
    schema = "some_schema",
    func = "some_func",
    perform_request = F
  )
  expect_s3_class(resp_funcs_get, "httr2_request")

  resp_funcs_list <- db_uc_funcs_list(
    catalog = "some_catalog",
    schema = "some_schema",
    perform_request = F
  )
  expect_s3_class(resp_funcs_list, "httr2_request")

})

skip_on_cran()
skip_unless_authenticated()
skip_unless_aws_workspace()

test_that("Unity Catalog API", {

  expect_no_error({
    resp_summary <- db_uc_metastore_summary()
  })
  expect_type(resp_summary, "list")

  expect_no_error({
    resp_sc_list <- db_uc_storage_creds_list()
  })
  expect_type(resp_sc_list, "list")

  expect_no_error({
    resp_sc_get <- db_uc_storage_creds_get(
      name = resp_sc_list[[1]]$name
    )
  })
  expect_type(resp_sc_get, "list")

  expect_no_error({
    resp_el_list <- db_uc_external_loc_list()
  })
  expect_type(resp_el_list, "list")

  expect_no_error({
    resp_el_get <- db_uc_external_loc_get(
      name = resp_el_list[[1]][[1]]$name
    )
  })
  expect_type(resp_el_get, "list")

  expect_no_error({
    resp_catalog_list <- db_uc_catalogs_list()
  })
  expect_type(resp_catalog_list, "list")

  expect_no_error({
    resp_catalog_get <- db_uc_catalogs_get(
      catalog = resp_catalog_list[[1]]$name
    )
  })
  expect_type(resp_catalog_get, "list")

  expect_no_error({
    resp_schema_list <- db_uc_schemas_list(
      catalog = "main"
    )
  })
  expect_type(resp_schema_list, "list")

  expect_no_error({
    resp_schema_get <- db_uc_schemas_get(
      catalog = "main",
      schema = resp_schema_list[[1]]$name
    )
  })
  expect_type(resp_schema_get, "list")

})