def test__poll(self, time_mock):
stream_route = Mock(spec=StreamRoute)
dh_config = {
'external_dataset_res_id': 'external_ds',
'stream_id': 'test_stream_id',
'stream_route': stream_route
}
bdh = BaseDataHandler(dh_config)
execute_acquire_sample_mock = MagicMock()
bdh.execute_acquire_sample = execute_acquire_sample_mock
bdh._params = {'POLLING_INTERVAL': 1}
glet = spawn(bdh._poll)
self.counter = 0
def inc_counter(*args):
self.counter += 1
if self.counter == 2:
bdh._polling = False
time_mock.sleep.side_effect = inc_counter
glet.join(timeout=5)
self.assertTrue(execute_acquire_sample_mock.call_count >= 1)
def setUp(self):
self._rr_cli = Mock()
self._stream_registrar = Mock()
dh_config = {'external_dataset_res_id': 'external_ds'}
self._bdh = BaseDataHandler(rr_cli=self._rr_cli,
stream_registrar=self._stream_registrar,
dh_config=dh_config)
def test__poll(self, time_mock, execute_acquire_data_mock):
bdh = BaseDataHandler(None, None, None)
bdh._params = {'POLLING_INTERVAL': 1}
glet = spawn(bdh._poll)
self.counter = 0
def inc_counter(*args):
self.counter += 1
if self.counter == 2:
bdh._polling = False
time_mock.sleep.side_effect = inc_counter
glet.join(timeout=5)
self.assertTrue(execute_acquire_data_mock.call_count >= 2)
def setUp(self):
dh_config = {'external_dataset_res_id': 'external_ds'}
self._bdh = BaseDataHandler(dh_config=dh_config)
