Index: scripts/slave/recipe_modules/generator_script/test_api.py |
diff --git a/scripts/slave/recipe_modules/generator_script/test_api.py b/scripts/slave/recipe_modules/generator_script/test_api.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..bdcfde13ed0cb69e1f7eca8c771a65852dd2add4 |
--- /dev/null |
+++ b/scripts/slave/recipe_modules/generator_script/test_api.py |
@@ -0,0 +1,9 @@ |
+from recipe_engine import recipe_test_api |
+ |
+class GeneratorScriptTestApi(recipe_test_api.RecipeTestApi): |
+ def __call__(self, script_name, *steps): |
+ assert all(isinstance(s, dict) for s in steps) |
+ return self.step_data( |
+ 'gen step(%s)' % script_name, |
+ self.m.json.output(list(steps)) |
+ ) |