bellingcat/auto-archiver
1
0
Fork 0
mirror of https://github.com/bellingcat/auto-archiver.git synced 2026-06-12 21:28:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update gsheet_feeder references in tests.

Browse Source
This commit is contained in:
erinhmclark
2025-02-25 21:38:41 +00:00
parent 077b56c150
commit d775e4612e
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/feeders/test_gworksheet.py
View File

@@ -1,7 +1,7 @@
# Note this isn't a feeder, but contained as utility of the gsheet feeder module # Note this isn't a feeder, but contained as utility of the gsheet feeder module
import pytest import pytest
from auto_archiver.modules.gsheet_feeder import GWorksheet from auto_archiver.modules.gsheet_feeder_db import GWorksheet
class TestGWorksheet: class TestGWorksheet:

12
tests/test_orchestrator.py
View File

@@ -78,7 +78,7 @@ def test_help(orchestrator, basic_parser, capsys):
assert "--logging.level" in logs assert "--logging.level" in logs
# individual module configs # 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): 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" assert orchestrator.formatters[0].name == "example_module"
def test_load_settings_for_module_from_commandline(orchestrator, test_args): 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) orchestrator.setup(args)
assert len(orchestrator.feeders) == 1 assert len(orchestrator.feeders) == 1
assert orchestrator.feeders[0].name == "gsheet_feeder" assert orchestrator.feeders[0].name == "gsheet_feeder_db"
assert orchestrator.config['gsheet_feeder']['sheet_id'] == "123" assert orchestrator.config['gsheet_feeder_db']['sheet_id'] == "123"
def test_multiple_orchestrator(test_args): 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() o1 = ArchivingOrchestrator()
with pytest.raises(ValueError) as exit_error: 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) o1.setup(o1_args)