diff options
Diffstat (limited to 'python/tests/harvest_datacite.py')
| -rw-r--r-- | python/tests/harvest_datacite.py | 15 |
1 files changed, 8 insertions, 7 deletions
diff --git a/python/tests/harvest_datacite.py b/python/tests/harvest_datacite.py index 0a7584bd..40e5f711 100644 --- a/python/tests/harvest_datacite.py +++ b/python/tests/harvest_datacite.py @@ -1,4 +1,3 @@ - import datetime import json @@ -12,13 +11,12 @@ def test_datacite_harvest_date(mocker): # mock out the harvest state object so it doesn't try to actually connect # to Kafka - mocker.patch('fatcat_tools.harvest.harvest_common.HarvestState.initialize_from_kafka') + mocker.patch("fatcat_tools.harvest.harvest_common.HarvestState.initialize_from_kafka") # mock day request to datacite API - with open('tests/files/datacite_api.json', 'r') as f: + with open("tests/files/datacite_api.json", "r") as f: resp = json.loads(f.readline()) - responses.add(responses.GET, 'https://api.datacite.org/dois', - json=resp, status=200) + responses.add(responses.GET, "https://api.datacite.org/dois", json=resp, status=200) harvester = HarvestDataciteWorker( kafka_hosts="dummy", @@ -34,10 +32,13 @@ def test_datacite_harvest_date(mocker): assert len(responses.calls) == 1 # ensure email was included in User-Agent - assert "mailto:test@fatcat.wiki" in responses.calls[0].request.headers['User-Agent'] + assert "mailto:test@fatcat.wiki" in responses.calls[0].request.headers["User-Agent"] # check that correct date param was passed as expected - assert "query=updated%3A%5B2019-02-03T00%3A00%3A00.000Z+TO+2019-02-03T23%3A59%3A59.999Z%5D" in responses.calls[0].request.url + assert ( + "query=updated%3A%5B2019-02-03T00%3A00%3A00.000Z+TO+2019-02-03T23%3A59%3A59.999Z%5D" + in responses.calls[0].request.url + ) # check that we published the expected number of DOI objects were published # to the (mock) kafka topic |