def _create_task_log(self, host, num_instances): task_log = objects.TaskLog(context.get_admin_context()) task_log.task_name = 'instance_usage_audit' task_log.period_beginning = '2012-06-01 00:00:00' task_log.period_ending = '2012-07-01 00:00:00' task_log.host = host task_log.task_items = num_instances task_log.message = ('Instance usage audit ran for host %s, %s ' 'instances in 0.01 seconds.' % (host, num_instances)) task_log.begin_task() task_log.errors = 1 task_log.end_task()
def test_end_task(self, mock_end_task): task_log = objects.TaskLog(self.context) task_log.task_name = fake_task_log['task_name'] task_log.period_beginning = fake_task_log['period_beginning'] task_log.period_ending = fake_task_log['period_ending'] task_log.host = fake_task_log['host'] task_log.errors = fake_task_log['errors'] task_log.message = fake_task_log['message'] task_log.end_task() mock_end_task.assert_called_once_with( self.context, fake_task_log['task_name'], fake_task_log['period_beginning'].replace(tzinfo=iso8601.UTC), fake_task_log['period_ending'].replace(tzinfo=iso8601.UTC), fake_task_log['host'], errors=fake_task_log['errors'], message=fake_task_log['message'])