test_that("ConnectorDatabricksVolume creation fails", {
  withr::local_options(
    list(connector.verbosity_level = "quiet")
  )
  # Invalid full path
  expect_error(ConnectorDatabricksVolume$new(full_path = 1))

  # Invalid catalog
  expect_error(ConnectorDatabricksVolume$new(catalog = 1))

  # Invalid schema
  expect_error(
    ConnectorDatabricksVolume$new(
      catalog = "DATABRICKS_CATALOG_NAME",
      schema = 1
    )
  )

  # Invalid path
  expect_error(
    ConnectorDatabricksVolume$new(
      catalog = "DATABRICKS_CATALOG_NAME",
      schema = "DATABRICKS_SCHEMA_NAME",
      path = 1
    )
  )

  # Invalid force
  expect_error(
    ConnectorDatabricksVolume$new(
      catalog = "DATABRICKS_CATALOG_NAME",
      schema = "DATABRICKS_SCHEMA_NAME",
      path = "path",
      force = 1
    )
  )

  # Invalid extra class
  expect_error(
    ConnectorDatabricksVolume$new(
      catalog = "DATABRICKS_CATALOG_NAME",
      schema = "DATABRICKS_SCHEMA_NAME",
      path = "path",
      extra_class = 1
    )
  )
})

test_that("connector_databricks_volume creation fails", {
  # Invalid full path
  expect_error(connector_databricks_volume(full_path = 1))

  # Invalid catalog
  expect_error(connector_databricks_volume(catalog = 1))

  # Invalid schema
  expect_error(
    connector_databricks_volume(catalog = "DATABRICKS_CATALOG_NAME", schema = 1)
  )

  # Invalid path
  expect_error(
    connector_databricks_volume(
      catalog = "DATABRICKS_CATALOG_NAME",
      schema = "DATABRICKS_SCHEMA_NAME",
      path = 1
    )
  )

  # Invalid force
  expect_error(
    connector_databricks_volume(
      catalog = "DATABRICKS_CATALOG_NAME",
      schema = "DATABRICKS_SCHEMA_NAME",
      path = "path",
      force = 1
    )
  )

  # Invalid extra class
  expect_error(
    connector_databricks_volume(
      catalog = "DATABRICKS_CATALOG_NAME",
      schema = "DATABRICKS_SCHEMA_NAME",
      path = "path",
      extra_class = 1
    )
  )
})

skip_offline_test()

test_that("ConnectorDatabricksVolume creation works", {
  ### Using already existing volume
  # Create connectors using two different ways with valid full path
  cons <- list(
    ConnectorDatabricksVolume$new(
      full_path = setup_db_volume_path,
      force = TRUE
    ),
    connector_databricks_volume(full_path = setup_db_volume_path)
  )

  for (con in cons) {
    checkmate::expect_r6(
      x = con,
      classes = c("Connector", "ConnectorFS", "ConnectorDatabricksVolume"),
      public = c(
        "list_content_cnt",
        "create_directory_cnt",
        "read_cnt",
        "write_cnt",
        "remove_cnt",
        "upload_cnt",
        "download_cnt",
        "remove_directory_cnt"
      ),
      private = c(".path", ".full_path", ".catalog", ".schema")
    )
  }

  # Create connector using individual parameters
  con2 <- ConnectorDatabricksVolume$new(
    catalog = setup_db_catalog,
    schema = setup_db_schema,
    path = "local_test_volume",
    force = TRUE
  )
  checkmate::expect_r6(
    x = con2,
    classes = c("Connector", "ConnectorFS", "ConnectorDatabricksVolume"),
    public = c(
      "list_content_cnt",
      "create_directory_cnt",
      "read_cnt",
      "write_cnt",
      "remove_cnt",
      "upload_cnt",
      "download_cnt",
      "remove_directory_cnt"
    ),
    private = c(".path", ".full_path", ".catalog", ".schema")
  )

  # Create connector using individual parameters and extra class
  con3 <- ConnectorDatabricksVolume$new(
    catalog = setup_db_catalog,
    schema = setup_db_schema,
    path = "local_test_volume",
    extra_class = "extra_class",
    force = TRUE
  )
  checkmate::expect_r6(
    x = con3,
    classes = c(
      "Connector",
      "ConnectorFS",
      "ConnectorDatabricksVolume",
      "extra_class"
    ),
    public = c(
      "list_content_cnt",
      "create_directory_cnt",
      "read_cnt",
      "write_cnt",
      "remove_cnt",
      "upload_cnt",
      "download_cnt",
      "remove_directory_cnt"
    ),
    private = c(".path", ".full_path", ".catalog", ".schema")
  )

  # Create connector using individual parameters with force parameter
  con4 <- ConnectorDatabricksVolume$new(
    catalog = setup_db_catalog,
    schema = setup_db_schema,
    path = "local_test_volume/force",
    force = TRUE
  )
  checkmate::expect_r6(
    x = con4,
    classes = c("Connector", "ConnectorFS", "ConnectorDatabricksVolume"),
    public = c(
      "list_content_cnt",
      "create_directory_cnt",
      "read_cnt",
      "write_cnt",
      "remove_cnt",
      "upload_cnt",
      "download_cnt",
      "remove_directory_cnt"
    ),
    private = c(".path", ".full_path", ".catalog", ".schema")
  )
})

