[Author Prev][Author Next][Thread Prev][Thread Next][Author Index][Thread Index]

[tor-commits] [stem/master] Unit test downloading descriptors from orport



commit 04186deb315d7b2d3ec1074757604d5f98656116
Author: Damian Johnson <atagar@xxxxxxxxxxxxxx>
Date:   Mon Apr 23 19:37:57 2018 -0700

    Unit test downloading descriptors from orport
---
 test/unit/descriptor/remote.py | 88 +++++++++++++++++++++++++++++++-----------
 1 file changed, 65 insertions(+), 23 deletions(-)

diff --git a/test/unit/descriptor/remote.py b/test/unit/descriptor/remote.py
index b2cfa7c1..7d92c695 100644
--- a/test/unit/descriptor/remote.py
+++ b/test/unit/descriptor/remote.py
@@ -29,9 +29,9 @@ except ImportError:
 
 try:
   # added in python 3.3
-  from unittest.mock import patch, Mock
+  from unittest.mock import patch, Mock, MagicMock
 except ImportError:
-  from mock import patch, Mock
+  from mock import patch, Mock, MagicMock
 
 # The urlopen() method is in a different location depending on if we're using
 # python 2.x or 3.x. The 2to3 converter accounts for this in imports, but not
@@ -126,10 +126,20 @@ HEADER = '\r\n'.join([
   'Content-Encoding: %s',
 ])
 
+ORPORT_DESCRIPTOR = 'HTTP/1.0 200 OK\n' + HEADER + '\n\n' + TEST_DESCRIPTOR
 
-def _urlopen_mock(data, encoding = 'identity'):
-  urlopen_mock = Mock()
-  urlopen_mock().read.return_value = data
+
+def _orport_mock(data):
+  connect_mock = MagicMock()
+  relay_mock = connect_mock().__enter__()
+  circ_mock = relay_mock.create_circuit().__enter__()
+  circ_mock.send().data = data
+  return connect_mock
+
+
+def _dirport_mock(data, encoding = 'identity'):
+  dirport_mock = Mock()
+  dirport_mock().read.return_value = data
 
   if stem.prereq.is_python_3():
     headers = HTTPMessage()
@@ -138,11 +148,11 @@ def _urlopen_mock(data, encoding = 'identity'):
       key, value = line.split(': ', 1)
       headers.add_header(key, encoding if key == 'Content-Encoding' else value)
 
-    urlopen_mock().headers = headers
+    dirport_mock().headers = headers
   else:
-    urlopen_mock().headers = HTTPMessage(io.BytesIO(HEADER % encoding))
+    dirport_mock().headers = HTTPMessage(io.BytesIO(HEADER % encoding))
 
-  return urlopen_mock
+  return dirport_mock
 
 
 class TestDescriptorDownloader(unittest.TestCase):
@@ -150,6 +160,38 @@ class TestDescriptorDownloader(unittest.TestCase):
     # prevent our mocks from impacting other tests
     stem.descriptor.remote.SINGLETON_DOWNLOADER = None
 
+  @patch('stem.client.Relay.connect', _orport_mock(ORPORT_DESCRIPTOR))
+  def test_using_orport(self):
+    """
+    Download a descriptor through the ORPort.
+    """
+
+    reply = stem.descriptor.remote.their_server_descriptor(
+      endpoints = [stem.ORPort('12.34.56.78', 1100)],
+      fall_back_to_authority = False,
+      validate = True,
+    )
+
+    self.assertEqual(1, len(list(reply)))
+    self.assertEqual('moria1', list(reply)[0].nickname)
+    self.assertEqual(5, len(reply.reply_headers))
+
+  @patch(URL_OPEN, _dirport_mock(TEST_DESCRIPTOR))
+  def test_using_dirport(self):
+    """
+    Download a descriptor through the DirPort.
+    """
+
+    reply = stem.descriptor.remote.their_server_descriptor(
+      endpoints = [stem.DirPort('12.34.56.78', 1100)],
+      fall_back_to_authority = False,
+      validate = True,
+    )
+
+    self.assertEqual(1, len(list(reply)))
+    self.assertEqual('moria1', list(reply)[0].nickname)
+    self.assertEqual(5, len(reply.reply_headers))
+
   def test_gzip_url_override(self):
     query = stem.descriptor.remote.Query(TEST_RESOURCE, start = False)
     self.assertEqual([Compression.PLAINTEXT], query.compression)
