Fix Tests

data-dot-all · Jun 28, 2024 · 4a25243 · 4a25243
1 parent bb111c4
commit 4a25243
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/backend/dataall/modules/catalog/tasks/catalog_indexer_task.py b/backend/dataall/modules/catalog/tasks/catalog_indexer_task.py
@@ -22,7 +22,7 @@ class CatalogIndexerTask:
     """
 
     @classmethod
-    def index_objects(cls, engine, with_deletes):
+    def index_objects(cls, engine, with_deletes='False'):
         try:
             indexed_object_uris = []
             with engine.scoped_session() as session:

diff --git a/tests/modules/s3_datasets/tasks/test_dataset_catalog_indexer.py b/tests/modules/s3_datasets/tasks/test_dataset_catalog_indexer.py
@@ -1,6 +1,6 @@
 import pytest
 
-from dataall.modules.catalog.tasks.catalog_indexer_task import index_objects
+from dataall.modules.catalog.tasks.catalog_indexer_task import CatalogIndexerTask
 from dataall.modules.s3_datasets.db.dataset_models import DatasetTable, S3Dataset
 
 
@@ -54,6 +54,6 @@ def test_catalog_indexer(db, org, env, sync_dataset, table, mocker):
     mocker.patch(
         'dataall.modules.s3_datasets.indexers.dataset_indexer.DatasetIndexer.upsert', return_value=sync_dataset
     )
-    indexed_objects_counter = index_objects(engine=db)
+    indexed_objects_counter = CatalogIndexerTask.index_objects(engine=db)
     # Count should be One table + One Dataset = 2
     assert indexed_objects_counter == 2