| Index: appengine/findit/waterfall/test/waterfall_config_test.py
|
| diff --git a/appengine/findit/waterfall/test/waterfall_config_test.py b/appengine/findit/waterfall/test/waterfall_config_test.py
|
| index cdb27bbf9272fa4c7fde8d34a94ae31150b7ddef..49da9dd30af393728bf78543e71b15a391700526 100644
|
| --- a/appengine/findit/waterfall/test/waterfall_config_test.py
|
| +++ b/appengine/findit/waterfall/test/waterfall_config_test.py
|
| @@ -4,7 +4,7 @@
|
|
|
| from testing_utils import testing
|
|
|
| -from model import wf_config
|
| +from model.wf_config import FinditConfig
|
| from waterfall import waterfall_config
|
|
|
|
|
| @@ -13,22 +13,25 @@ class MastersTest(testing.AppengineTestCase):
|
|
|
| def setUp(self):
|
| super(MastersTest, self).setUp()
|
| + self.mock_current_user(user_email='test@chromium.org', is_admin=True)
|
| +
|
| + config_data = {
|
| + 'masters_to_blacklisted_steps': {
|
| + 'master1': ['step1', 'step2', 'step3'],
|
| + 'master2': ['step4', 'step5', 'step6']
|
| + },
|
| + 'builders_to_trybots': {
|
| + 'master1': {
|
| + 'builder1': {
|
| + 'mastername': 'tryserver1',
|
| + 'buildername': 'trybot1',
|
| + }
|
| + }
|
| + }
|
| + }
|
| +
|
| + FinditConfig.Get().Update(**config_data)
|
|
|
| - class MockSettings():
|
| - masters_to_blacklisted_steps = {
|
| - 'master1': ['step1', 'step2', 'step3'],
|
| - 'master2': ['step4', 'step5', 'step6']
|
| - }
|
| - builders_to_trybots = {
|
| - 'master1': {
|
| - 'builder1': {
|
| - 'mastername': 'tryserver1',
|
| - 'buildername': 'trybot1',
|
| - }
|
| - }
|
| - }
|
| -
|
| - self.mock(wf_config, 'Settings', MockSettings)
|
|
|
| def testMasterIsSupported(self):
|
| self.assertTrue(waterfall_config.MasterIsSupported('master1'))
|
|
|