def setUp(self): process_name = PROCESS_SITE_DAILY self.aggregator = SiteDailyAggregator(process_name) self.publishers = PublishersPool(self.aggregator.logger) self.uow_id = create_and_insert_unit_of_work(PROCESS_SITE_DAILY, 0, 1, 2011091220)
class DailySiteAggregatorUnitTest(unittest.TestCase): def setUp(self): process_name = PROCESS_SITE_DAILY self.aggregator = SiteDailyAggregator(process_name) self.publishers = PublishersPool(self.aggregator.logger) self.uow_id = create_and_insert_unit_of_work(PROCESS_SITE_DAILY, 0, 1, 2011091220) def tearDown(self): # cleaning up DB uow_dao = UnitOfWorkDao(self.aggregator.logger) uow_dao.remove(self.uow_id) del self.aggregator def test_aggregation(self): message = TestMessage(process_name=self.aggregator.process_name, uow_id=self.uow_id) self.aggregator._mq_callback(message)
class DailySiteAggregatorUnitTest(unittest.TestCase): def create_unit_of_work(self, logger, process_name, first_object_id, last_object_id, timestamp): """ method is used to insert unit_of_work """ unit_of_work = UnitOfWorkEntry() unit_of_work.set_timestamp(timestamp) unit_of_work.set_start_timestamp(timestamp) unit_of_work.set_end_timestamp(timestamp) unit_of_work.set_start_id(first_object_id) unit_of_work.set_end_id(last_object_id) unit_of_work.set_source_collection(None) unit_of_work.set_target_collection(None) unit_of_work.set_state(UnitOfWorkEntry.STATE_REQUESTED) unit_of_work.set_process_name(process_name) unit_of_work.set_number_of_retries(0) uow_id = unit_of_work_helper.insert(logger, unit_of_work) return uow_id def setUp(self): process_name = PROCESS_SITE_DAILY self.aggregator = SiteDailyAggregator(process_name) self.publishers = PublishersPool(self.aggregator.logger) self.uow_id = self.create_unit_of_work(self.aggregator.logger, process_name, 0, 1, 2011091220) def tearDown(self): # cleaning up DB unit_of_work_helper.remove(self.aggregator.logger, self.uow_id) del self.aggregator def test_aggregation(self): message = TestMessage() message.body = str(self.uow_id) self.aggregator._mq_callback(message)