[tor-commits] [stem/master] Test collector_caching example

[atagar at torproject.org]

commit 4bfaa5b7cdba1a533f7241a28e0e4286e1be081e
Author: Damian Johnson <atagar at torproject.org>
Date:   Sat Sep 26 15:14:57 2020 -0700

    Test collector_caching example
---
 test/unit/examples.py | 22 ++++++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

diff --git a/test/unit/examples.py b/test/unit/examples.py
index 521a7937..7c801ece 100644
--- a/test/unit/examples.py
+++ b/test/unit/examples.py
@@ -19,6 +19,7 @@ from stem.descriptor.bandwidth_file import BandwidthFile
 from stem.descriptor.extrainfo_descriptor import RelayExtraInfoDescriptor
 from stem.descriptor.router_status_entry import RouterStatusEntryV3
 from stem.descriptor.server_descriptor import RelayDescriptor
+from stem.exit_policy import ExitPolicy
 from stem.response import ControlMessage
 from unittest.mock import Mock, patch
 
@@ -97,6 +98,12 @@ EXPECTED_COLLECTOR_CACHING = """\
   flubber (5C2124E6C5DD75C3C17C03EEA5A51812773DE671)
 """
 
+EXPECTED_COLLECTOR_READING = """\
+1 relays published an exiting policy today...
+
+  caerSidi (4F0C867DF0EF68160568C826838F482CEA7CFE44)
+"""
+
 
 class TestExamples(unittest.TestCase):
   def setUp(self):
@@ -256,8 +263,19 @@ class TestExamples(unittest.TestCase):
 
     self.assertEqual(EXPECTED_COLLECTOR_CACHING, stdout_mock.getvalue())
 
-  def test_collector_reading(self):
-    pass
+  @patch('stem.descriptor.collector.CollecTor.get_server_descriptors')
+  @patch('sys.stdout', new_callable = io.StringIO)
+  def test_collector_reading(self, stdout_mock, server_desc_mock):
+    server_desc_mock.return_value = [
+      RelayDescriptor.create({
+        'router': 'caerSidi 71.35.133.197 9001 0 0',
+        'fingerprint': '4F0C 867D F0EF 6816 0568 C826 838F 482C EA7C FE44',
+      }, exit_policy = ExitPolicy('accept *:*')),
+    ]
+
+    import collector_reading
+
+    self.assertEqual(EXPECTED_COLLECTOR_READING, stdout_mock.getvalue())
 
   def test_compare_flags(self):
     pass