Committed PyCon sprint work.

1 files changed, 35 insertions, 0 deletions

diff --git a/functional_tests/test_buggy_generators.py b/functional_tests/test_buggy_generators.py
new file mode 100644
index 0000000..e710e4a
--- /dev/null
+++ b/functional_tests/test_buggy_generators.py
@@ -0,0 +1,35 @@
+import os
+import unittest
+from cStringIO import StringIO
+from nose.core import TestProgram
+from nose.config import Config
+
+here = os.path.dirname(__file__)
+support = os.path.join(here, 'support')
+
+
+class TestRunner(unittest.TextTestRunner):
+    def _makeResult(self):
+        self.result = unittest._TextTestResult(
+            self.stream, self.descriptions, self.verbosity)
+        return self.result
+
+    
+class TestBuggyGenerators(unittest.TestCase):
+    def test_run_buggy_generators(self):
+        stream = StringIO()
+        runner = TestRunner(stream=stream)
+        prog = TestProgram(
+            argv=['nosetests',
+                  os.path.join(support, 'test_buggy_generators.py')],
+            testRunner=runner,
+            config=Config(),
+            exit=False)
+        res = runner.result
+        print stream.getvalue()
+        self.assertEqual(res.testsRun, 12,
+                         "Expected to run 12 tests, ran %s" % res.testsRun)
+        assert not res.wasSuccessful()
+        assert len(res.errors) == 4
+        assert not res.failures
+