summaryrefslogtreecommitdiffstats
path: root/tests/test_issue_db.py
blob: b29eea0f065a22e7e7e876df8985125b43716164 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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)