blob: 965b92527f6ec302f078ebc55ef9534edfa8519b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
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)
|