diff options
author | Bryan Newbold <bnewbold@robocracy.org> | 2021-11-02 18:13:14 -0700 |
---|---|---|
committer | Bryan Newbold <bnewbold@robocracy.org> | 2021-11-02 18:13:14 -0700 |
commit | cdfd6b85b386b7bbf9d5a5179ef26970b6e5a4e7 (patch) | |
tree | 5e4034027b51f3ee4d2a488bb2cbb7a75c3bd0d8 /python/tests/import_matched.py | |
parent | 78f08280edea4ff65ca613ad30005c45cc48dea6 (diff) | |
download | fatcat-cdfd6b85b386b7bbf9d5a5179ef26970b6e5a4e7.tar.gz fatcat-cdfd6b85b386b7bbf9d5a5179ef26970b6e5a4e7.zip |
fmt (black): tests/
Diffstat (limited to 'python/tests/import_matched.py')
-rw-r--r-- | python/tests/import_matched.py | 30 |
1 files changed, 16 insertions, 14 deletions
diff --git a/python/tests/import_matched.py b/python/tests/import_matched.py index aad8e799..4ade6e04 100644 --- a/python/tests/import_matched.py +++ b/python/tests/import_matched.py @@ -1,4 +1,3 @@ - import json import pytest @@ -11,39 +10,42 @@ from fatcat_tools.importers import JsonLinePusher, MatchedImporter def matched_importer(api): yield MatchedImporter(api) + # TODO: use API to check that entities actually created... def test_matched_importer_basic(matched_importer): - with open('tests/files/example_matched.json', 'r') as f: + with open("tests/files/example_matched.json", "r") as f: JsonLinePusher(matched_importer, f).run() + def test_matched_importer(matched_importer): last_index = matched_importer.api.get_changelog(limit=1)[0].index - with open('tests/files/example_matched.json', 'r') as f: + with open("tests/files/example_matched.json", "r") as f: matched_importer.bezerk_mode = True counts = JsonLinePusher(matched_importer, f).run() - assert counts['insert'] == 2 - assert counts['exists'] == 0 - assert counts['skip'] == 11 + assert counts["insert"] == 2 + assert counts["exists"] == 0 + assert counts["skip"] == 11 # fetch most recent editgroup - change = matched_importer.api.get_changelog_entry(index=last_index+1) + change = matched_importer.api.get_changelog_entry(index=last_index + 1) eg = change.editgroup assert eg.description assert "file-to-release" in eg.description.lower() - assert eg.extra['git_rev'] - assert "fatcat_tools.MatchedImporter" in eg.extra['agent'] + assert eg.extra["git_rev"] + assert "fatcat_tools.MatchedImporter" in eg.extra["agent"] # re-insert; should skip - with open('tests/files/example_matched.json', 'r') as f: + with open("tests/files/example_matched.json", "r") as f: matched_importer.reset() matched_importer.bezerk_mode = False counts = JsonLinePusher(matched_importer, f).run() - assert counts['insert'] == 0 - assert counts['exists'] == 2 - assert counts['skip'] == 11 + assert counts["insert"] == 0 + assert counts["exists"] == 2 + assert counts["skip"] == 11 + def test_matched_dict_parse(matched_importer): - with open('tests/files/example_matched.json', 'r') as f: + with open("tests/files/example_matched.json", "r") as f: raw = json.loads(f.readline()) f = matched_importer.parse_record(raw) assert f.sha1 == "00242a192acc258bdfdb151943419437f440c313" |