OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright 2015 The LUCI Authors. All rights reserved. | 2 # Copyright 2015 The LUCI Authors. All rights reserved. |
3 # Use of this source code is governed under the Apache License, Version 2.0 | 3 # Use of this source code is governed under the Apache License, Version 2.0 |
4 # that can be found in the LICENSE file. | 4 # that can be found in the LICENSE file. |
5 | 5 |
6 import os | |
7 import sys | |
8 import unittest | 6 import unittest |
9 | 7 |
10 import test_env | 8 import test_env |
11 | 9 |
12 from recipe_engine import loader, recipe_api, config | 10 from recipe_engine import loader, recipe_api, config |
13 import mock | 11 import mock |
14 | 12 |
15 | 13 |
16 class TestRecipeScript(unittest.TestCase): | 14 class TestRecipeScript(unittest.TestCase): |
17 def testReturnSchemaHasValidClass(self): | 15 def testReturnSchemaHasValidClass(self): |
18 with self.assertRaises(ValueError): | 16 with self.assertRaises(ValueError): |
19 script = loader.RecipeScript({'RETURN_SCHEMA': 3}, 'test_script') | 17 loader.RecipeScript({'RETURN_SCHEMA': 3}, 'test_script') |
20 | 18 |
21 def testRunChecksReturnType(self): | 19 def testRunChecksReturnType(self): |
22 sentinel = object() | 20 sentinel = object() |
23 mocked_return = object() | 21 mocked_return = object() |
24 class FakeReturn(object): | 22 class FakeReturn(object): |
25 def as_jsonish(_self, hidden=sentinel): | 23 def as_jsonish(_self, hidden=sentinel): |
26 self.assertEqual(True, hidden) | 24 self.assertEqual(True, hidden) |
27 | 25 |
28 return mocked_return | 26 return mocked_return |
29 | 27 |
(...skipping 103 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
133 | 131 |
134 with mock.patch( | 132 with mock.patch( |
135 'recipe_engine.loader._invoke_with_properties') as mocked_invoke: | 133 'recipe_engine.loader._invoke_with_properties') as mocked_invoke: |
136 loader.invoke_with_properties(TestClass, None, None) | 134 loader.invoke_with_properties(TestClass, None, None) |
137 args, _ = mocked_invoke.call_args | 135 args, _ = mocked_invoke.call_args |
138 self.assertTrue(['api', 'foo', 'bar'] in args) | 136 self.assertTrue(['api', 'foo', 'bar'] in args) |
139 | 137 |
140 | 138 |
141 if __name__ == '__main__': | 139 if __name__ == '__main__': |
142 unittest.main() | 140 unittest.main() |
OLD | NEW |