| 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 |