Emit CURRENT_TIMESTAMP annotation

recipe_engine/simulation_test.py

 # Copyright 2014 The LUCI Authors. All rights reserved.
 # Use of this source code is governed under the Apache License, Version 2.0
 # that can be found in the LICENSE file.
 
 """Provides simulator test coverage for individual recipes."""
 
 import logging
 import re
 import os
 import sys
 
 from .third_party import expect_tests
 
 # This variable must be set in the dynamic scope of the functions in this file.
 # We do this instead of passing because the threading system of expect tests
 # doesn't know how to serialize it.
 _UNIVERSE = None
 
 def RunRecipe(test_data):
   from . import config_types
   from . import loader
   from . import run
   from . import step_runner
   from . import stream
 
   config_types.ResetTostringFns()
-  stream_engine = stream.StreamEngineInvariants()
-  step_runner = step_runner.SimulationStepRunner(stream_engine, test_data)
+  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'])
-  api = loader.create_recipe_api(recipe_script.LOADED_DEPS, engine, test_data)
-  result = engine.run(recipe_script, api)
+    engine = run.RecipeEngine(step_runner, test_data.properties, _UNIVERSE)
+    recipe_script = _UNIVERSE.load_recipe(test_data.properties['recipe'])
+    api = loader.create_recipe_api(recipe_script.LOADED_DEPS, engine, test_data)
+    result = engine.run(recipe_script, api)
 
-  # Don't include tracebacks in expectations because they are too sensitive to
-  # change.
-  result.result.pop('traceback', None)
+    # Don't include tracebacks in expectations because they are too sensitive to
+    # change.
+    result.result.pop('traceback', None)
 
-  return expect_tests.Result(step_runner.steps_ran + [result.result])
+    return expect_tests.Result(step_runner.steps_ran + [result.result])
 
 
 def test_gen_coverage():
   cover = []
 
   for path in _UNIVERSE.recipe_dirs:
     cover.append(os.path.join(path, '*'))
 
   for path in _UNIVERSE.module_dirs:
     cover.append(os.path.join(path, '*', 'example.py'))
@@ skipping 60 matching lines @@
                   'TESTING_SLAVENAME']:
     if env_var in os.environ:
       logging.warn("Ignoring %s environment variable." % env_var)
       os.environ.pop(env_var)
 
   global _UNIVERSE
   _UNIVERSE = universe
 
   expect_tests.main('recipe_simulation_test', GenerateTests,
                     cover_omit=cover_omit(), args=args)