test_that("ConnectorDatabricksVolume methods work", {
  ### Testing scenario

  ## 1. Test directory method works
  # Create connector using valid full path
  con <- connector_databricks_volume(full_path = setup_db_volume_path)

  # Create a test directory
  test_directory <- "test_directory"
  new_directory <- con$create_directory_cnt(name = test_directory)
  checkmate::assert_r6(new_directory, "ConnectorDatabricksVolume")
  expect_true(basename(new_directory$full_path) == "test_directory")

  # List contents of root folder
  list_of_items <- con$list_content_cnt()

  # Check if directory exists
  expect_contains(list_of_items, basename(test_directory))

  # Delete test directory
  con$remove_directory_cnt(test_directory)

  # List contents of root folder
  list_of_items <- con$list_content_cnt()

  # Check if directory exists
  expect_false(basename(test_directory) %in% list_of_items)

  ## 2. Test file method works
  # Write a test file
  con$write_cnt(x = tibble::tibble(x = 1, y = 2), name = "test_file.csv")

  # List contents of root folder
  list_of_items <- con$list_content_cnt()

  # Check if file exists
  expect_contains(list_of_items, basename("test_file.csv"))

  # Download the file
  con$download_cnt("test_file.csv", "test_file.csv")

  # Check if file exists
  expect_true(file.exists("test_file.csv"))

  # Read the file
  file_content <- con$read_cnt("test_file.csv")

  # Check if file content is correct
  expect_equal(file_content, tibble::tibble(x = 1, y = 2))

  # Delete test file
  con$remove_cnt("test_file.csv")

  # List contents of root folder
  list_of_items <- con$list_content_cnt()

  # Check if file exists
  expect_false(basename("test_file.csv") %in% list_of_items)

  # Write a test file
  write.table(x = tibble::tibble(x = 1, y = 2), file = "test_file.csv")

  # Create a test file
  con$upload_cnt("test_file.csv", "test_file.csv")

  # List contents of root folder
  list_of_items <- con$list_content_cnt()

  # Check if file exists
  expect_contains(list_of_items, basename("test_file.csv"))

  # Delete test file
  con$remove_cnt("test_file.csv")

  # Delete local test file
  unlink("test_file.csv")
})

test_that("ConnectorDatabricksVolume upload/download works", {
  create_nested_directories("nested_structure_volumes", 3, 2)
  withr::defer(
    fs::dir_delete("nested_structure_volumes")
  )

  expect_no_failure(setup_volume_connector$upload_directory_cnt(
    src = "nested_structure_volumes",
    overwrite = TRUE
  ))

  expect_no_failure(setup_volume_connector$remove_directory_cnt(
    name = "nested_structure_volumes"
  ))

  expect_no_failure(
    setup_volume_connector |>
      upload_directory_cnt(
        src = "nested_structure_volumes"
      )
  )

  withr::defer({
    setup_volume_connector |>
      remove_directory_cnt(
        name = "nested_structure_volumes"
      )
    expect_error(brickster::db_volume_dir_exists(paste0(
      "/Volumes/",
      setup_db_volume_path,
      "/nested_structure_volumes"
    )))
  })

  expect_no_failure(setup_volume_connector$download_directory_cnt(
    src = "nested_structure_volumes",
    dest = "nested_structure_volumes_downloaded"
  ))

  expect_no_failure(
    setup_volume_connector |>
      download_directory_cnt(
        src = "nested_structure_volumes",
        dest = "nested_structure_volumes_downloaded_pipe"
      )
  )

  withr::defer({
    fs::dir_delete("nested_structure_volumes_downloaded")
    fs::dir_delete("nested_structure_volumes_downloaded_pipe")
  })
})