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
