def test_fill_input_queue(self):
bridge = EdgeDataBridge(self.config)
return_value = [
{'id': uuid.uuid4().hex,
'dateModified': datetime.datetime.utcnow().isoformat()}
]
bridge.feeder.get_resource_items = MagicMock(return_value=return_value)
self.assertEqual(bridge.input_queue.qsize(), 0)
bridge.fill_input_queue()
self.assertEqual(bridge.input_queue.qsize(), 1)
self.assertEqual(bridge.input_queue.get(), return_value[0])
def test_fill_input_queue(self):
bridge = EdgeDataBridge(self.config)
bridge.workers_config['historical'] = True
mock_api_client = MagicMock()
client_dict = {
'id': uuid.uuid4().hex,
'request_interval': 0.02,
'client': mock_api_client
}
bridge.api_clients_queue.put(client_dict)
return_value = [{
'id': uuid.uuid4().hex,
'dateModified': datetime.datetime.utcnow().isoformat()
}]
resource_item_return_value = {
'id': return_value[0]['id'],
'x_revision_n': randint(10, 99)
}
mock_api_client.get_resource_item_historical.return_value = resource_item_return_value
bridge.feeder.get_resource_items = MagicMock(return_value=return_value)
self.assertEqual(bridge.input_queue.qsize(), 0)
bridge.fill_input_queue()
self.assertEqual(bridge.resource_items_queue.qsize(),
resource_item_return_value['x_revision_n'])
bridge.resource_items_queue = Queue()
mock_api_client.get_resource_item_historical.side_effect = RequestFailed(
munchify({'status_code': 429}))
bridge.fill_input_queue()
self.assertEqual(bridge.resource_items_queue.qsize(), 0)
bridge = EdgeDataBridge(self.config)
return_value = [{
'id': uuid.uuid4().hex,
'dateModified': datetime.datetime.utcnow().isoformat()
}]
bridge.feeder.get_resource_items = MagicMock(return_value=return_value)
self.assertEqual(bridge.input_queue.qsize(), 0)
bridge.fill_input_queue()
self.assertEqual(bridge.input_queue.qsize(), 1)
self.assertEqual(bridge.input_queue.get(), return_value[0])