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

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

Issue 1497783003: [Findit] Use Findit's own versioned config. (Closed) Base URL: https://chromium.googlesource.com/infra/infra.git@versioned_model
Patch Set: Created 5 years 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
« no previous file with comments | « appengine/findit/model/wf_config.py ('k') | appengine/findit/waterfall/waterfall_config.py » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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 from testing_utils import testing 5 from testing_utils import testing
6 6
7 from model import wf_config 7 from model.wf_config import FinditConfig
8 from waterfall import waterfall_config 8 from waterfall import waterfall_config
9 9
10 10
11 11
12 class MastersTest(testing.AppengineTestCase): 12 class MastersTest(testing.AppengineTestCase):
13 13
14 def setUp(self): 14 def setUp(self):
15 super(MastersTest, self).setUp() 15 super(MastersTest, self).setUp()
16 self.mock_current_user(user_email='test@chromium.org', is_admin=True)
16 17
17 class MockSettings(): 18 config_data = {
18 masters_to_blacklisted_steps = { 19 'masters_to_blacklisted_steps': {
19 'master1': ['step1', 'step2', 'step3'], 20 'master1': ['step1', 'step2', 'step3'],
20 'master2': ['step4', 'step5', 'step6'] 21 'master2': ['step4', 'step5', 'step6']
21 } 22 },
22 builders_to_trybots = { 23 'builders_to_trybots': {
23 'master1': { 24 'master1': {
24 'builder1': { 25 'builder1': {
25 'mastername': 'tryserver1', 26 'mastername': 'tryserver1',
26 'buildername': 'trybot1', 27 'buildername': 'trybot1',
27 } 28 }
28 } 29 }
29 } 30 }
31 }
30 32
31 self.mock(wf_config, 'Settings', MockSettings) 33 FinditConfig.Get().Update(**config_data)
34
32 35
33 def testMasterIsSupported(self): 36 def testMasterIsSupported(self):
34 self.assertTrue(waterfall_config.MasterIsSupported('master1')) 37 self.assertTrue(waterfall_config.MasterIsSupported('master1'))
35 self.assertFalse(waterfall_config.MasterIsSupported('blabla')) 38 self.assertFalse(waterfall_config.MasterIsSupported('blabla'))
36 39
37 def testStepIsSupportedForMaster(self): 40 def testStepIsSupportedForMaster(self):
38 self.assertFalse( 41 self.assertFalse(
39 waterfall_config.StepIsSupportedForMaster('step1', 'master1')) 42 waterfall_config.StepIsSupportedForMaster('step1', 'master1'))
40 self.assertTrue( 43 self.assertTrue(
41 waterfall_config.StepIsSupportedForMaster('step4', 'master1')) 44 waterfall_config.StepIsSupportedForMaster('step4', 'master1'))
42 self.assertFalse( 45 self.assertFalse(
43 waterfall_config.StepIsSupportedForMaster('blabla', 'blabla')) 46 waterfall_config.StepIsSupportedForMaster('blabla', 'blabla'))
44 47
45 def testGetTrybotForWaterfallBuilder(self): 48 def testGetTrybotForWaterfallBuilder(self):
46 self.assertEqual( 49 self.assertEqual(
47 ('tryserver1', 'trybot1'), 50 ('tryserver1', 'trybot1'),
48 waterfall_config.GetTrybotForWaterfallBuilder('master1', 'builder1')) 51 waterfall_config.GetTrybotForWaterfallBuilder('master1', 'builder1'))
49 self.assertEqual( 52 self.assertEqual(
50 (None, None), 53 (None, None),
51 waterfall_config.GetTrybotForWaterfallBuilder('master2', 'builder2')) 54 waterfall_config.GetTrybotForWaterfallBuilder('master2', 'builder2'))
OLDNEW
« no previous file with comments | « appengine/findit/model/wf_config.py ('k') | appengine/findit/waterfall/waterfall_config.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698