def test_processing(self): processor = Raw2DataProcessor() now = datetime.now() source = [(10, 'some_source1', 'cat1', datetime_2_str(now-timedelta(minutes=59)), 'some_data'), (11, 'some_source2', 'cat2', datetime_2_str(now-timedelta(hours=57)), 'some_data2'), (12, 'some_source1', 'cat1', datetime_2_str(now-timedelta(minutes=46)), 'some_data3')] expected_items = [StatDataItem(10, 'some_source1', 'cat1', (now-timedelta(minutes=59)).replace(microsecond=0), 'some_data'), StatDataItem(11, 'some_source2', 'cat2', (now-timedelta(hours=57)).replace(microsecond=0), 'some_data2'), StatDataItem(12, 'some_source1', 'cat1', (now-timedelta(minutes=46)).replace(microsecond=0), 'some_data3')] actual = processor.process(source) self.assertIsInstance(actual, StatData) self.assertEquals((10, 12), actual.id_range) self.assertEquals(expected_items, actual.stat_data_items)
def test_exception_when_clear(self): self._logger.info('execute() enter') self._storage.get_data().AndReturn([(13, 's1', 'c1', datetime_2_str(self._now), 'data1')]) self._user_identity_provider.get_user_identity().AndReturn(self._user_id) dest_data = '<?xml version="1.0" ?><data_packet user_id="' + str(self._user_id) + '">' +\ '<data_item><source>s1</source><category>c1</category><timemarker>' + self._str_now + '</timemarker><data>data1</data></data_item>' +\ '</data_packet>' self._logger.getChild('unreliable_task_executer').AndReturn(self._child_logger) self._child_logger.info('execute() enter') self._child_logger.info('execute(): iteration number 1') self._endpoint.send(dest_data).AndReturn(True) self._child_logger.info('execute() exit with result successfully') self._storage.clear((13, 13)).AndRaise(Exception()) self._logger.exception('exception in execute()') self._test_common_action(False)
def test_exception_when_get_user_identity(self): self._logger.info('execute() enter') self._storage.get_data().AndReturn([(13, 's1', 'c1', datetime_2_str(self._now), 'data1')]) self._user_identity_provider.get_user_identity().AndRaise(Exception()) self._logger.exception('exception in execute()') self._test_common_action(False)