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

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

Issue 2203643002: [Findit] Use new mock for unittests. (Closed) Base URL: https://chromium.googlesource.com/infra/infra.git@0708-pipeline-refactor
Patch Set: rebase Created 4 years, 4 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
1 # Copyright 2015 The Chromium Authors. All rights reserved. 1 # Copyright 2015 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 import mock 5 import mock
6 6
7 from common.waterfall import buildbucket_client 7 from common.waterfall import buildbucket_client
8 from common.waterfall import failure_type 8 from common.waterfall import failure_type
9 from model.wf_try_job import WfTryJob 9 from model.wf_try_job import WfTryJob
10 from model.wf_try_job_data import WfTryJobData 10 from model.wf_try_job_data import WfTryJobData
(...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after
51 '/%s/builds/%s') % ( 51 '/%s/builds/%s') % (
52 master_name, builder_name, build_number) 52 master_name, builder_name, build_number)
53 } 53 }
54 try_job_pipeline = ScheduleTryJobPipeline() 54 try_job_pipeline = ScheduleTryJobPipeline()
55 properties = try_job_pipeline._GetBuildProperties( 55 properties = try_job_pipeline._GetBuildProperties(
56 master_name, builder_name, build_number, 1, 2, failure_type.TEST, None) 56 master_name, builder_name, build_number, 1, 2, failure_type.TEST, None)
57 57
58 self.assertEqual(properties, expected_properties) 58 self.assertEqual(properties, expected_properties)
59 59
60 @mock.patch.object(schedule_try_job_pipeline, 'buildbucket_client') 60 @mock.patch.object(schedule_try_job_pipeline, 'buildbucket_client')
61 def testTriggerTryJob(self, mock_other): 61 def testTriggerTryJob(self, mock_module):
62 master_name = 'm' 62 master_name = 'm'
63 builder_name = 'b' 63 builder_name = 'b'
64 response = { 64 response = {
65 'build': { 65 'build': {
66 'id': '1', 66 'id': '1',
67 'url': 'url', 67 'url': 'url',
68 'status': 'SCHEDULED', 68 'status': 'SCHEDULED',
69 } 69 }
70 } 70 }
71 results = [(None, buildbucket_client.BuildbucketBuild(response['build']))] 71 results = [(None, buildbucket_client.BuildbucketBuild(response['build']))]
72 mock_other.TriggerTryJobs.return_value = results 72 mock_module.TriggerTryJobs.return_value = results
73 73
74 try_job_pipeline = ScheduleTryJobPipeline() 74 try_job_pipeline = ScheduleTryJobPipeline()
75 build_id = try_job_pipeline._TriggerTryJob( 75 build_id = try_job_pipeline._TriggerTryJob(
76 master_name, builder_name, {}, []) 76 master_name, builder_name, {}, [])
77 77
78 self.assertEqual(build_id, '1') 78 self.assertEqual(build_id, '1')
79 79
80 80
81 def testCreateTryJobData(self): 81 def testCreateTryJobData(self):
82 build_id = 1 82 build_id = 1
83 master_name = 'm' 83 master_name = 'm'
84 builder_name = 'b' 84 builder_name = 'b'
85 build_number = 1 85 build_number = 1
86 try_job_type = failure_type.GetDescriptionForFailureType(failure_type.TEST) 86 try_job_type = failure_type.GetDescriptionForFailureType(failure_type.TEST)
87 87
88 ScheduleTryJobPipeline()._CreateTryJobData( 88 ScheduleTryJobPipeline()._CreateTryJobData(
89 build_id, master_name, builder_name, build_number, try_job_type, 89 build_id, master_name, builder_name, build_number, try_job_type,
90 False, False) 90 False, False)
91 91
92 data = WfTryJobData.Get(build_id) 92 data = WfTryJobData.Get(build_id)
93 93
94 self.assertIsNotNone(data) 94 self.assertIsNotNone(data)
95 self.assertEqual(data.master_name, master_name) 95 self.assertEqual(data.master_name, master_name)
96 self.assertEqual(data.builder_name, builder_name) 96 self.assertEqual(data.builder_name, builder_name)
97 self.assertEqual(data.build_number, build_number) 97 self.assertEqual(data.build_number, build_number)
98 self.assertEqual(data.try_job_type, try_job_type) 98 self.assertEqual(data.try_job_type, try_job_type)
99 self.assertFalse(data.has_compile_targets) 99 self.assertFalse(data.has_compile_targets)
100 self.assertFalse(data.has_heuristic_results) 100 self.assertFalse(data.has_heuristic_results)
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698