refactor remaining importers

1 files changed, 5 insertions, 3 deletions

diff --git a/python/tests/import_journal_metadata.py b/python/tests/import_journal_metadata.py
index 81334bc6..0263f706 100644
--- a/python/tests/import_journal_metadata.py
+++ b/python/tests/import_journal_metadata.py
@@ -1,6 +1,6 @@
 
 import pytest
-from fatcat_tools.importers import JournalMetadataImporter
+from fatcat_tools.importers import JournalMetadataImporter, CsvPusher
 from fixtures import api
 
 
@@ -11,11 +11,13 @@ def journal_metadata_importer(api):
 # TODO: use API to check that entities actually created...
 def test_journal_metadata_importer_batch(journal_metadata_importer):
     with open('tests/files/journal_extra_metadata.snip.csv', 'r') as f:
-        journal_metadata_importer.process_csv_batch(f)
+        CsvPusher(journal_metadata_importer, f).run()
 
 def test_journal_metadata_importer(journal_metadata_importer):
     with open('tests/files/journal_extra_metadata.snip.csv', 'r') as f:
-        journal_metadata_importer.process_csv_source(f)
+        journal_metadata_importer.bezerk_mode = True
+        journal_metadata_importer.serial_mode = True
+        CsvPusher(journal_metadata_importer, f).run()
 
     # fetch most recent editgroup
     changes = journal_metadata_importer.api.get_changelog(limit=1)