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

[atagar at torproject.org]

commit 085018b9bace37a9758d5a0fbe447086612a8e8a
Author: Damian Johnson <atagar at torproject.org>
Date:   Fri Sep 25 17:21:56 2020 -0700

    Test collector_caching example
    
    Ooph. Despite looking simple it took me a few hours to get these mocks right...
---
 test/unit/examples.py | 28 ++++++++++++++++++++++++++--
 1 file changed, 26 insertions(+), 2 deletions(-)

diff --git a/test/unit/examples.py b/test/unit/examples.py
index 566611c5..521a7937 100644
--- a/test/unit/examples.py
+++ b/test/unit/examples.py
@@ -91,6 +91,12 @@ Server descriptor digest invalid, expected A106452D87BD7B803B6CE916291ED368DC5BD
 Extrainfo descriptor digest is correct
 """
 
+EXPECTED_COLLECTOR_CACHING = """\
+  krypton (3E2F63E2356F52318B536A12B6445373808A5D6C)
+  dizum (7EA6EAD6FD83083C538F44038BBFA077587DD755)
+  flubber (5C2124E6C5DD75C3C17C03EEA5A51812773DE671)
+"""
+
 
 class TestExamples(unittest.TestCase):
   def setUp(self):
@@ -229,8 +235,26 @@ class TestExamples(unittest.TestCase):
             module.validate_relay(fingerprint)
             self.assertEqual(EXPECTED_CHECK_DIGESTS_BAD % server_desc.digest(), stdout_mock.getvalue())
 
-  def test_collector_caching(self):
-    pass
+  @patch('stem.descriptor.collector.File.download', Mock())
+  @patch('stem.descriptor.collector.CollecTor.files')
+  @patch('sys.stdout', new_callable = io.StringIO)
+  def test_collector_caching(self, stdout_mock, files_mock):
+    files_mock.return_value = [stem.descriptor.collector.File(
+     'archive/relay-descriptors/server-descriptors/server-descriptors-2005-12.tar',
+      ['server-descriptor 1.0'],
+      1348620,
+      '0RrqB5aMY46vTeEHYqnbPVFGZQi1auJkzyHyt0NNDcw=',
+      '2005-12-15 01:42',
+      '2005-12-17 11:06',
+      '2016-06-24 08:12',
+    )]
+
+    server_desc = list(stem.descriptor.parse_file(os.path.join(DESC_DIR, 'collector', 'server-descriptors-2005-12-cropped.tar')))
+
+    with patch('stem.descriptor.parse_file', Mock(return_value = server_desc)):
+      import collector_caching
+
+    self.assertEqual(EXPECTED_COLLECTOR_CACHING, stdout_mock.getvalue())
 
   def test_collector_reading(self):
     pass