Block SDK-managed secondary connector peers
This commit is contained in:
@@ -2341,74 +2341,13 @@ class ApiSmokeTests(unittest.TestCase):
|
||||
self.assertEqual(unchanged["file"]["id"], nextcloud_file["id"])
|
||||
self.assertEqual(unchanged["current_version_id"], synced["current_version_id"])
|
||||
|
||||
case = self
|
||||
|
||||
class _FakeSmbStat:
|
||||
st_size = 18
|
||||
st_mtime = 1783425600
|
||||
st_mtime_ns = 1783425600000000000
|
||||
|
||||
class _FakeSmbEntry:
|
||||
def __init__(self, name: str, is_dir: bool, size: int = 18) -> None:
|
||||
self.name = name
|
||||
self._is_dir = is_dir
|
||||
self._stat = _FakeSmbStat()
|
||||
self._stat.st_size = size
|
||||
|
||||
def is_dir(self) -> bool:
|
||||
return self._is_dir
|
||||
|
||||
def stat(self):
|
||||
return self._stat
|
||||
|
||||
class _FakeSmbScandir:
|
||||
def __enter__(self):
|
||||
return iter([
|
||||
_FakeSmbEntry("Archive", True, 0),
|
||||
_FakeSmbEntry("smb-notice.txt", False, 18),
|
||||
])
|
||||
|
||||
def __exit__(self, exc_type, exc, traceback):
|
||||
return False
|
||||
|
||||
class _FakeSmbFile:
|
||||
def __enter__(self):
|
||||
return self
|
||||
|
||||
def __exit__(self, exc_type, exc, traceback):
|
||||
return False
|
||||
|
||||
def read(self, size: int) -> bytes:
|
||||
case.assertGreater(size, 18)
|
||||
return b"smb connector data"
|
||||
|
||||
class _FakeSmbClient:
|
||||
def scandir(self, path: str, **kwargs):
|
||||
case.assertEqual(path, r"\\127.0.0.1\files\root\shared")
|
||||
case.assertEqual(kwargs["port"], 1445)
|
||||
case.assertEqual(kwargs["username"], "govoplan")
|
||||
case.assertEqual(kwargs["password"], "super-secret-smb")
|
||||
case.assertTrue(kwargs["require_signing"])
|
||||
return _FakeSmbScandir()
|
||||
|
||||
def stat(self, path: str, **kwargs):
|
||||
case.assertEqual(path, r"\\127.0.0.1\files\root\shared\smb-notice.txt")
|
||||
case.assertEqual(kwargs["port"], 1445)
|
||||
return _FakeSmbStat()
|
||||
|
||||
def open_file(self, path: str, mode: str, **kwargs):
|
||||
case.assertEqual(path, r"\\127.0.0.1\files\root\shared\smb-notice.txt")
|
||||
case.assertEqual(mode, "rb")
|
||||
case.assertEqual(kwargs["password"], "super-secret-smb")
|
||||
return _FakeSmbFile()
|
||||
|
||||
fake_smb = _FakeSmbClient()
|
||||
with patch("govoplan_files.backend.storage.connector_browse._smbclient_module", return_value=fake_smb):
|
||||
with patch("govoplan_files.backend.storage.connector_browse._smbclient_module") as smb_sdk:
|
||||
smb_browse = self.client.get("/api/v1/files/connectors/profiles/tenant-smb/browse?path=shared", headers=headers)
|
||||
self.assertEqual(smb_browse.status_code, 200, smb_browse.text)
|
||||
self.assertEqual([(item["kind"], item["path"]) for item in smb_browse.json()["items"]], [("folder", "shared/Archive"), ("file", "shared/smb-notice.txt")])
|
||||
self.assertEqual(smb_browse.status_code, 400, smb_browse.text)
|
||||
self.assertIn("redirects/referrals", smb_browse.json()["detail"])
|
||||
smb_sdk.assert_not_called()
|
||||
|
||||
with patch("govoplan_files.backend.storage.connector_imports._smbclient_module", return_value=fake_smb):
|
||||
with patch("govoplan_files.backend.storage.connector_imports._smbclient_module") as smb_import_sdk:
|
||||
smb_import = self.client.post(
|
||||
"/api/v1/files/connectors/profiles/tenant-smb/import",
|
||||
headers=headers,
|
||||
@@ -2420,12 +2359,9 @@ class ApiSmokeTests(unittest.TestCase):
|
||||
"target_folder": "imports",
|
||||
},
|
||||
)
|
||||
self.assertEqual(smb_import.status_code, 200, smb_import.text)
|
||||
smb_file = smb_import.json()["files"][0]
|
||||
self.assertEqual(smb_file["display_path"], "imports/smb-notice.txt")
|
||||
self.assertEqual(smb_file["source_provenance"]["provider"], "smb")
|
||||
self.assertEqual(smb_file["source_provenance"]["external_id"], "files:shared/smb-notice.txt")
|
||||
self.assertEqual(smb_file["source_provenance"]["metadata"]["share"], "files")
|
||||
self.assertEqual(smb_import.status_code, 400, smb_import.text)
|
||||
self.assertIn("redirects/referrals", smb_import.json()["detail"])
|
||||
smb_import_sdk.assert_not_called()
|
||||
|
||||
filtered = self.client.get("/api/v1/files/connectors/profiles?provider=smb", headers=headers)
|
||||
self.assertEqual(filtered.status_code, 200, filtered.text)
|
||||
|
||||
Reference in New Issue
Block a user