Add LogDog / annotation protobuf support.

recipe_engine/unittests/stream_test.py

 #!/usr/bin/env python
 # Copyright 2015 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.
 
 import cStringIO
 import os
 import sys
 import unittest
 
@@ skipping 102 matching lines @@
     self.assertEqual(
         stringio.getvalue().splitlines(),
         self._example_annotations().splitlines())
 
   def test_example_wellformed(self):
     with stream.StreamEngineInvariants() as engine:
       self._example(engine)
 
   def test_product_with_invariants_on_example(self):
     stringio = cStringIO.StringIO()
-    engine = stream.ProductStreamEngine(
-        stream.StreamEngineInvariants(),
+    engine = stream.StreamEngineInvariants.wrap(
         stream.AnnotatorStreamEngine(
             stringio, emit_timestamps=True, time_fn=self.fake_time))
     with engine:
       self._example(engine)
     self.assertEqual(stringio.getvalue(), self._example_annotations())
 
   def test_noop(self):
     with stream.NoopStreamEngine() as engine:
       self._example(engine)
 
@@ skipping 25 matching lines @@
 
   def test_buildbot_status_constraint(self):
     with stream.StreamEngineInvariants() as engine:
       foo = engine.make_step_stream('foo')
       foo.set_step_status('FAILURE')
       with self.assertRaises(AssertionError):
         foo.set_step_status('SUCCESS')
 
 if __name__ == '__main__':
   unittest.main()