OLD | NEW |
(Empty) | |
| 1 #!/usr/bin/env python |
| 2 # Copyright 2015 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. |
| 5 |
| 6 import os |
| 7 import sys |
| 8 import unittest |
| 9 |
| 10 sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname( |
| 11 os.path.abspath(__file__))))) |
| 12 |
| 13 from recipe_engine import loader, recipe_api, config |
| 14 |
| 15 class TestProperties(unittest.TestCase): |
| 16 def _makeProp(self, *args, **kwargs): |
| 17 return recipe_api.Property(*args, **kwargs) |
| 18 |
| 19 def testDefault(self): |
| 20 """Tests the default option of properties.""" |
| 21 for default in (1, object(), "test", None): |
| 22 prop = self._makeProp(default=default) |
| 23 self.assertEqual(default, prop.interpret(recipe_api.Property.sentinel)) |
| 24 |
| 25 def testRequired(self): |
| 26 """Tests that a required property errors when not provided.""" |
| 27 prop = self._makeProp() |
| 28 with self.assertRaises(ValueError): |
| 29 prop.interpret(recipe_api.Property.sentinel) |
| 30 |
| 31 def testTypeSingle(self): |
| 32 """Tests a simple typed property.""" |
| 33 prop = self._makeProp(kind=bool) |
| 34 with self.assertRaises(TypeError): |
| 35 prop.interpret(1) |
| 36 |
| 37 self.assertEqual(True, prop.interpret(True)) |
| 38 |
| 39 def testTypeFancy(self): |
| 40 """Tests a config style type property.""" |
| 41 prop = self._makeProp(kind=config.List(int)) |
| 42 for value in (1, "hi", [3, "test"]): |
| 43 with self.assertRaises(TypeError): |
| 44 prop.interpret(value) |
| 45 |
| 46 self.assertEqual([2, 3], prop.interpret([2, 3])) |
| 47 |
| 48 class TestInvoke(unittest.TestCase): |
| 49 def invoke(self, callable, all_properties, prop_defs, **kwargs): |
| 50 return loader.invoke_with_properties( |
| 51 callable, all_properties, prop_defs, **kwargs) |
| 52 |
| 53 def testInvokeFuncSimple(self): |
| 54 """Simple test of invoke.""" |
| 55 def func(): |
| 56 pass |
| 57 |
| 58 self.assertEqual(self.invoke(func, {}, {}), None) |
| 59 |
| 60 def testInvokeFuncComplex(self): |
| 61 """Tests invoke with two different properties.""" |
| 62 def func(a, b): |
| 63 return a |
| 64 |
| 65 prop_defs = { |
| 66 'a': recipe_api.Property(), |
| 67 'b': recipe_api.Property(), |
| 68 } |
| 69 |
| 70 props = { |
| 71 'a': 1, |
| 72 'b': 2, |
| 73 } |
| 74 self.assertEqual(1, self.invoke(func, props, prop_defs)) |
| 75 |
| 76 def testInvokeClass(self): |
| 77 """Tests invoking a class.""" |
| 78 class test(object): |
| 79 def __init__(self, a, b): |
| 80 self.answer = a |
| 81 |
| 82 prop_defs = { |
| 83 'a': recipe_api.Property(), |
| 84 'b': recipe_api.Property(), |
| 85 } |
| 86 |
| 87 props = { |
| 88 'a': 1, |
| 89 'b': 2, |
| 90 } |
| 91 self.assertEqual(1, self.invoke(test, props, prop_defs).answer) |
| 92 |
| 93 def testMissingProperty(self): |
| 94 """Tests that invoke raises an error when missing a property.""" |
| 95 def func(a): |
| 96 return a |
| 97 |
| 98 with self.assertRaises(recipe_api.UndefinedPropertyException): |
| 99 self.invoke(func, {}, {}) |
| 100 |
| 101 if __name__ == '__main__': |
| 102 unittest.main() |
OLD | NEW |