diff --git a/ingestify/tests/conftest.py b/ingestify/tests/conftest.py index 2cee097..33faeae 100644 --- a/ingestify/tests/conftest.py +++ b/ingestify/tests/conftest.py @@ -10,3 +10,8 @@ def datastore_dir(): os.environ["TEST_DIR"] = tmpdirname os.environ["INGESTIFY_RUN_EAGER"] = "true" yield tmpdirname + + +@pytest.fixture(scope="session") +def config_file(): + return os.path.abspath(os.path.dirname(__file__) + "/config.yaml") diff --git a/ingestify/tests/test_engine.py b/ingestify/tests/test_engine.py index b7cf7bc..6c3855b 100644 --- a/ingestify/tests/test_engine.py +++ b/ingestify/tests/test_engine.py @@ -134,8 +134,8 @@ def fetch_dataset_files( } -def test_engine(): - engine = get_engine("config.yaml", "main") +def test_engine(config_file): + engine = get_engine(config_file, "main") add_extract_job( engine, SimpleFakeSource("fake-source"), competition_id=1, season_id=2 @@ -170,12 +170,12 @@ def test_engine(): assert len(datasets) == 1 -def test_iterator_source(): +def test_iterator_source(config_file): """Test when a Source returns a Iterator to do Batch processing. Every batch must be executed right away. """ - engine = get_engine("config.yaml", "main") + engine = get_engine(config_file, "main") batch_source = None