OLD | NEW |
1 #!/usr/bin/python | 1 #!/usr/bin/python |
2 | 2 |
3 import logging, os, unittest | 3 import logging, os, unittest |
4 import common | 4 import common |
5 from autotest_lib.client.common_lib import enum, global_config, host_protections | 5 from autotest_lib.client.common_lib import enum, global_config, host_protections |
6 from autotest_lib.database import database_connection | 6 from autotest_lib.database import database_connection |
7 from autotest_lib.frontend import setup_django_environment | 7 from autotest_lib.frontend import setup_django_environment |
8 from autotest_lib.frontend.afe import frontend_test_utils, models | 8 from autotest_lib.frontend.afe import frontend_test_utils, models |
9 from autotest_lib.frontend.afe import model_attributes | 9 from autotest_lib.frontend.afe import model_attributes |
10 from autotest_lib.scheduler import drone_manager, email_manager, monitor_db | 10 from autotest_lib.scheduler import drone_manager, email_manager, host_scheduler |
11 from autotest_lib.scheduler import scheduler_models | 11 from autotest_lib.scheduler import monitor_db, scheduler_models |
12 | 12 |
13 # translations necessary for scheduler queries to work with SQLite | 13 # translations necessary for scheduler queries to work with SQLite |
14 _re_translator = database_connection.TranslatingDatabase.make_regexp_translator | 14 _re_translator = database_connection.TranslatingDatabase.make_regexp_translator |
15 _DB_TRANSLATORS = ( | 15 _DB_TRANSLATORS = ( |
16 _re_translator(r'NOW\(\)', 'time("now")'), | 16 _re_translator(r'NOW\(\)', 'time("now")'), |
17 _re_translator(r'LAST_INSERT_ID\(\)', 'LAST_INSERT_ROWID()'), | 17 _re_translator(r'LAST_INSERT_ID\(\)', 'LAST_INSERT_ROWID()'), |
18 # older SQLite doesn't support group_concat, so just don't bother until | 18 # older SQLite doesn't support group_concat, so just don't bother until |
19 # it arises in an important query | 19 # it arises in an important query |
20 _re_translator(r'GROUP_CONCAT\((.*?)\)', r'\1'), | 20 _re_translator(r'GROUP_CONCAT\((.*?)\)', r'\1'), |
21 ) | 21 ) |
(...skipping 702 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
724 host=queue_entry.host, | 724 host=queue_entry.host, |
725 task=models.SpecialTask.Task.VERIFY, | 725 task=models.SpecialTask.Task.VERIFY, |
726 requested_by=models.User.current_user()) | 726 requested_by=models.User.current_user()) |
727 models.SpecialTask.objects.create( | 727 models.SpecialTask.objects.create( |
728 host=queue_entry.host, | 728 host=queue_entry.host, |
729 task=models.SpecialTask.Task.CLEANUP, | 729 task=models.SpecialTask.Task.CLEANUP, |
730 queue_entry=queue_entry, | 730 queue_entry=queue_entry, |
731 is_complete=True, | 731 is_complete=True, |
732 requested_by=models.User.current_user()) | 732 requested_by=models.User.current_user()) |
733 | 733 |
734 self.assertRaises(monitor_db.SchedulerError, self._initialize_test) | 734 self.assertRaises(host_scheduler.SchedulerError, self._initialize_test) |
735 | 735 |
736 | 736 |
737 def _test_recover_verifying_hqe_helper(self, task, pidfile_type): | 737 def _test_recover_verifying_hqe_helper(self, task, pidfile_type): |
738 _, queue_entry = self._make_job_and_queue_entry() | 738 _, queue_entry = self._make_job_and_queue_entry() |
739 queue_entry.status = HqeStatus.VERIFYING | 739 queue_entry.status = HqeStatus.VERIFYING |
740 queue_entry.save() | 740 queue_entry.save() |
741 | 741 |
742 special_task = models.SpecialTask.objects.create( | 742 special_task = models.SpecialTask.objects.create( |
743 host=queue_entry.host, task=task, queue_entry=queue_entry) | 743 host=queue_entry.host, task=task, queue_entry=queue_entry) |
744 | 744 |
(...skipping 323 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1068 self.assert_(job_keyval_path in attached_files, attached_files) | 1068 self.assert_(job_keyval_path in attached_files, attached_files) |
1069 keyval_contents = attached_files[job_keyval_path] | 1069 keyval_contents = attached_files[job_keyval_path] |
1070 keyval_dict = dict(line.strip().split('=', 1) | 1070 keyval_dict = dict(line.strip().split('=', 1) |
1071 for line in keyval_contents.splitlines()) | 1071 for line in keyval_contents.splitlines()) |
1072 self.assert_('job_queued' in keyval_dict, keyval_dict) | 1072 self.assert_('job_queued' in keyval_dict, keyval_dict) |
1073 self.assertEquals(keyval_dict['mykey'], 'myvalue') | 1073 self.assertEquals(keyval_dict['mykey'], 'myvalue') |
1074 | 1074 |
1075 | 1075 |
1076 if __name__ == '__main__': | 1076 if __name__ == '__main__': |
1077 unittest.main() | 1077 unittest.main() |
OLD | NEW |