Skip to content

Commit 1366776

Browse files
test: update test_update1.py fixture for unified stores
- Update mock_stores_update fixture to use config.stores - Remove project_name (now embedded in location path) - Simplify fixture using unified stores API
1 parent 662471b commit 1366776

File tree

1 file changed

+11
-12
lines changed

1 file changed

+11
-12
lines changed

tests/integration/test_update1.py

Lines changed: 11 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -23,30 +23,29 @@ class Thing(dj.Manual):
2323

2424
@pytest.fixture(scope="module")
2525
def mock_stores_update(tmpdir_factory):
26-
"""Configure object storage stores for update tests."""
27-
og_project_name = dj.config.object_storage.project_name
28-
og_stores = dict(dj.config.object_storage.stores)
26+
"""Configure stores for update tests using unified stores system."""
27+
og_stores = dict(dj.config.stores)
2928

30-
# Configure stores
31-
dj.config.object_storage.project_name = "djtest"
32-
store_location = str(tmpdir_factory.mktemp("store"))
29+
# Configure stores (location includes project context)
30+
store_location = str(tmpdir_factory.mktemp("store")) + "/djtest"
3331
repo_stage = str(tmpdir_factory.mktemp("repo_stage"))
34-
repo_location = str(tmpdir_factory.mktemp("repo_loc"))
35-
dj.config.object_storage.stores["update_store"] = dict(
32+
repo_location = str(tmpdir_factory.mktemp("repo_loc")) + "/djtest"
33+
34+
dj.config.stores["update_store"] = dict(
3635
protocol="file",
3736
location=store_location,
3837
)
39-
dj.config.object_storage.stores["update_repo"] = dict(
38+
dj.config.stores["update_repo"] = dict(
4039
stage=repo_stage,
4140
protocol="file",
4241
location=repo_location,
4342
)
43+
4444
yield {"update_store": {"location": store_location}, "update_repo": {"stage": repo_stage, "location": repo_location}}
4545

4646
# Restore original
47-
dj.config.object_storage.project_name = og_project_name
48-
dj.config.object_storage.stores.clear()
49-
dj.config.object_storage.stores.update(og_stores)
47+
dj.config.stores.clear()
48+
dj.config.stores.update(og_stores)
5049

5150

5251
@pytest.fixture

0 commit comments

Comments
 (0)