| OLD | NEW |
| (Empty) | |
| 1 # Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. |
| 4 |
| 5 from common.waterfall import failure_type |
| 6 from model import analysis_approach_type |
| 7 from model import suspected_cl_status |
| 8 from model.base_build_model import BaseBuildModel |
| 9 from model.wf_suspected_cl import WfSuspectedCL |
| 10 from waterfall import suspected_cl_util |
| 11 from waterfall.test import wf_testcase |
| 12 |
| 13 |
| 14 class SuspectedCLUtilTest(wf_testcase.WaterfallTestCase): |
| 15 |
| 16 def testCreateWfSuspectedCL(self): |
| 17 approach = analysis_approach_type.HEURISTIC |
| 18 master_name = 'm' |
| 19 builder_name = 'b' |
| 20 build_number = 123 |
| 21 compile_failure_type = failure_type.COMPILE |
| 22 repo_name = 'chromium' |
| 23 revision = 'r1' |
| 24 commit_position = 1 |
| 25 failures = {'compile': []} |
| 26 top_score = 5 |
| 27 |
| 28 self.assertIsNone(WfSuspectedCL.Get(repo_name, revision)) |
| 29 |
| 30 suspected_cl_util.UpdateSuspectedCL( |
| 31 repo_name, revision, commit_position, approach, master_name, |
| 32 builder_name, build_number, compile_failure_type, failures, top_score) |
| 33 |
| 34 expected_builds = { |
| 35 BaseBuildModel.CreateBuildId(master_name, builder_name, build_number ):{ |
| 36 'approaches': [approach], |
| 37 'failure_type': compile_failure_type, |
| 38 'failures': failures, |
| 39 'status': None, |
| 40 'top_score': top_score |
| 41 } |
| 42 } |
| 43 |
| 44 suspected_cl = WfSuspectedCL.Get(repo_name, revision) |
| 45 |
| 46 self.assertIsNotNone(suspected_cl) |
| 47 self.assertEqual( |
| 48 [analysis_approach_type.HEURISTIC], suspected_cl.approaches) |
| 49 self.assertEqual([compile_failure_type], suspected_cl.failure_type) |
| 50 self.assertEqual(expected_builds, suspected_cl.builds) |
| 51 |
| 52 def testUpdateWfSuspectedCLAddAnotherApproach(self): |
| 53 approach = analysis_approach_type.TRY_JOB |
| 54 master_name = 'm' |
| 55 builder_name = 'b' |
| 56 build_number = 122 |
| 57 test_failure_type = failure_type.TEST |
| 58 repo_name = 'chromium' |
| 59 revision = 'r2' |
| 60 commit_position = 2 |
| 61 failures = {'step_1': ['test1', 'test2']} |
| 62 top_score = None |
| 63 |
| 64 suspected_cl = WfSuspectedCL.Create(repo_name, revision, commit_position) |
| 65 suspected_cl.approaches = [analysis_approach_type.HEURISTIC] |
| 66 suspected_cl.builds = { |
| 67 BaseBuildModel.CreateBuildId( |
| 68 master_name, builder_name, build_number ): { |
| 69 'approaches': [analysis_approach_type.HEURISTIC], |
| 70 'failure_type': test_failure_type, |
| 71 'failures': failures, |
| 72 'status': None, |
| 73 'top_score': 4 |
| 74 } |
| 75 } |
| 76 suspected_cl.failure_type = [test_failure_type] |
| 77 suspected_cl.put() |
| 78 |
| 79 suspected_cl_util.UpdateSuspectedCL( |
| 80 repo_name, revision, commit_position, approach, master_name, |
| 81 builder_name, build_number, test_failure_type, failures, top_score) |
| 82 |
| 83 expected_builds = { |
| 84 BaseBuildModel.CreateBuildId( |
| 85 master_name, builder_name, build_number ): { |
| 86 'approaches': [ |
| 87 analysis_approach_type.HEURISTIC, |
| 88 analysis_approach_type.TRY_JOB], |
| 89 'failure_type': test_failure_type, |
| 90 'failures': failures, |
| 91 'status': None, |
| 92 'top_score': 4 |
| 93 } |
| 94 } |
| 95 |
| 96 expected_approaches = [ |
| 97 analysis_approach_type.HEURISTIC, |
| 98 analysis_approach_type.TRY_JOB] |
| 99 |
| 100 suspected_cl = WfSuspectedCL.Get(repo_name, revision) |
| 101 |
| 102 self.assertIsNotNone(suspected_cl) |
| 103 self.assertEqual(expected_approaches, suspected_cl.approaches) |
| 104 self.assertEqual([test_failure_type], suspected_cl.failure_type) |
| 105 self.assertEqual(expected_builds, suspected_cl.builds) |
| 106 |
| 107 def testUpdateWfSuspectedCLAddSameBuild(self): |
| 108 approach = analysis_approach_type.HEURISTIC |
| 109 master_name = 'm' |
| 110 builder_name = 'b' |
| 111 build_number = 122 |
| 112 test_failure_type = failure_type.TEST |
| 113 repo_name = 'chromium' |
| 114 revision = 'r2' |
| 115 commit_position = 2 |
| 116 failures = {'step_1': ['test1', 'test2']} |
| 117 top_score = 4 |
| 118 |
| 119 suspected_cl = WfSuspectedCL.Create(repo_name, revision, commit_position) |
| 120 suspected_cl.approaches = [analysis_approach_type.HEURISTIC] |
| 121 suspected_cl.builds = { |
| 122 BaseBuildModel.CreateBuildId( |
| 123 master_name, builder_name, build_number ): { |
| 124 'approaches': [analysis_approach_type.HEURISTIC], |
| 125 'failure_type': test_failure_type, |
| 126 'failures': failures, |
| 127 'status': None, |
| 128 'top_score': 4 |
| 129 } |
| 130 } |
| 131 suspected_cl.failure_type = [test_failure_type] |
| 132 suspected_cl.put() |
| 133 |
| 134 suspected_cl_util.UpdateSuspectedCL( |
| 135 repo_name, revision, commit_position, approach, master_name, |
| 136 builder_name, build_number, test_failure_type, failures, top_score) |
| 137 |
| 138 expected_builds = { |
| 139 BaseBuildModel.CreateBuildId( |
| 140 master_name, builder_name, build_number ): { |
| 141 'approaches': [analysis_approach_type.HEURISTIC], |
| 142 'failure_type': test_failure_type, |
| 143 'failures': failures, |
| 144 'status': None, |
| 145 'top_score': 4 |
| 146 } |
| 147 } |
| 148 |
| 149 expected_approaches = [analysis_approach_type.HEURISTIC] |
| 150 |
| 151 suspected_cl = WfSuspectedCL.Get(repo_name, revision) |
| 152 |
| 153 self.assertIsNotNone(suspected_cl) |
| 154 self.assertEqual(expected_approaches, suspected_cl.approaches) |
| 155 self.assertEqual([test_failure_type], suspected_cl.failure_type) |
| 156 self.assertEqual(expected_builds, suspected_cl.builds) |
| 157 |
| 158 def testUpdateWfSuspectedCLAddAnotherHeuristic(self): |
| 159 approach = analysis_approach_type.HEURISTIC |
| 160 master_name = 'm' |
| 161 builder_name = 'b' |
| 162 build_number = 122 |
| 163 test_failure_type = failure_type.TEST |
| 164 repo_name = 'chromium' |
| 165 revision = 'r2' |
| 166 commit_position = 2 |
| 167 failures = {'step_1': ['test1', 'test2']} |
| 168 top_score = 4 |
| 169 |
| 170 suspected_cl = WfSuspectedCL.Create(repo_name, revision, commit_position) |
| 171 suspected_cl.approaches = [analysis_approach_type.HEURISTIC] |
| 172 suspected_cl.builds = { |
| 173 BaseBuildModel.CreateBuildId( |
| 174 master_name, builder_name, build_number-1): { |
| 175 'approaches': [analysis_approach_type.HEURISTIC], |
| 176 'failure_type': test_failure_type, |
| 177 'failures': {'step': ['test']}, |
| 178 'status': suspected_cl_status.CORRECT, |
| 179 'top_score': 4 |
| 180 }, |
| 181 BaseBuildModel.CreateBuildId( |
| 182 master_name, builder_name, build_number - 2): { |
| 183 'approaches': [analysis_approach_type.HEURISTIC], |
| 184 'failure_type': test_failure_type, |
| 185 'failures': failures, |
| 186 'status': suspected_cl_status.CORRECT, |
| 187 'top_score': 4 |
| 188 } |
| 189 } |
| 190 suspected_cl.failure_type = [test_failure_type] |
| 191 suspected_cl.put() |
| 192 |
| 193 suspected_cl_util.UpdateSuspectedCL( |
| 194 repo_name, revision, commit_position, approach, master_name, |
| 195 builder_name, build_number, test_failure_type, failures, top_score) |
| 196 |
| 197 expected_builds = { |
| 198 BaseBuildModel.CreateBuildId( |
| 199 master_name, builder_name, build_number-1): { |
| 200 'approaches': [analysis_approach_type.HEURISTIC], |
| 201 'failure_type': test_failure_type, |
| 202 'failures': {'step': ['test']}, |
| 203 'status': suspected_cl_status.CORRECT, |
| 204 'top_score': 4 |
| 205 }, |
| 206 BaseBuildModel.CreateBuildId( |
| 207 master_name, builder_name, build_number - 2): { |
| 208 'approaches': [analysis_approach_type.HEURISTIC], |
| 209 'failure_type': test_failure_type, |
| 210 'failures': failures, |
| 211 'status': suspected_cl_status.CORRECT, |
| 212 'top_score': 4 |
| 213 }, |
| 214 BaseBuildModel.CreateBuildId( |
| 215 master_name, builder_name, build_number): { |
| 216 'approaches': [analysis_approach_type.HEURISTIC], |
| 217 'failure_type': test_failure_type, |
| 218 'failures': failures, |
| 219 'status': suspected_cl_status.CORRECT, |
| 220 'top_score': 4 |
| 221 } |
| 222 } |
| 223 |
| 224 suspected_cl = WfSuspectedCL.Get(repo_name, revision) |
| 225 |
| 226 self.assertIsNotNone(suspected_cl) |
| 227 self.assertEqual( |
| 228 [analysis_approach_type.HEURISTIC], suspected_cl.approaches) |
| 229 self.assertEqual([test_failure_type], suspected_cl.failure_type) |
| 230 self.assertEqual(expected_builds, suspected_cl.builds) |
| 231 |
| 232 def testGetCLInfo(self): |
| 233 self.assertEqual(['chromium', 'rev1'], |
| 234 suspected_cl_util.GetCLInfo('chromium/rev1')) |
| OLD | NEW |