def test_get_special_tasks(self): self._setup_special_tasks() tasks = rpc_interface.get_special_tasks(host__hostname='host1', queue_entry__isnull=True) self.assertEquals(len(tasks), 2) self.assertEquals(tasks[0]['task'], models.SpecialTask.Task.VERIFY) self.assertEquals(tasks[0]['is_active'], False) self.assertEquals(tasks[0]['is_complete'], True)
def test_reverify_hosts(self): hostname_list = rpc_interface.reverify_hosts(id__in=[1, 2]) self.assertEquals(hostname_list, ['host1', 'host2']) tasks = rpc_interface.get_special_tasks() self.assertEquals(len(tasks), 2) self.assertEquals(set(task['host']['id'] for task in tasks), set([1, 2])) task = tasks[0] self.assertEquals(task['task'], models.SpecialTask.Task.VERIFY) self.assertEquals(task['requested_by'], 'autotest_system')
def test_get_latest_special_task(self): # a particular usage of get_special_tasks() self._setup_special_tasks() self.task2.time_started = datetime.datetime(2009, 1, 2) self.task2.save() tasks = rpc_interface.get_special_tasks( host__hostname='host1', task=models.SpecialTask.Task.VERIFY, time_started__isnull=False, sort_by=['-time_started'], query_limit=1) self.assertEquals(len(tasks), 1) self.assertEquals(tasks[0]['id'], 2)
def test_get_latest_special_task(self): # a particular usage of get_special_tasks() self._setup_special_tasks() self.task2.time_started = datetime.datetime(2009, 1, 2) self.task2.save() tasks = rpc_interface.get_special_tasks( host__hostname='host1', task=models.SpecialTask.Task.VERIFY, time_started__isnull=False, sort_by=['-time_started'], query_limit=1) self.assertEqual(len(tasks), 1) self.assertEqual(tasks[0]['id'], 2)