From 3161319d0fe3feafe2b5fed48b9383680f9f0547 Mon Sep 17 00:00:00 2001 From: Bryan Newbold Date: Fri, 23 Oct 2020 00:06:50 -0700 Subject: basic test for issue-db pipeline --- tests/test_issue_db.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) create mode 100644 tests/test_issue_db.py (limited to 'tests/test_issue_db.py') 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) -- cgit v1.2.3