| 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 from common.findit_testcase import FinditTestCase |
| 8 | |
| 9 from model import analysis_status | 8 from model import analysis_status |
| 10 from model.flake.flake_analysis_request import BuildStep | 9 from model.flake.flake_analysis_request import BuildStep |
| 11 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 |
| 12 | 12 |
| 13 | 13 |
| 14 class FlakeAnalysisRequestTest(unittest.TestCase): | 14 class FlakeAnalysisRequestTest(FinditTestCase): |
| 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 14 matching lines...) Expand all Loading... |
| 39 request = FlakeAnalysisRequest.Create('flaky_test', False, 123) | 39 request = FlakeAnalysisRequest.Create('flaky_test', False, 123) |
| 40 self.assertTrue(request.AddBuildStep('m', 'b1', 1, 's', t1)) | 40 self.assertTrue(request.AddBuildStep('m', 'b1', 1, 's', t1)) |
| 41 self.assertTrue(request.AddBuildStep('m', 'b2', 10, 's', t2)) | 41 self.assertTrue(request.AddBuildStep('m', 'b2', 10, 's', t2)) |
| 42 self.assertFalse(request.AddBuildStep('m', 'b2', 11, 's', t3)) | 42 self.assertFalse(request.AddBuildStep('m', 'b2', 11, 's', t3)) |
| 43 self.assertTrue(request.AddBuildStep('m', 'b2', 9, 's', t4)) | 43 self.assertTrue(request.AddBuildStep('m', 'b2', 9, 's', t4)) |
| 44 self.assertEqual(2, len(request.build_steps), request.build_steps) | 44 self.assertEqual(2, len(request.build_steps), request.build_steps) |
| 45 self.assertEqual(BuildStep.Create('m', 'b1', 1, 's', t1), | 45 self.assertEqual(BuildStep.Create('m', 'b1', 1, 's', t1), |
| 46 request.build_steps[0]) | 46 request.build_steps[0]) |
| 47 self.assertEqual(BuildStep.Create('m', 'b2', 9, 's', t4), | 47 self.assertEqual(BuildStep.Create('m', 'b2', 9, 's', t4), |
| 48 request.build_steps[1]) | 48 request.build_steps[1]) |
| 49 |
| 50 def testCopyFrom(self): |
| 51 request1 = FlakeAnalysisRequest.Create('flaky_test', False, 123) |
| 52 |
| 53 request2 = FlakeAnalysisRequest.Create('flaky_test', True, 456) |
| 54 request2.AddBuildStep('m', 'b1', 1, 's', datetime(2016, 10, 1)) |
| 55 request2.user_emails = ['email'] |
| 56 analysis = MasterFlakeAnalysis.Create('m', 'b', 100, 's', 't') |
| 57 analysis.Save() |
| 58 request2.analyses.append(analysis.key) |
| 59 |
| 60 request1.CopyFrom(request2) |
| 61 |
| 62 self.assertEqual(request2.is_step, request1.is_step) |
| 63 self.assertEqual(request2.bug_id, request1.bug_id) |
| 64 self.assertEqual(request2.user_emails, request1.user_emails) |
| 65 self.assertEqual(request2.build_steps, request1.build_steps) |
| 66 self.assertEqual(request2.analyses, request1.analyses) |
| OLD | NEW |