OLD | NEW |
1 import json | 1 import json |
2 | 2 |
3 from recipe_engine import recipe_test_api | 3 from recipe_engine import recipe_test_api |
4 | 4 |
5 class JsonTestApi(recipe_test_api.RecipeTestApi): | 5 class JsonTestApi(recipe_test_api.RecipeTestApi): |
6 @recipe_test_api.placeholder_step_data | 6 @recipe_test_api.placeholder_step_data |
7 @staticmethod | 7 @staticmethod |
8 def output(data, retcode=None): | 8 def output(data, retcode=None): |
9 return json.dumps(data), retcode | 9 return json.dumps(data), retcode |
10 | 10 |
11 def output_stream(self, data, stream='stdout', retcode=None): | 11 def output_stream(self, data, stream='stdout', retcode=None): |
12 assert stream in ('stdout', 'stderr') | 12 assert stream in ('stdout', 'stderr') |
13 ret = recipe_test_api.StepTestData() | 13 ret = recipe_test_api.StepTestData() |
14 setattr(ret, stream, self.output(data, retcode).unwrap_placeholder()) | 14 setattr(ret, stream, self.output(data, retcode).unwrap_placeholder()) |
15 return ret | 15 return ret |
OLD | NEW |