| Index: recipe_engine/simulation_test.py
|
| diff --git a/recipe_engine/simulation_test.py b/recipe_engine/simulation_test.py
|
| index 87c81ff8a0618d9e807033e40cd96390dd860bb3..81091a0a362f8a19f24cdd09110c20df6e2f3613 100644
|
| --- a/recipe_engine/simulation_test.py
|
| +++ b/recipe_engine/simulation_test.py
|
| @@ -4,8 +4,6 @@
|
|
|
| """Provides simulator test coverage for individual recipes."""
|
|
|
| -import StringIO
|
| -import contextlib
|
| import json
|
| import logging
|
| import os
|
| @@ -13,7 +11,6 @@
|
| import sys
|
|
|
| from . import env
|
| -from . import stream
|
| import expect_tests
|
|
|
| # This variable must be set in the dynamic scope of the functions in this file.
|
| @@ -58,24 +55,6 @@
|
| return expect_tests.Result(raw_expectations)
|
|
|
|
|
| -class SimulationAnnotatorStreamEngine(stream.AnnotatorStreamEngine):
|
| -
|
| - def __init__(self):
|
| - self._step_buffer_map = {}
|
| - super(SimulationAnnotatorStreamEngine, self).__init__(
|
| - self.step_buffer(None))
|
| -
|
| - def step_buffer(self, step_name):
|
| - return self._step_buffer_map.setdefault(step_name, StringIO.StringIO())
|
| -
|
| - def _new_step_stream(self, step_name, allow_subannotations, nest_level):
|
| - return self._create_step_stream(
|
| - step_name,
|
| - self.step_buffer(step_name),
|
| - allow_subannotations,
|
| - nest_level)
|
| -
|
| -
|
| def RunRecipe(test_data):
|
| """Actually runs the recipe given the GenTests-supplied test_data."""
|
| from . import config_types
|
| @@ -85,14 +64,8 @@
|
| from . import stream
|
|
|
| config_types.ResetTostringFns()
|
| -
|
| - annotator = SimulationAnnotatorStreamEngine()
|
| - stream_engine = stream.ProductStreamEngine(
|
| - stream.StreamEngineInvariants(),
|
| - annotator)
|
| - with stream_engine:
|
| - step_runner = step_runner.SimulationStepRunner(stream_engine, test_data,
|
| - annotator)
|
| + with stream.StreamEngineInvariants() as stream_engine:
|
| + step_runner = step_runner.SimulationStepRunner(stream_engine, test_data)
|
|
|
| engine = run.RecipeEngine(step_runner, test_data.properties, _UNIVERSE)
|
| recipe_script = _UNIVERSE.load_recipe(test_data.properties['recipe'])
|
|
|