| OLD | NEW |
| 1 # Copyright 2016 The Chromium Authors. All rights reserved. | 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 | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 from datetime import datetime | 5 from datetime import datetime |
| 6 | 6 |
| 7 import unittest | 7 import unittest |
| 8 | 8 |
| 9 from model import analysis_status | 9 from model import analysis_status |
| 10 from model.flake.master_flake_analysis import DataPoint |
| 10 from model.flake.master_flake_analysis import MasterFlakeAnalysis | 11 from model.flake.master_flake_analysis import MasterFlakeAnalysis |
| 11 | 12 |
| 12 | 13 |
| 13 class MasterFlakeAnalysisTest(unittest.TestCase): | 14 class MasterFlakeAnalysisTest(unittest.TestCase): |
| 14 | 15 |
| 15 def testMasterFlakeAnalysisStatusIsCompleted(self): | 16 def testMasterFlakeAnalysisStatusIsCompleted(self): |
| 16 for status in (analysis_status.COMPLETED, analysis_status.ERROR): | 17 for status in (analysis_status.COMPLETED, analysis_status.ERROR): |
| 17 analysis = MasterFlakeAnalysis.Create('m', 'b', 123, 's', 't') | 18 analysis = MasterFlakeAnalysis.Create('m', 'b', 123, 's', 't') |
| 18 analysis.status = status | 19 analysis.status = status |
| 19 self.assertTrue(analysis.completed) | 20 self.assertTrue(analysis.completed) |
| (...skipping 57 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 77 | 78 |
| 78 def testMasterFlakeAnalysisStatusDescriptionError(self): | 79 def testMasterFlakeAnalysisStatusDescriptionError(self): |
| 79 analysis = MasterFlakeAnalysis.Create('m', 'b', 123, 's', 't') | 80 analysis = MasterFlakeAnalysis.Create('m', 'b', 123, 's', 't') |
| 80 analysis.status = analysis_status.ERROR | 81 analysis.status = analysis_status.ERROR |
| 81 self.assertEqual('Error', analysis.status_description) | 82 self.assertEqual('Error', analysis.status_description) |
| 82 | 83 |
| 83 def testMasterFlakeAnalysisStepTestName(self): | 84 def testMasterFlakeAnalysisStepTestName(self): |
| 84 analysis = MasterFlakeAnalysis.Create('m', 'b', 123, 's', 't') | 85 analysis = MasterFlakeAnalysis.Create('m', 'b', 123, 's', 't') |
| 85 self.assertEqual('s', analysis.step_name) | 86 self.assertEqual('s', analysis.step_name) |
| 86 self.assertEqual('t', analysis.test_name) | 87 self.assertEqual('t', analysis.test_name) |
| 88 |
| 89 def testReset(self): |
| 90 analysis = MasterFlakeAnalysis.Create('m', 'b', 123, 's', 't') |
| 91 analysis.swarming_rerun_results = [{}] |
| 92 analysis.status = analysis_status.RUNNING |
| 93 analysis.correct_regression_range = True |
| 94 analysis.correct_culprit = False |
| 95 analysis.correct_culprit = None |
| 96 analysis.data_points = [DataPoint()] |
| 97 analysis.suspected_flake_build_number = 123 |
| 98 analysis.Reset() |
| 99 |
| 100 self.assertEqual([], analysis.swarming_rerun_results) |
| 101 self.assertEqual(analysis_status.PENDING, analysis.status) |
| 102 self.assertIsNone(analysis.correct_regression_range) |
| 103 self.assertIsNone(analysis.correct_culprit) |
| 104 self.assertIsNone(analysis.completed_time) |
| 105 self.assertIsNone(analysis.suspected_flake_build_number) |
| 106 self.assertEqual([], analysis.data_points) |
| OLD | NEW |