def test__acquire_data_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_data( 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()
def test__acquire_data_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_data( 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)
def test__acquire_data_no_constraints(self, gevent_mock, _new_data_constraints_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() _new_data_constraints_mock.return_value = { 'constraints': 'test_constraints' } _get_data_mock.return_value = data_generator config = {} publisher = Mock() unlock_new_data_callback = Mock() BaseDataHandler._acquire_data( config=config, publisher=publisher, unlock_new_data_callback=unlock_new_data_callback) _init_acquisition_cycle_mock.assert_called_once_with(config) _new_data_constraints_mock.assert_called_once_with(config) _get_data_mock.assert_called_once_with(config) _publish_data_mock.assert_called_once_with(publisher, data_generator)
def test__acquire_data_with_constraints(self, _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] _get_data_mock.return_value = data_generator config = {'constraints' : 'test_constraints'} publisher = Mock() unlock_new_data_callback = Mock() BaseDataHandler._acquire_data(config=config, publisher=publisher, unlock_new_data_callback=unlock_new_data_callback) _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)
def test__acquire_data_with_constraints(self, _publish_data_mock, _get_data_mock, _init_acquisition_cycle_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'}} publisher = Mock() unlock_new_data_callback = Mock() update_new_data_check_attachment = Mock() BaseDataHandler._acquire_data(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_historical_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)
def test__acquire_data_with_constraints_testing_flag( self, _publish_data_mock, _get_data_mock, _init_acquisition_cycle_mock, EventPublisher_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', 'TESTING': True} publisher = Mock() unlock_new_data_callback = Mock() EventPublisher_mock.publish_event = Mock() BaseDataHandler._acquire_data( config=config, publisher=publisher, unlock_new_data_callback=unlock_new_data_callback) _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) EventPublisher_mock.publish_event.assert_called_once()