Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(270)

Side by Side Diff: appengine/findit/waterfall/test/suspected_cl_util_test.py

Issue 2230103002: [Findit] Pipeline change to save suspected cls to data store. (Closed) Base URL: https://chromium.googlesource.com/infra/infra.git@0808-resubmit-suspected_cl_model
Patch Set: . Created 4 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(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'))
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698