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

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

Issue 2491473002: [Findit] Implementing swarming task error detection (Closed)
Patch Set: Rebase Created 4 years, 1 month 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 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 model.flake.flake_swarming_task import FlakeSwarmingTask 5 from model.flake.flake_swarming_task import FlakeSwarmingTask
6 from waterfall.test import wf_testcase 6 from waterfall.test import wf_testcase
7 from waterfall.trigger_flake_swarming_task_pipeline import ( 7 from waterfall.trigger_flake_swarming_task_pipeline import (
8 TriggerFlakeSwarmingTaskPipeline) 8 TriggerFlakeSwarmingTaskPipeline)
9 9
10 10
11 class TriggerFlakeSwarmingTaskPipelineTest(wf_testcase.WaterfallTestCase): 11 class TriggerFlakeSwarmingTaskPipelineTest(wf_testcase.WaterfallTestCase):
12 12
13 def testGetArgs(self): 13 def testGetArgs(self):
14 master_name = 'm' 14 master_name = 'm'
15 builder_name = 'b' 15 builder_name = 'b'
16 build_number = 123 16 build_number = 123
17 step_name = 's' 17 step_name = 's'
18 tests = ['t'] 18 tests = ['t']
19
20 self.assertEqual( 19 self.assertEqual(
21 (master_name, builder_name, build_number, step_name, tests[0]), 20 (master_name, builder_name, build_number, step_name, tests[0]),
22 TriggerFlakeSwarmingTaskPipeline()._GetArgs( 21 TriggerFlakeSwarmingTaskPipeline()._GetArgs(
23 master_name, builder_name, build_number, step_name, tests)) 22 master_name, builder_name, build_number, step_name, tests))
24 23
25 def testGetSwarmingTask(self): 24 def testGetSwarmingTask(self):
26 master_name = 'm' 25 master_name = 'm'
27 builder_name = 'b' 26 builder_name = 'b'
28 build_number = 123 27 build_number = 123
29 step_name = 's' 28 step_name = 's'
30 test_name = 't' 29 test_name = 't'
31 30
32 FlakeSwarmingTask.Create( 31 FlakeSwarmingTask.Create(
33 master_name, builder_name, build_number, step_name, test_name).put() 32 master_name, builder_name, build_number, step_name, test_name).put()
34 33
35 task = TriggerFlakeSwarmingTaskPipeline()._GetSwarmingTask( 34 task = TriggerFlakeSwarmingTaskPipeline()._GetSwarmingTask(
36 master_name, builder_name, build_number, step_name, test_name) 35 master_name, builder_name, build_number, step_name, test_name)
37
38 self.assertEqual(master_name, task.master_name) 36 self.assertEqual(master_name, task.master_name)
39 self.assertEqual(builder_name, task.builder_name) 37 self.assertEqual(builder_name, task.builder_name)
40 self.assertEqual(build_number, task.build_number) 38 self.assertEqual(build_number, task.build_number)
41 self.assertEqual(step_name, task.step_name) 39 self.assertEqual(step_name, task.step_name)
42 self.assertEqual(test_name, task.test_name) 40 self.assertEqual(test_name, task.test_name)
43 41
44 def testCreateSwarmingTask(self): 42 def testCreateSwarmingTask(self):
45 master_name = 'm' 43 master_name = 'm'
46 builder_name = 'b' 44 builder_name = 'b'
47 build_number = 123 45 build_number = 123
48 step_name = 's' 46 step_name = 's'
49 test_name = 't' 47 test_name = 't'
50 48
51 task = TriggerFlakeSwarmingTaskPipeline()._CreateSwarmingTask( 49 task = TriggerFlakeSwarmingTaskPipeline()._CreateSwarmingTask(
52 master_name, builder_name, build_number, step_name, test_name) 50 master_name, builder_name, build_number, step_name, test_name)
53 self.assertEqual(master_name, task.master_name) 51 self.assertEqual(master_name, task.master_name)
54 self.assertEqual(builder_name, task.builder_name) 52 self.assertEqual(builder_name, task.builder_name)
55 self.assertEqual(build_number, task.build_number) 53 self.assertEqual(build_number, task.build_number)
56 self.assertEqual(step_name, task.step_name) 54 self.assertEqual(step_name, task.step_name)
57 self.assertEqual(test_name, task.test_name) 55 self.assertEqual(test_name, task.test_name)
58 56
59 def testGetIterationsToRerun(self): 57 def testGetIterationsToRerun(self):
60 expected_iterations = 50 58 expected_iterations = 50
61 self.UpdateUnitTestConfigSettings( 59 self.UpdateUnitTestConfigSettings(
62 config_property='check_flake_settings', 60 config_property='check_flake_settings',
63 override_data={'iterations_to_rerun': expected_iterations}) 61 override_data={'iterations_to_rerun': expected_iterations})
64 self.assertEqual( 62 self.assertEqual(
65 expected_iterations, 63 expected_iterations,
66 TriggerFlakeSwarmingTaskPipeline()._GetIterationsToRerun()) 64 TriggerFlakeSwarmingTaskPipeline()._GetIterationsToRerun())
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698