@@ -177,7 +219,7 @@ class TestDescriptorDownloader(unittest.TestCase):
       query = stem.descriptor.remote.Query(TEST_RESOURCE, compression = Compression.LZMA, start = False)
       self.assertEqual([Compression.PLAINTEXT], query.compression)
 
-  @patch(URL_OPEN, _urlopen_mock(read_resource('compressed_identity'), encoding = 'identity'))
+  @patch(URL_OPEN, _dirport_mock(read_resource('compressed_identity'), encoding = 'identity'))
   def test_compression_plaintext(self):
     """
     Download a plaintext descriptor.
@@ -192,7 +234,7 @@ class TestDescriptorDownloader(unittest.TestCase):
     self.assertEqual(1, len(descriptors))
     self.assertEqual('moria1', descriptors[0].nickname)
 
-  @patch(URL_OPEN, _urlopen_mock(read_resource('compressed_gzip'), encoding = 'gzip'))
+  @patch(URL_OPEN, _dirport_mock(read_resource('compressed_gzip'), encoding = 'gzip'))
   def test_compression_gzip(self):
     """
     Download a gip compressed descriptor.
@@ -207,7 +249,7 @@ class TestDescriptorDownloader(unittest.TestCase):
     self.assertEqual(1, len(descriptors))
     self.assertEqual('moria1', descriptors[0].nickname)
 
-  @patch(URL_OPEN, _urlopen_mock(read_resource('compressed_zstd'), encoding = 'x-zstd'))
+  @patch(URL_OPEN, _dirport_mock(read_resource('compressed_zstd'), encoding = 'x-zstd'))
   def test_compression_zstd(self):
     """
     Download a zstd compressed descriptor.
@@ -226,7 +268,7 @@ class TestDescriptorDownloader(unittest.TestCase):
     self.assertEqual(1, len(descriptors))
     self.assertEqual('moria1', descriptors[0].nickname)
 
-  @patch(URL_OPEN, _urlopen_mock(read_resource('compressed_lzma'), encoding = 'x-tor-lzma'))
+  @patch(URL_OPEN, _dirport_mock(read_resource('compressed_lzma'), encoding = 'x-tor-lzma'))
   def test_compression_lzma(self):
     """
     Download a lzma compressed descriptor.
@@ -245,7 +287,7 @@ class TestDescriptorDownloader(unittest.TestCase):
     self.assertEqual(1, len(descriptors))
     self.assertEqual('moria1', descriptors[0].nickname)
 
-  @patch(URL_OPEN, _urlopen_mock(TEST_DESCRIPTOR))
+  @patch(URL_OPEN, _dirport_mock(TEST_DESCRIPTOR))
   def test_reply_headers(self):
     query = stem.descriptor.remote.get_server_descriptors('9695DFC35FFEB861329B9F1AB04C46397020CE31', start = False)
     self.assertEqual(None, query.reply_headers)  # initially we don't have a reply
@@ -268,7 +310,7 @@ class TestDescriptorDownloader(unittest.TestCase):
     self.assertEqual(1, len(descriptors))
     self.assertEqual('moria1', descriptors[0].nickname)
 
-  @patch(URL_OPEN, _urlopen_mock(TEST_DESCRIPTOR))
+  @patch(URL_OPEN, _dirport_mock(TEST_DESCRIPTOR))
   def test_query_download(self):
     """
     Check Query functionality when we successfully download a descriptor.
@@ -293,7 +335,7 @@ class TestDescriptorDownloader(unittest.TestCase):
     self.assertEqual('9695DFC35FFEB861329B9F1AB04C46397020CE31', desc.fingerprint)
     self.assertEqual(TEST_DESCRIPTOR.strip(), desc.get_bytes())
 
-  @patch(URL_OPEN, _urlopen_mock(b'some malformed stuff'))
+  @patch(URL_OPEN, _dirport_mock(b'some malformed stuff'))
   def test_query_with_malformed_content(self):
     """
     Query with malformed descriptor content.
