def test_should_schedule_invoking_task_for_retry(self): self.result.retry_task = common.RetryTasks.INVOKE_SAME_TASK # Schedule this test method as the passed-in 'retry' function. retry_rpc_method = server.schedule_order_retry_tasks( self.test_should_schedule_invoking_task_for_retry, self.result, None, # Not used. *self.args, **self.kwargs) database_utils.get_session().commit() # Flush to the database. self.assertEqual( 'test_should_schedule_invoking_task_for_retry', retry_rpc_method)
def test_should_schedule_certificate_status_task_for_retry(self): self.result.retry_task = ( common.RetryTasks.INVOKE_CERT_STATUS_CHECK_TASK) # Schedule this test method as the passed-in 'retry' function. retry_rpc_method = server.schedule_order_retry_tasks( None, # Should be ignored for non-self retries. self.result, None, # Not used. *self.args, **self.kwargs) database_utils.get_session().commit() # Flush to the database. self.assertEqual('check_certificate_status', retry_rpc_method) self._verify_retry_task_entity('check_certificate_status')
def setUp(self): super(WhenCallingScheduleOrderRetryTasks, self).setUp() self.project = database_utils.create_project() self.order = database_utils.create_order(self.project) database_utils.get_session().commit() self.repo = repositories.OrderRetryTaskRepo() self.result = common.FollowOnProcessingStatusDTO() self.args = ['args-foo', 'args-bar'] self.kwargs = {'order_id': self.order.id, 'foo': 1, 'bar': 2} self.date_to_retry_at = ( datetime.datetime.utcnow() + datetime.timedelta(milliseconds=self.result.retry_msec))
def setUp(self): super(WhenCallingScheduleOrderRetryTasks, self).setUp() self.project = database_utils.create_project() self.order = database_utils.create_order(self.project) database_utils.get_session().commit() self.repo = repositories.OrderRetryTaskRepo() self.result = common.FollowOnProcessingStatusDTO() self.args = ['args-foo', 'args-bar'] self.kwargs = {'order_id': self.order.id, 'foo': 1, 'bar': 2} self.date_to_retry_at = ( datetime.datetime.utcnow() + datetime.timedelta( milliseconds=self.result.retry_msec) )
def test_should_schedule_certificate_status_task_for_retry(self): self.result.retry_task = ( common.RetryTasks.INVOKE_CERT_STATUS_CHECK_TASK ) # Schedule this test method as the passed-in 'retry' function. retry_rpc_method = server.schedule_order_retry_tasks( None, # Should be ignored for non-self retries. self.result, None, # Not used. *self.args, **self.kwargs) database_utils.get_session().commit() # Flush to the database. self.assertEqual( 'check_certificate_status', retry_rpc_method) self._verify_retry_task_entity( 'check_certificate_status')
def _create_retry_task(self): # Add one retry task: task = 'test_task' args = ('foo', 'bar') kwargs = {'k_foo': 1, 'k_bar': 2} order = database_utils.create_order() retry = models.OrderRetryTask() retry.order_id = order.id retry.retry_at = datetime.datetime.utcnow() retry.retry_task = task retry.retry_args = args retry.retry_kwargs = kwargs retry_repo = repositories.get_order_retry_tasks_repository() retry_repo.create_from(retry) database_utils.get_session().commit() return args, kwargs, retry_repo
def _create_retry_task(self): # Add one retry task: task = 'test_task' args = ('foo', 'bar') kwargs = {'k_foo': 1, 'k_bar': 2} order = database_utils.create_order() retry = models.OrderRetryTask() retry.order_id = order.id retry.retry_at = datetime.datetime.utcnow() retry.retry_task = task retry.retry_args = args retry.retry_kwargs = kwargs retry_repo = repositories.get_order_retry_tasks_repository() retry_repo.create_from(retry) database_utils.get_session().commit() return args, kwargs, retry_repo