chore: import upstream snapshot with attribution

2026-07-13 12:30:44 +08:00
commit bcbd1bdb22
5748 changed files with 562488 additions and 0 deletions
@@ -0,0 +1,99 @@
+from mirage.accessor import databricks_volume as accessor_module
+from mirage.accessor.databricks_volume import DatabricksVolumeAccessor
+from mirage.resource.databricks_volume import DatabricksVolumeConfig
+
+
+class FakeWorkspaceClient:
+    calls: list[dict] = []
+
+    def __init__(self, **kwargs) -> None:
+        self.calls.append(kwargs)
+        self.files = object()
+
+
+class FakeWorkspaceConfig:
+
+    def __init__(self, **kwargs) -> None:
+        for key, value in kwargs.items():
+            setattr(self, key, value)
+
+
+def test_accessor_passes_timeout_to_workspace_client(monkeypatch):
+    FakeWorkspaceClient.calls = []
+    monkeypatch.setattr(
+        accessor_module,
+        "WorkspaceClient",
+        FakeWorkspaceClient,
+    )
+    monkeypatch.setattr(
+        accessor_module,
+        "WorkspaceConfig",
+        FakeWorkspaceConfig,
+    )
+    config = DatabricksVolumeConfig(
+        catalog="main",
+        schema="default",
+        volume="agent_files",
+        host="https://example.cloud.databricks.com",
+        token="secret",
+        timeout=17,
+    )
+    accessor = DatabricksVolumeAccessor(config)
+    assert accessor.files is not None
+    sdk_config = FakeWorkspaceClient.calls[0]["config"]
+    assert sdk_config.host == "https://example.cloud.databricks.com"
+    assert sdk_config.token == "secret"
+    assert sdk_config.http_timeout_seconds == 17
+
+
+def test_accessor_sets_pat_auth_type_for_explicit_token(monkeypatch):
+    FakeWorkspaceClient.calls = []
+    monkeypatch.setattr(
+        accessor_module,
+        "WorkspaceClient",
+        FakeWorkspaceClient,
+    )
+    monkeypatch.setattr(
+        accessor_module,
+        "WorkspaceConfig",
+        FakeWorkspaceConfig,
+    )
+    config = DatabricksVolumeConfig(
+        catalog="main",
+        schema="default",
+        volume="agent_files",
+        host="https://example.cloud.databricks.com",
+        token="request-token",
+    )
+    accessor = DatabricksVolumeAccessor(config)
+
+    assert accessor.files is not None
+    sdk_config = FakeWorkspaceClient.calls[0]["config"]
+    assert sdk_config.auth_type == "pat"
+    assert sdk_config.token == "request-token"
+
+
+def test_accessor_leaves_auth_type_unset_for_profile(monkeypatch):
+    FakeWorkspaceClient.calls = []
+    monkeypatch.setattr(
+        accessor_module,
+        "WorkspaceClient",
+        FakeWorkspaceClient,
+    )
+    monkeypatch.setattr(
+        accessor_module,
+        "WorkspaceConfig",
+        FakeWorkspaceConfig,
+    )
+    config = DatabricksVolumeConfig(
+        catalog="main",
+        schema="default",
+        volume="agent_files",
+        profile="default",
+    )
+    accessor = DatabricksVolumeAccessor(config)
+
+    assert accessor.files is not None
+    sdk_config = FakeWorkspaceClient.calls[0]["config"]
+    assert not hasattr(sdk_config, "auth_type")
+    assert sdk_config.profile == "default"