Chromium Code Reviews| Index: appengine/findit/handlers/test/triage_analysis_test.py |
| diff --git a/appengine/findit/handlers/test/triage_analysis_test.py b/appengine/findit/handlers/test/triage_analysis_test.py |
| index acc0a0fc8ade8d85ba82ad7ed1caef4795ae0072..059bb55549e8d6854362fb89b3924c6446f4497e 100644 |
| --- a/appengine/findit/handlers/test/triage_analysis_test.py |
| +++ b/appengine/findit/handlers/test/triage_analysis_test.py |
| @@ -136,7 +136,9 @@ class TriageAnalysisTest(testing.AppengineTestCase): |
| '/triage-analysis', |
| params={'url': build_url, 'correct': True, 'format': 'json'}) |
| self.assertEquals(200, response.status_int) |
| - self.assertEquals({'success': True}, response.json_body) |
| + self.assertEquals( |
| + {'success': True, 'num_duplicate_analyses': 0}, |
|
lijeffrey
2016/06/27 20:38:22
nit: I would put this second field on a separate l
|
| + response.json_body) |
| def testIncompleteTriage(self): |
| build_url = buildbot.CreateBuildUrl( |
| @@ -145,7 +147,20 @@ class TriageAnalysisTest(testing.AppengineTestCase): |
| '/triage-analysis', |
| params={'url': build_url, 'correct': True, 'format': 'json'}) |
| self.assertEquals(200, response.status_int) |
| - self.assertEquals({'success': False}, response.json_body) |
| + self.assertEquals( |
| + {'success': False, 'num_duplicate_analyses': 0}, |
| + response.json_body) |
| + |
| + def testDuplicatePropertyWorksWhenNotDuplicate(self): |
| + analysis = WfAnalysis.Get( |
| + self.master_name, self.builder_name, self.build_number_found) |
| + self.assertFalse(analysis.duplicate) |
| + |
| + def testDuplicatePropertyWorksWhenDuplicate(self): |
| + analysis = WfAnalysis.Get( |
| + self.master_name, self.builder_name, self.build_number_found) |
| + analysis.result_status = result_status.FOUND_CORRECT_DUPLICATE |
| + self.assertTrue(analysis.duplicate) |
| def testAnalysesMatch(self): |
| analysis_with_empty_failures = WfAnalysis.Create( |
| @@ -364,11 +379,7 @@ class TriageAnalysisTest(testing.AppengineTestCase): |
| analysis.result = { |
| 'failures': [ |
| { |
| - 'suspected_cls': [ |
| - { |
| - 'revision': 'abc', |
| - } |
| - ], |
| + 'suspected_cls': self.suspected_cls, |
| 'step_name': 'turing_test', |
| } |
| ] |
| @@ -376,6 +387,7 @@ class TriageAnalysisTest(testing.AppengineTestCase): |
| analysis.result_status = result_status.FOUND_UNTRIAGED |
| analysis.build_start_time = build_start_time |
| analysis.status = analysis_status.COMPLETED |
| + analysis.suspected_cls = self.suspected_cls |
| analysis.put() |
| return analysis |
| @@ -447,7 +459,7 @@ class TriageAnalysisTest(testing.AppengineTestCase): |
| self.assertEquals( |
| len(triage_analysis._GetDuplicateAnalyses(analysis_original)), 0) |
| - def testTriageDuplicateResults(self): |
| + def testTriageDuplicateResultsFoundCorrectDuplicate(self): |
| # Yesterday, UTC Noon. |
| original_time = (datetime.utcnow() - timedelta(days=1)).replace( |
| hour=12, minute=0, second=0, microsecond=0) |
| @@ -456,10 +468,25 @@ class TriageAnalysisTest(testing.AppengineTestCase): |
| # Create another analysis at the same time (also yesterday). |
| self.createAnalysis(311, original_time) |
| - triage_analysis._TriageDuplicateResults(analysis_original, True) |
| + triage_analysis._TriageDuplicateResults(analysis_original, is_correct=True) |
| second_analysis = WfAnalysis.Get(self.master_name, self.builder_name, 311) |
| - self.assertEquals(result_status.NOT_FOUND_CORRECT, |
| + self.assertEquals(result_status.FOUND_CORRECT_DUPLICATE, |
| second_analysis.result_status) |
| + def testTriageDuplicateResultsFoundIncorrectDuplicate(self): |
| + # Yesterday, UTC Noon. |
| + original_time = (datetime.utcnow() - timedelta(days=1)).replace( |
| + hour=12, minute=0, second=0, microsecond=0) |
| + analysis_original = self.createAnalysis(312, original_time) |
| + |
| + # Create another analysis at the same time (also yesterday). |
| + self.createAnalysis(313, original_time) |
| + |
| + triage_analysis._TriageDuplicateResults(analysis_original, is_correct=False) |
| + |
| + second_analysis = WfAnalysis.Get(self.master_name, self.builder_name, 313) |
| + |
| + self.assertEquals(result_status.FOUND_INCORRECT_DUPLICATE, |
| + second_analysis.result_status) |