commit 437ea486834ae9e320675ec0c80aa4ecf293fcc6 Author: Damian Johnson atagar@torproject.org Date: Tue Sep 29 18:39:46 2020 -0700
Test queue_listener example --- test/unit/examples.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-)
diff --git a/test/unit/examples.py b/test/unit/examples.py index 0b4d7082..6e32a0f2 100644 --- a/test/unit/examples.py +++ b/test/unit/examples.py @@ -772,8 +772,18 @@ class TestExamples(unittest.TestCase):
self.assertEqual(EXPECTED_PERSISTING_A_CONSENSUS, stdout_mock.getvalue())
- def test_queue_listener(self): - pass + @patch('stem.control.Controller.from_port', spec = Controller) + @patch('sys.stdout', new_callable = io.StringIO) + def test_queue_listener(self, stdout_mock, from_port_mock): + bw_event = ControlMessage.from_str('650 BW 15 25', 'EVENT', normalize = True) + + controller = from_port_mock().__enter__() + controller.add_event_listener.side_effect = lambda handler, event_type: handler(bw_event) + + with patch('time.time', Mock(side_effect = [1, 1, 10])): + import queue_listener + + self.assertEqual('I got a BW event for 15 bytes downloaded and 25 bytes uploaded\n', stdout_mock.getvalue())
def test_read_with_parse_file(self): pass