| OLD | NEW |
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
| 5 | 5 |
| 6 """Unit tests for verification/try_job_on_rietveld.py.""" | 6 """Unit tests for verification/try_job_on_rietveld.py.""" |
| 7 | 7 |
| 8 import logging | 8 import logging |
| 9 import os | 9 import os |
| 10 import random | 10 import random |
| (...skipping 59 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 70 revision, key, completed) | 70 revision, key, completed) |
| 71 return self.builders[builder].builds[number] | 71 return self.builders[builder].builds[number] |
| 72 | 72 |
| 73 | 73 |
| 74 def _posted(builders): | 74 def _posted(builders): |
| 75 return 'trigger_try_jobs(42, 23, \'CQ\', False, \'HEAD\', %s)' % str(builders) | 75 return 'trigger_try_jobs(42, 23, \'CQ\', False, \'HEAD\', %s)' % str(builders) |
| 76 | 76 |
| 77 | 77 |
| 78 def gen_job_pending(**kwargs): | 78 def gen_job_pending(**kwargs): |
| 79 value = { | 79 value = { |
| 80 '__persistent_module__': 'verification.try_job_on_rietveld', | |
| 81 '__persistent_type__': 'RietveldTryJobPending', | 80 '__persistent_type__': 'RietveldTryJobPending', |
| 82 'builder': None, | 81 'builder': None, |
| 83 'clobber': False, | 82 'clobber': False, |
| 84 'requested_steps': [], | 83 'requested_steps': [], |
| 85 'revision': None, | 84 'revision': None, |
| 86 'tries': 1, | 85 'tries': 1, |
| 87 } | 86 } |
| 88 assert all(arg in value for arg in kwargs) | 87 assert all(arg in value for arg in kwargs) |
| 89 value.update(kwargs) | 88 value.update(kwargs) |
| 90 return value | 89 return value |
| 91 | 90 |
| 92 | 91 |
| 93 def gen_job(**kwargs): | 92 def gen_job(**kwargs): |
| 94 value = { | 93 value = { |
| 95 '__persistent_module__': 'verification.try_job_on_rietveld', | |
| 96 '__persistent_type__': 'RietveldTryJob', | 94 '__persistent_type__': 'RietveldTryJob', |
| 97 'build': None, | 95 'build': None, |
| 98 'builder': None, | 96 'builder': None, |
| 99 'clobber': False, | 97 'clobber': False, |
| 100 'completed': False, | 98 'completed': False, |
| 101 'init_time': 1, | 99 'init_time': 1, |
| 102 'requested_steps': [], | 100 'requested_steps': [], |
| 103 'revision': None, | 101 'revision': None, |
| 104 'started': 1, | 102 'started': 1, |
| 105 'steps_failed': [], | 103 'steps_failed': [], |
| 106 'steps_passed': [], | 104 'steps_passed': [], |
| 107 'tries': 1, | 105 'tries': 1, |
| 108 'parent_key': None, | 106 'parent_key': None, |
| 109 } | 107 } |
| 110 assert all(arg in value for arg in kwargs) | 108 assert all(arg in value for arg in kwargs) |
| 111 value.update(kwargs) | 109 value.update(kwargs) |
| 112 return value | 110 return value |
| 113 | 111 |
| 114 | 112 |
| 115 def gen_jobs(**kwargs): | 113 def gen_jobs(**kwargs): |
| 116 value = { | 114 value = { |
| 117 '__persistent_module__': 'verification.try_job_on_rietveld', | |
| 118 '__persistent_type__': 'RietveldTryJobs', | 115 '__persistent_type__': 'RietveldTryJobs', |
| 119 'builders_and_tests': {}, | 116 'builders_and_tests': {}, |
| 120 'triggered_builders': {}, | 117 'triggered_builders': {}, |
| 121 'error_message': None, | 118 'error_message': None, |
| 122 'irrelevant': [], | 119 'irrelevant': [], |
| 123 'pendings': [], | 120 'pendings': [], |
| 124 'skipped': False, | 121 'skipped': False, |
| 125 'try_jobs': {}, | 122 'try_jobs': {}, |
| 126 } | 123 } |
| 127 assert all(arg in value for arg in kwargs) | 124 assert all(arg in value for arg in kwargs) |
| (...skipping 796 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 924 self.assertEquals({}, jobs.tests_need_to_be_run()) | 921 self.assertEquals({}, jobs.tests_need_to_be_run()) |
| 925 self.assertEquals({'tester1': ['test3']}, jobs.tests_waiting_for_result()) | 922 self.assertEquals({'tester1': ['test3']}, jobs.tests_waiting_for_result()) |
| 926 | 923 |
| 927 | 924 |
| 928 if __name__ == '__main__': | 925 if __name__ == '__main__': |
| 929 logging.basicConfig( | 926 logging.basicConfig( |
| 930 level=[logging.WARNING, logging.INFO, logging.DEBUG][ | 927 level=[logging.WARNING, logging.INFO, logging.DEBUG][ |
| 931 min(sys.argv.count('-v'), 2)], | 928 min(sys.argv.count('-v'), 2)], |
| 932 format='%(levelname)5s %(module)15s(%(lineno)3d): %(message)s') | 929 format='%(levelname)5s %(module)15s(%(lineno)3d): %(message)s') |
| 933 unittest.main() | 930 unittest.main() |
| OLD | NEW |