Index: scripts/slave/recipe_modules/raw_io/test_api.py |
diff --git a/scripts/slave/recipe_modules/raw_io/test_api.py b/scripts/slave/recipe_modules/raw_io/test_api.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..a0e3f4178d719b1f63621e402cc3cd7faea20eac |
--- /dev/null |
+++ b/scripts/slave/recipe_modules/raw_io/test_api.py |
@@ -0,0 +1,21 @@ |
+from recipe_engine import recipe_test_api |
+ |
+class RawIOTestApi(recipe_test_api.RecipeTestApi): # pragma: no cover |
+ @recipe_test_api.placeholder_step_data |
+ @staticmethod |
+ def output(data, retcode=None): |
+ return data, retcode |
+ |
+ @recipe_test_api.placeholder_step_data |
+ @staticmethod |
+ def output_dir(files_dict, retcode=None): |
+ assert type(files_dict) is dict |
+ assert all(type(key) is str for key in files_dict.keys()) |
+ assert all(type(value) is str for value in files_dict.values()) |
+ return files_dict, retcode |
+ |
+ def stream_output(self, data, stream='stdout', retcode=None): |
+ ret = recipe_test_api.StepTestData() |
+ assert stream in ('stdout', 'stderr') |
+ setattr(ret, stream, self.output(data, retcode).unwrap_placeholder()) |
+ return ret |