| 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 import mock | 6 import mock |
| 7 | 7 |
| 8 from common.findit_testcase import FinditTestCase | 8 from libs.testcase import TestCase |
| 9 from model.flake.flake_analysis_request import BuildStep | 9 from model.flake.flake_analysis_request import BuildStep |
| 10 from model.flake.flake_analysis_request import FlakeAnalysisRequest | 10 from model.flake.flake_analysis_request import FlakeAnalysisRequest |
| 11 from model.flake.master_flake_analysis import MasterFlakeAnalysis | 11 from model.flake.master_flake_analysis import MasterFlakeAnalysis |
| 12 | 12 |
| 13 | 13 |
| 14 class FlakeAnalysisRequestTest(FinditTestCase): | 14 class FlakeAnalysisRequestTest(TestCase): |
| 15 | 15 |
| 16 def testStripMasterPrefix(self): | 16 def testStripMasterPrefix(self): |
| 17 cases = { | 17 cases = { |
| 18 'master.tryserver.chromium.linux': 'tryserver.chromium.linux', | 18 'master.tryserver.chromium.linux': 'tryserver.chromium.linux', |
| 19 'chromium.linux': 'chromium.linux', | 19 'chromium.linux': 'chromium.linux', |
| 20 } | 20 } |
| 21 for original_name, expected_name in cases.iteritems(): | 21 for original_name, expected_name in cases.iteritems(): |
| 22 self.assertEqual(expected_name, | 22 self.assertEqual(expected_name, |
| 23 BuildStep._StripMasterPrefix(original_name)) | 23 BuildStep._StripMasterPrefix(original_name)) |
| 24 | 24 |
| (...skipping 102 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 127 def testFindMatchingAnalysisForConfiguration(self, _): | 127 def testFindMatchingAnalysisForConfiguration(self, _): |
| 128 request = FlakeAnalysisRequest.Create('test', False, 123) | 128 request = FlakeAnalysisRequest.Create('test', False, 123) |
| 129 analysis = MasterFlakeAnalysis.Create('m', 'b', 123, 's', 'test') | 129 analysis = MasterFlakeAnalysis.Create('m', 'b', 123, 's', 'test') |
| 130 analysis.Save() | 130 analysis.Save() |
| 131 request.analyses.append(analysis.key) | 131 request.analyses.append(analysis.key) |
| 132 request.Save() | 132 request.Save() |
| 133 | 133 |
| 134 self.assertEqual( | 134 self.assertEqual( |
| 135 analysis, | 135 analysis, |
| 136 request.FindMatchingAnalysisForConfiguration('m', 'b')) | 136 request.FindMatchingAnalysisForConfiguration('m', 'b')) |
| OLD | NEW |