[tor-commits] [stem/master] Mocking stdout for tutorial tests

[atagar at torproject.org]

commit 1749855559044f2547920fa206f72c6fdf7c86f4
Author: Damian Johnson <atagar at torproject.org>
Date:   Fri Feb 22 08:53:21 2013 -0800

    Mocking stdout for tutorial tests
    
    Mocking stdout so the tests can be a little more faithful to the tutorial
    examples.
---
 test/unit/tutorial.py |   21 +++++++++++++++------
 1 files changed, 15 insertions(+), 6 deletions(-)

diff --git a/test/unit/tutorial.py b/test/unit/tutorial.py
index 5880f53..21db41b 100644
--- a/test/unit/tutorial.py
+++ b/test/unit/tutorial.py
@@ -4,14 +4,23 @@ Tests for the examples given in stem's tutorial.
 
 from __future__ import with_statement
 
+import StringIO
+import sys
 import unittest
 
 from test import mocking
 
 
 class TestTutorial(unittest.TestCase):
+  stdout, stdout_real = None, None
+
+  def setUp(self):
+    self.stdout, self.stdout_real = StringIO.StringIO(), sys.stdout
+    sys.stdout = self.stdout
+
   def tearDown(self):
     mocking.revert_mocking()
+    sys.stdout = self.stdout_real
 
   def test_the_little_relay_that_could(self):
     from stem.control import Controller
@@ -30,12 +39,12 @@ class TestTutorial(unittest.TestCase):
     bytes_read = controller.get_info("traffic/read")
     bytes_written = controller.get_info("traffic/written")
 
-    expected_line = "My Tor relay has read 1234 bytes and written 5678."
-    printed_line = "My Tor relay has read %s bytes and written %s." % (bytes_read, bytes_written)
-    self.assertEqual(expected_line, printed_line)
+    print "My Tor relay has read %s bytes and written %s." % (bytes_read, bytes_written)
 
     controller.close()
 
+    self.assertEqual("My Tor relay has read 1234 bytes and written 5678.\n", self.stdout.getvalue())
+
   def test_mirror_mirror_on_the_wall(self):
     from stem.descriptor.server_descriptor import RelayDescriptor
     from stem.descriptor.reader import DescriptorReader
@@ -76,13 +85,13 @@ class TestTutorial(unittest.TestCase):
 
     for bw_value in sorted(bw_to_relay.keys(), reverse = True):
       for nickname in bw_to_relay[bw_value]:
-        expected_line = "%i. speedyexit (102.13 KB/s)" % count
-        printed_line = "%i. %s (%s/s)" % (count, nickname, str_tools.get_size_label(bw_value, 2))
-        self.assertEqual(expected_line, printed_line)
+        print "%i. %s (%s/s)" % (count, nickname, str_tools.get_size_label(bw_value, 2))
 
         count += 1
 
         if count > 15:
           return
 
+    expected_stdout = "".join(["%i. speedyexit (102.13 KB/s)\n" % i for i in xrange(1, 4)])
+    self.assertEqual(expected_stdout, self.stdout.getvalue())
     self.assertEqual(4, count)