def test__acquire_sample_with_constraints_testing_flag(
self, _publish_data_mock, _get_data_mock,
_init_acquisition_cycle_mock, EventPublisher_mock,
_constraints_for_historical_request_mock):
granule1 = Mock(spec=Granule)
granule2 = Mock(spec=Granule)
granule3 = Mock(spec=Granule)
data_generator = [granule1, granule2, granule3]
_get_data_mock.return_value = data_generator
config = {
'constraints': {
'test_constraints': 'value'
},
'TESTING': True
}
publisher = Mock()
unlock_new_data_callback = Mock()
update_new_data_check_attachment = Mock()
EventPublisher_mock.publish_event = Mock()
BaseDataHandler._acquire_sample(
config=config,
publisher=publisher,
unlock_new_data_callback=unlock_new_data_callback,
update_new_data_check_attachment=update_new_data_check_attachment)
_init_acquisition_cycle_mock.assert_called_once_with(config)
_get_data_mock.assert_called_once_with(config)
_publish_data_mock.assert_called_once_with(
publisher, data_generator, config,
update_new_data_check_attachment)
_constraints_for_historical_request_mock.assert_called_once_with(
config)
EventPublisher_mock.publish_event.assert_called_once()
def test__acquire_sample_no_constraints(self, gevent_mock,
_constraints_for_new_request_mock,
_publish_data_mock,
_get_data_mock,
_init_acquisition_cycle_mock):
granule1 = Mock(spec=Granule)
granule2 = Mock(spec=Granule)
granule3 = Mock(spec=Granule)
data_generator = [granule1, granule2, granule3]
gevent_mock.getcurrent = Mock()
_constraints_for_new_request_mock.return_value = {
'constraints': {
'test_constraints': 'value'
}
}
_get_data_mock.return_value = data_generator
config = {}
publisher = Mock()
unlock_new_data_callback = Mock()
update_new_data_check_attachment = Mock()
BaseDataHandler._acquire_sample(config=config,
publisher=publisher,
unlock_new_data_callback=unlock_new_data_callback,
update_new_data_check_attachment=update_new_data_check_attachment)
_init_acquisition_cycle_mock.assert_called_once_with(config)
_constraints_for_new_request_mock.assert_called_once_with(config)
_get_data_mock.assert_called_once_with(config)
_publish_data_mock.assert_called_once_with(publisher, data_generator, config, update_new_data_check_attachment)
def test__acquire_sample_no_constraints(self, gevent_mock,
_constraints_for_new_request_mock,
_publish_data_mock, _get_data_mock,
_init_acquisition_cycle_mock):
granule1 = Mock(spec=Granule)
granule2 = Mock(spec=Granule)
granule3 = Mock(spec=Granule)
data_generator = [granule1, granule2, granule3]
gevent_mock.getcurrent = Mock()
_constraints_for_new_request_mock.return_value = {
'constraints': {
'test_constraints': 'value'
}
}
_get_data_mock.return_value = data_generator
config = {}
publisher = Mock()
unlock_new_data_callback = Mock()
update_new_data_check_attachment = Mock()
BaseDataHandler._acquire_sample(
config=config,
publisher=publisher,
unlock_new_data_callback=unlock_new_data_callback,
update_new_data_check_attachment=update_new_data_check_attachment)
_init_acquisition_cycle_mock.assert_called_once_with(config)
_constraints_for_new_request_mock.assert_called_once_with(config)
_get_data_mock.assert_called_once_with(config)
_publish_data_mock.assert_called_once_with(
publisher, data_generator, config,
update_new_data_check_attachment)
def test__acquire_sample_with_constraints_testing_flag(
self,
_publish_data_mock,
_get_data_mock,
_init_acquisition_cycle_mock,
EventPublisher_mock,
_constraints_for_historical_request_mock,
):
granule1 = Mock(spec=Granule)
granule2 = Mock(spec=Granule)
granule3 = Mock(spec=Granule)
data_generator = [granule1, granule2, granule3]
_get_data_mock.return_value = data_generator
config = {"constraints": {"test_constraints": "value"}, "TESTING": True}
publisher = Mock()
unlock_new_data_callback = Mock()
update_new_data_check_attachment = Mock()
EventPublisher_mock.publish_event = Mock()
BaseDataHandler._acquire_sample(
config=config,
publisher=publisher,
unlock_new_data_callback=unlock_new_data_callback,
update_new_data_check_attachment=update_new_data_check_attachment,
)
_init_acquisition_cycle_mock.assert_called_once_with(config)
_get_data_mock.assert_called_once_with(config)
_publish_data_mock.assert_called_once_with(publisher, data_generator)
_constraints_for_historical_request_mock.assert_called_once_with(config)
EventPublisher_mock.publish_event.assert_called_once()
