basic test for issue-db pipeline

1 files changed, 26 insertions, 0 deletions

diff --git a/tests/test_issue_db.py b/tests/test_issue_db.py
new file mode 100644
index 0000000..b29eea0
--- /dev/null
+++ b/tests/test_issue_db.py
@@ -0,0 +1,26 @@
+
+import elasticsearch
+import fatcat_openapi_client
+
+from fatcat_scholar.issue_db import IssueDB
+from fatcat_scholar.config import settings
+
+
+def test_issue_db_basics() -> None:
+
+    api_conf = fatcat_openapi_client.Configuration()
+    api_conf.host = settings.FATCAT_API_HOST
+    api = fatcat_openapi_client.DefaultApi(
+        fatcat_openapi_client.ApiClient(api_conf)
+    )
+
+    es_client = elasticsearch.Elasticsearch(settings.FATCAT_ELASTICSEARCH_BACKEND)
+
+    issue_db = IssueDB(settings.SCHOLAR_ISSUEDB_PATH)
+    issue_db.init_db()
+
+    with open('tests/files/sim_collections.json', 'r') as f:
+        issue_db.load_pubs(f.readlines(), api)
+
+    with open('tests/files/sim_items.json', 'r') as f:
+        issue_db.load_issues(f.readlines(), es_client)