OLD | NEW |
1 #!/usr/bin/python -u | 1 #!/usr/bin/python -u |
2 # | 2 # |
3 # Copyright 2008 Google Inc. All Rights Reserved. | 3 # Copyright 2008 Google Inc. All Rights Reserved. |
4 | 4 |
5 | 5 |
6 """Tests for job.""" | 6 """Tests for job.""" |
7 | 7 |
8 import copy, getpass, unittest, sys, os | 8 import copy, getpass, unittest, sys, os |
9 | 9 |
10 import common | 10 import common |
(...skipping 813 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
824 file_temp = cli_mock.create_file(self.ctrl_file) | 824 file_temp = cli_mock.create_file(self.ctrl_file) |
825 self.run_cmd(argv=['atest', 'job', 'create', '-f', file_temp.name, | 825 self.run_cmd(argv=['atest', 'job', 'create', '-f', file_temp.name, |
826 'test_job0', '-m', 'host0', '-d', 'dep1, dep2 ', | 826 'test_job0', '-m', 'host0', '-d', 'dep1, dep2 ', |
827 '--ignore_site_file'], | 827 '--ignore_site_file'], |
828 rpcs=[('create_job', data, True, 42)], | 828 rpcs=[('create_job', data, True, 42)], |
829 out_words_ok=['test_job0', 'Created'], | 829 out_words_ok=['test_job0', 'Created'], |
830 out_words_no=['Uploading', 'Done']) | 830 out_words_no=['Uploading', 'Done']) |
831 file_temp.clean() | 831 file_temp.clean() |
832 | 832 |
833 | 833 |
| 834 def test_execute_create_job_with_control_and_comma_dependencies(self): |
| 835 data = self.data.copy() |
| 836 data['dependencies'] = ['dep2,False', 'dep1,True'] |
| 837 file_temp = cli_mock.create_file(self.ctrl_file) |
| 838 self.run_cmd(argv=['atest', 'job', 'create', '-f', file_temp.name, |
| 839 'test_job0', '-m', 'host0', '-d', |
| 840 'dep1\,True, dep2\,False ', '--ignore_site_file'], |
| 841 rpcs=[('create_job', data, True, 42)], |
| 842 out_words_ok=['test_job0', 'Created'], |
| 843 out_words_no=['Uploading', 'Done']) |
| 844 file_temp.clean() |
| 845 |
| 846 |
834 def test_execute_create_job_with_synch_count(self): | 847 def test_execute_create_job_with_synch_count(self): |
835 data = self.data.copy() | 848 data = self.data.copy() |
836 data['synch_count'] = 2 | 849 data['synch_count'] = 2 |
837 file_temp = cli_mock.create_file(self.ctrl_file) | 850 file_temp = cli_mock.create_file(self.ctrl_file) |
838 self.run_cmd(argv=['atest', 'job', 'create', '-f', file_temp.name, | 851 self.run_cmd(argv=['atest', 'job', 'create', '-f', file_temp.name, |
839 'test_job0', '-m', 'host0', '-y', '2', | 852 'test_job0', '-m', 'host0', '-y', '2', |
840 '--ignore_site_file'], | 853 '--ignore_site_file'], |
841 rpcs=[('create_job', data, True, 42)], | 854 rpcs=[('create_job', data, True, 42)], |
842 out_words_ok=['test_job0', 'Created'], | 855 out_words_ok=['test_job0', 'Created'], |
843 out_words_no=['Uploading', 'Done']) | 856 out_words_no=['Uploading', 'Done']) |
(...skipping 11 matching lines...) Expand all Loading... |
855 True, | 868 True, |
856 {'control_file' : self.ctrl_file, | 869 {'control_file' : self.ctrl_file, |
857 'synch_count' : 1, | 870 'synch_count' : 1, |
858 'is_server' : False, | 871 'is_server' : False, |
859 'dependencies' : ['dep3']}), | 872 'dependencies' : ['dep3']}), |
860 ('create_job', data, True, 42)], | 873 ('create_job', data, True, 42)], |
861 out_words_ok=['test_job0', 'Created'], | 874 out_words_ok=['test_job0', 'Created'], |
862 out_words_no=['Uploading', 'Done']) | 875 out_words_no=['Uploading', 'Done']) |
863 | 876 |
864 | 877 |
| 878 def test_execute_create_job_with_test_and_comma_dependencies(self): |
| 879 data = self.data.copy() |
| 880 data['dependencies'] = ['dep1,True', 'dep2,False', 'dep3,123'] |
| 881 self.run_cmd(argv=['atest', 'job', 'create', '-t', 'sleeptest', |
| 882 'test_job0', '-m', 'host0', '-d', |
| 883 'dep1\,True dep2\,False ', '--ignore_site_file'], |
| 884 rpcs=[('generate_control_file', |
| 885 {'tests': ['sleeptest']}, |
| 886 True, |
| 887 {'control_file' : self.ctrl_file, |
| 888 'synch_count' : 1, |
| 889 'is_server' : False, |
| 890 'dependencies' : ['dep3,123']}), |
| 891 ('create_job', data, True, 42)], |
| 892 out_words_ok=['test_job0', 'Created'], |
| 893 out_words_no=['Uploading', 'Done']) |
| 894 |
| 895 |
865 def test_execute_create_job_with_kernel(self): | 896 def test_execute_create_job_with_kernel(self): |
866 data = self.data.copy() | 897 data = self.data.copy() |
867 data['control_file'] = self.kernel_ctrl_file | 898 data['control_file'] = self.kernel_ctrl_file |
868 self.run_cmd(argv=['atest', 'job', 'create', '-t', 'sleeptest', | 899 self.run_cmd(argv=['atest', 'job', 'create', '-t', 'sleeptest', |
869 '-k', 'kernel', 'test_job0', '-m', 'host0', | 900 '-k', 'kernel', 'test_job0', '-m', 'host0', |
870 '--ignore_site_file'], | 901 '--ignore_site_file'], |
871 rpcs=[('generate_control_file', | 902 rpcs=[('generate_control_file', |
872 {'tests': ['sleeptest'], | 903 {'tests': ['sleeptest'], |
873 'kernel': [{'version': 'kernel'}], | 904 'kernel': [{'version': 'kernel'}], |
874 'do_push_packages': True}, | 905 'do_push_packages': True}, |
(...skipping 562 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1437 def test_execute_job_abort(self): | 1468 def test_execute_job_abort(self): |
1438 self.run_cmd(argv=['atest', 'job', 'abort', '180', | 1469 self.run_cmd(argv=['atest', 'job', 'abort', '180', |
1439 '--ignore_site_file'], | 1470 '--ignore_site_file'], |
1440 rpcs=[('abort_host_queue_entries', | 1471 rpcs=[('abort_host_queue_entries', |
1441 {'job__id__in': ['180']}, True, None)], | 1472 {'job__id__in': ['180']}, True, None)], |
1442 out_words_ok=['Aborting', '180']) | 1473 out_words_ok=['Aborting', '180']) |
1443 | 1474 |
1444 | 1475 |
1445 if __name__ == '__main__': | 1476 if __name__ == '__main__': |
1446 unittest.main() | 1477 unittest.main() |
OLD | NEW |