| # Copyright 2015 The Chromium OS Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """Unittests for main builder logic (__init__.py).""" |
| |
| from __future__ import print_function |
| |
| from chromite.cbuildbot import builders |
| from chromite.lib import cros_test_lib |
| |
| |
| class ModuleTest(cros_test_lib.MockTempDirTestCase): |
| """Module loading related tests""" |
| |
| def testGetBuilderClassNone(self): |
| """Check behavior when requesting missing builders.""" |
| self.assertRaises(ValueError, builders.GetBuilderClass, 'Foalksdjo') |
| self.assertRaises(ImportError, builders.GetBuilderClass, 'foo.Foalksdjo') |
| self.assertRaises(AttributeError, builders.GetBuilderClass, |
| 'misc_builders.Foalksdjo') |