@@ -321,12 +363,12 @@ class TestDescriptorDownloader(unittest.TestCase):
     self.assertRaises(ValueError, query.run)
 
   @patch(URL_OPEN)
-  def test_query_with_timeout(self, urlopen_mock):
+  def test_query_with_timeout(self, dirport_mock):
     def urlopen_call(*args, **kwargs):
       time.sleep(0.06)
       raise socket.timeout('connection timed out')
 
-    urlopen_mock.side_effect = urlopen_call
+    dirport_mock.side_effect = urlopen_call
 
     query = stem.descriptor.remote.Query(
       TEST_RESOURCE,
@@ -341,7 +383,7 @@ class TestDescriptorDownloader(unittest.TestCase):
     # Check that we don't make a third.
 
     self.assertRaises(socket.timeout, query.run)
-    self.assertEqual(2, urlopen_mock.call_count)
+    self.assertEqual(2, dirport_mock.call_count)
 
   def test_query_with_invalid_endpoints(self):
     invalid_endpoints = {
@@ -355,7 +397,7 @@ class TestDescriptorDownloader(unittest.TestCase):
       expected_error = re.escape('Endpoints must be an stem.ORPort, stem.DirPort, or two value tuple. ' + error_suffix)
       self.assertRaisesRegexp(ValueError, expected_error, stem.descriptor.remote.Query, TEST_RESOURCE, 'server-descriptor 1.0', endpoints = endpoints)
 
-  @patch(URL_OPEN, _urlopen_mock(TEST_DESCRIPTOR))
+  @patch(URL_OPEN, _dirport_mock(TEST_DESCRIPTOR))
   def test_can_iterate_multiple_times(self):
     query = stem.descriptor.remote.Query(
       TEST_RESOURCE,
@@ -381,7 +423,7 @@ class TestDescriptorDownloader(unittest.TestCase):
     self.assertTrue(len(fallback_directories) > 10)
     self.assertEqual('5.39.92.199', fallback_directories['0BEA4A88D069753218EAAAD6D22EA87B9A1319D6'].address)
 
-  @patch(URL_OPEN, _urlopen_mock(FALLBACK_DIR_CONTENT))
+  @patch(URL_OPEN, _dirport_mock(FALLBACK_DIR_CONTENT))
   def test_fallback_directories_from_remote(self):
     fallback_directories = stem.descriptor.remote.FallbackDirectory.from_remote()
     header = OrderedDict((('type', 'fallback'), ('version', '2.0.0'), ('timestamp', '20170526090242')))
@@ -465,15 +507,15 @@ class TestDescriptorDownloader(unittest.TestCase):
 
       self.assertEqual(expected, stem.descriptor.remote.FallbackDirectory.from_cache(tmp.name))
 
-  @patch(URL_OPEN, _urlopen_mock(b''))
+  @patch(URL_OPEN, _dirport_mock(b''))
   def test_fallback_directories_from_remote_empty(self):
     self.assertRaisesRegexp(IOError, 'did not have any content', stem.descriptor.remote.FallbackDirectory.from_remote)
 
-  @patch(URL_OPEN, _urlopen_mock(b'\n'.join(FALLBACK_DIR_CONTENT.splitlines()[1:])))
+  @patch(URL_OPEN, _dirport_mock(b'\n'.join(FALLBACK_DIR_CONTENT.splitlines()[1:])))
   def test_fallback_directories_from_remote_no_header(self):
     self.assertRaisesRegexp(IOError, 'does not have a type field indicating it is fallback directory metadata', stem.descriptor.remote.FallbackDirectory.from_remote)
 
-  @patch(URL_OPEN, _urlopen_mock(FALLBACK_DIR_CONTENT.replace(b'version=2.0.0', b'version')))
+  @patch(URL_OPEN, _dirport_mock(FALLBACK_DIR_CONTENT.replace(b'version=2.0.0', b'version')))
   def test_fallback_directories_from_remote_malformed_header(self):
     self.assertRaisesRegexp(IOError, 'Malformed fallback directory header line: /\* version \*/', stem.descriptor.remote.FallbackDirectory.from_remote)
 



_______________________________________________
tor-commits mailing list
tor-commits@xxxxxxxxxxxxxxxxxxxx
https://lists.torproject.org/cgi-bin/mailman/listinfo/tor-commits