diff options
Diffstat (limited to 'python/tests/import_journal_metadata.py')
| -rw-r--r-- | python/tests/import_journal_metadata.py | 27 |
1 files changed, 14 insertions, 13 deletions
diff --git a/python/tests/import_journal_metadata.py b/python/tests/import_journal_metadata.py index cdf4e8ed..ff9dcb7a 100644 --- a/python/tests/import_journal_metadata.py +++ b/python/tests/import_journal_metadata.py @@ -1,4 +1,3 @@ - import pytest from fixtures import * @@ -9,32 +8,34 @@ from fatcat_tools.importers import JournalMetadataImporter, JsonLinePusher def journal_metadata_importer(api): yield JournalMetadataImporter(api) + # TODO: use API to check that entities actually created... def test_journal_metadata_importer_batch(journal_metadata_importer): - with open('tests/files/journal_metadata.sample.json', 'r') as f: + with open("tests/files/journal_metadata.sample.json", "r") as f: JsonLinePusher(journal_metadata_importer, f).run() + def test_journal_metadata_importer(journal_metadata_importer): last_index = journal_metadata_importer.api.get_changelog(limit=1)[0].index - with open('tests/files/journal_metadata.sample.json', 'r') as f: + with open("tests/files/journal_metadata.sample.json", "r") as f: journal_metadata_importer.bezerk_mode = True counts = JsonLinePusher(journal_metadata_importer, f).run() - assert counts['insert'] == 20 - assert counts['exists'] == 0 - assert counts['skip'] == 0 + assert counts["insert"] == 20 + assert counts["exists"] == 0 + assert counts["skip"] == 0 # fetch most recent editgroup - change = journal_metadata_importer.api.get_changelog_entry(index=last_index+1) + change = journal_metadata_importer.api.get_changelog_entry(index=last_index + 1) eg = change.editgroup assert eg.description assert "container" in eg.description.lower() - assert eg.extra['git_rev'] - assert "fatcat_tools.JournalMetadataImporter" in eg.extra['agent'] + assert eg.extra["git_rev"] + assert "fatcat_tools.JournalMetadataImporter" in eg.extra["agent"] - with open('tests/files/journal_metadata.sample.json', 'r') as f: + with open("tests/files/journal_metadata.sample.json", "r") as f: journal_metadata_importer.reset() journal_metadata_importer.bezerk_mode = False counts = JsonLinePusher(journal_metadata_importer, f).run() - assert counts['insert'] == 0 - assert counts['exists'] == 20 - assert counts['skip'] == 0 + assert counts["insert"] == 0 + assert counts["exists"] == 20 + assert counts["skip"] == 0 |