commit fc902a4917b3fb96e6c11b71a8f24737bbae221f
Author: Damian Johnson <ata...@torproject.org>
Date: Mon Sep 21 16:28:30 2020 -0700
Test benchmark_server_descriptor_stem example
---
test/unit/examples.py | 33 +++++++++++++++++++++++----------
1 file changed, 23 insertions(+), 10 deletions(-)
diff --git a/test/unit/examples.py b/test/unit/examples.py
index 12e3d68f..6a00d9b7 100644
--- a/test/unit/examples.py
+++ b/test/unit/examples.py
@@ -33,20 +33,25 @@ Relay BD4172533C3F7271ABCCD9F057E06FD91547C42B
"""
+EXPECTED_BENCHMARK_SERVER_DESC_PREFIX = """\
+Finished measure_average_advertised_bandwidth('%s')
+ Total time: 0 seconds
+ Processed server descriptors: 5
+ Average advertised bandwidth: 313183
+ Time per server descriptor:
+""".rstrip()
-def run_example(module):
+
+def import_example(module_name):
"""
- Invoke the given example, returning its stdout.
+ Import this example module.
"""
original_path = list(sys.path)
sys.path.append(EXAMPLE_DIR)
try:
- with patch('sys.stdout', new_callable = io.StringIO) as stdout_mock:
- importlib.import_module(module)
-
- return stdout_mock.getvalue()
+ return importlib.import_module(module_name)
finally:
sys.path = original_path
@@ -70,7 +75,8 @@ class TestExamples(unittest.TestCase):
self.fail("Changed our examples directory? The following are untested:
%s" % ', '.join(missing))
@patch('stem.descriptor.remote.get_bandwidth_file')
- def test_bandwidth_stats(self, get_bandwidth_file_mock):
+ @patch('sys.stdout', new_callable = io.StringIO)
+ def test_bandwidth_stats(self, stdout_mock, get_bandwidth_file_mock):
get_bandwidth_file_mock().run.return_value = [BandwidthFile.create({
'content': [
'bw=1 bw_mean=807445 bw_median=911047 consensus_bandwidth=1190000
node_id=$FDCF49562E65B1CC219410009BD48A9EED387C77',
@@ -78,10 +84,17 @@ class TestExamples(unittest.TestCase):
],
})]
- self.assertEqual(EXPECTED_BANDWIDTH_STATS, run_example('bandwidth_stats'))
+ import_example('bandwidth_stats')
+ self.assertEqual(EXPECTED_BANDWIDTH_STATS, stdout_mock.getvalue())
- def test_benchmark_server_descriptor_stem(self):
- pass
+ @patch('sys.stdout', new_callable = io.StringIO)
+ def test_benchmark_server_descriptor_stem(self, stdout_mock):
+ path = os.path.join(test.STEM_BASE, 'test', 'unit', 'descriptor', 'data',
'collector', 'server-descriptors-2005-12-cropped.tar')
+
+ module = import_example('benchmark_server_descriptor_stem')
+ module.measure_average_advertised_bandwidth(path)
+
+
self.assertTrue(stdout_mock.getvalue().startswith(EXPECTED_BENCHMARK_SERVER_DESC_PREFIX
% path))
def test_benchmark_stem(self):
pass
_______________________________________________
tor-commits mailing list
tor-commits@lists.torproject.org
https://lists.torproject.org/cgi-bin/mailman/listinfo/tor-commits
