fix tests

iterative · Jul 12, 2024 · 4ca848f · 4ca848f
1 parent 7ec3fb3
commit 4ca848f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -332,6 +332,7 @@ class CloudTestCatalog:
     server: CloudServer
     working_dir: PosixPath
     catalog: Catalog
+    session: Session
 
     @property
     def src(self):
@@ -432,7 +433,10 @@ def get_cloud_test_catalog(cloud_server, tmp_path, id_generator, metastore, ware
         tmp_dir=str(tmpfile_dir),
         client_config=cloud_server.client_config,
     )
-    return CloudTestCatalog(server=cloud_server, working_dir=tmp_path, catalog=catalog)
+    session = Session.get(catalog=catalog)
+    return CloudTestCatalog(
+        server=cloud_server, working_dir=tmp_path, catalog=catalog, session=session
+    )
 
 
 @pytest.fixture

diff --git a/tests/func/test_datachain.py b/tests/func/test_datachain.py
@@ -16,7 +16,7 @@ def test_catalog_anon(catalog, anon):
 
 def test_from_storage(cloud_test_catalog):
     ctc = cloud_test_catalog
-    dc = DataChain.from_storage(ctc.src_uri, catalog=ctc.catalog)
+    dc = DataChain.from_storage(ctc.src_uri, session=ctc.session)
     assert dc.count() == 7
 
 
@@ -27,7 +27,7 @@ def new_signal(file: File) -> str:
         with file.open() as f:
             return file.name + " -> " + f.read().decode("utf-8")
 
-    dc = DataChain.from_storage(ctc.src_uri, catalog=ctc.catalog).map(signal=new_signal)
+    dc = DataChain.from_storage(ctc.src_uri, session=ctc.session).map(signal=new_signal)
     expected = {
         "description -> Cats and Dogs",
         "cat1 -> meow",