def test_get_articles_data(req_mock):
expected_payload = {
"consumer_key": "1234",
"access_token": "5678",
"count": 7,
"detailType": "complete",
}
api.get_articles_data(count=7)
req_mock.post.assert_called_once_with("test_url",
json=expected_payload,
timeout=5)
def test_get_articles_data(req_mock):
expected_payload = {
'consumer_key': '1234',
'access_token': '5678',
'count': 7,
'detailType': 'complete',
}
api.get_articles_data(count=7)
req_mock.post.assert_called_once_with('test_url',
json=expected_payload,
timeout=5)
def test_get_articles_data(req_mock):
expected_payload = {
'consumer_key': '1234',
'access_token': '5678',
'count': 7,
'detailType': 'complete',
}
api.get_articles_data(count=7)
req_mock.post.assert_called_once_with('test_url',
json=expected_payload,
timeout=5)
def refresh(self, count=None):
# request data from the API
articles = get_articles_data(count=count)
if articles and len(articles['recommendations']) > 0:
# if API gave us data, insert/update new data
# returns tuple: number updated, number deleted
return self.update_articles(articles['recommendations'])
# no data returned. something wrong.
return None, None
def refresh(self, count=None):
# request data from the API
articles = get_articles_data(count=count)
if articles and len(articles['recommendations']) > 0:
# if API gave us data, insert/update new data
# returns tuple: number updated, number deleted
return self.update_articles(articles['recommendations'])
# no data returned. something wrong.
return None, None
def test_refresh_articles_with_dupes():
setup_responses()
updated, deleted = PocketArticle.objects.refresh()
assert updated == 3
assert deleted == 0
# add dupes
articles = api.get_articles_data()["recommendations"]
api.complete_articles_data((None, a) for a in articles)
for article in articles:
PocketArticle.objects.create(**article)
assert PocketArticle.objects.count() == 7
PocketArticle.objects.refresh()
assert PocketArticle.objects.count() == 3
def test_refresh_articles_with_dupes():
setup_responses()
updated, deleted = PocketArticle.objects.refresh()
assert updated == 3
assert deleted == 0
# add dupes
articles = api.get_articles_data()['recommendations']
api.complete_articles_data((None, a) for a in articles)
for article in articles:
PocketArticle.objects.create(**article)
assert PocketArticle.objects.count() == 7
PocketArticle.objects.refresh()
assert PocketArticle.objects.count() == 3
def test_get_articles_data():
setup_responses()
articles = api.get_articles_data()
assert len(articles['recommendations']) == 3
assert articles['recommendations'][0]['id'] == 2262198874
assert articles['recommendations'][1]['id'] == 2248108002
def test_get_articles_data():
setup_responses()
articles = api.get_articles_data()
assert len(articles["recommendations"]) == 4
assert articles["recommendations"][0]["id"] == 2262198874
assert articles["recommendations"][2]["id"] == 2248108002
def test_get_articles_data():
setup_responses()
articles = api.get_articles_data()
assert len(articles['recommendations']) == 4
assert articles['recommendations'][0]['id'] == 2262198874
assert articles['recommendations'][2]['id'] == 2248108002
