def test_pub_locations_error(self):
with patch.object(QueryService, 'esearch') as mock:
mock.return_value = fx.dummy_error
result = queries.pubmed_locations(start=2010,
stop=2012,
term='test')
try:
next(result)
except QueryError as e:
pass
else:
self.assertFalse(True, "Exception should be Raised as QueryError")
def test_pub_locations_ok(self):
with patch.object(QueryService, 'esearch') as mock_search:
with patch.object(QueryService, 'efetch') as mock_fetch:
mock_search.return_value = fx.dummy_search_result
mock_fetch.return_value = fx.dummy_fetch_result
result = queries.pubmed_locations(start=2008, stop=2014, term='test')
self.assertIsInstance(result, types.GeneratorType)
res = list(result)
self.assertEqual(len(res), 1)
expected = [{'count': 2, 'country': 'united states'}]
self.assertEqual(res, expected)
