From d775e4612e7eef9746a64d5e7fec9fcb03ad1d3b Mon Sep 17 00:00:00 2001 From: erinhmclark Date: Tue, 25 Feb 2025 21:38:41 +0000 Subject: [PATCH] Update gsheet_feeder references in tests. --- tests/feeders/test_gworksheet.py | 2 +- tests/test_orchestrator.py | 12 ++++++------ 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/tests/feeders/test_gworksheet.py b/tests/feeders/test_gworksheet.py index 2b05504..b6a0b5c 100644 --- a/tests/feeders/test_gworksheet.py +++ b/tests/feeders/test_gworksheet.py @@ -1,7 +1,7 @@ # Note this isn't a feeder, but contained as utility of the gsheet feeder module import pytest -from auto_archiver.modules.gsheet_feeder import GWorksheet +from auto_archiver.modules.gsheet_feeder_db import GWorksheet class TestGWorksheet: diff --git a/tests/test_orchestrator.py b/tests/test_orchestrator.py index 752adb8..72f4949 100644 --- a/tests/test_orchestrator.py +++ b/tests/test_orchestrator.py @@ -78,7 +78,7 @@ def test_help(orchestrator, basic_parser, capsys): assert "--logging.level" in logs # individual module configs - assert "--gsheet_feeder.sheet_id" in logs + assert "--gsheet_feeder_db.sheet_id" in logs def test_add_custom_modules_path(orchestrator, test_args): @@ -154,22 +154,22 @@ def test_load_modules_from_commandline(orchestrator, test_args): assert orchestrator.formatters[0].name == "example_module" def test_load_settings_for_module_from_commandline(orchestrator, test_args): - args = test_args + ["--feeders", "gsheet_feeder", "--gsheet_feeder.sheet_id", "123", "--gsheet_feeder.service_account", "tests/data/test_service_account.json"] + args = test_args + ["--feeders", "gsheet_feeder_db", "--gsheet_feeder_db.sheet_id", "123", "--gsheet_feeder_db.service_account", "tests/data/test_service_account.json"] orchestrator.setup(args) assert len(orchestrator.feeders) == 1 - assert orchestrator.feeders[0].name == "gsheet_feeder" - assert orchestrator.config['gsheet_feeder']['sheet_id'] == "123" + assert orchestrator.feeders[0].name == "gsheet_feeder_db" + assert orchestrator.config['gsheet_feeder_db']['sheet_id'] == "123" def test_multiple_orchestrator(test_args): - o1_args = test_args + ["--feeders", "gsheet_feeder", "--gsheet_feeder.service_account", "tests/data/test_service_account.json"] + o1_args = test_args + ["--feeders", "gsheet_feeder_db", "--gsheet_feeder_db.service_account", "tests/data/test_service_account.json"] o1 = ArchivingOrchestrator() with pytest.raises(ValueError) as exit_error: - # this should fail because the gsheet_feeder requires a sheet_id / sheet + # this should fail because the gsheet_feeder_db requires a sheet_id / sheet o1.setup(o1_args)