aboutsummaryrefslogtreecommitdiffstats
path: root/tests/test_work_pipeline.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_work_pipeline.py')
-rw-r--r--tests/test_work_pipeline.py26
1 files changed, 26 insertions, 0 deletions
diff --git a/tests/test_work_pipeline.py b/tests/test_work_pipeline.py
new file mode 100644
index 0000000..439a186
--- /dev/null
+++ b/tests/test_work_pipeline.py
@@ -0,0 +1,26 @@
+from fatcat_scholar.issue_db import IssueDB
+from fatcat_scholar.sandcrawler import (
+ SandcrawlerPostgrestClient,
+ SandcrawlerMinioClient,
+)
+from fatcat_scholar.work_pipeline import *
+from fatcat_scholar.config import settings
+
+
+def test_run_transform() -> None:
+
+ issue_db = IssueDB(settings.SCHOLAR_ISSUEDB_PATH)
+ issue_db.init_db()
+
+ wp = WorkPipeline(
+ issue_db=issue_db,
+ sandcrawler_db_client=SandcrawlerPostgrestClient(
+ api_url=settings.SANDCRAWLER_DB_API
+ ),
+ sandcrawler_s3_client=SandcrawlerMinioClient(
+ host_url=settings.SANDCRAWLER_S3_API
+ ),
+ )
+
+ with open("tests/files/release_hsmo6p4smrganpb3fndaj2lon4_sans.json", "r") as f:
+ wp.run_releases(f.readlines())