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()