Add tests for set_focus

1 files changed, 16 insertions, 0 deletions

diff --git a/test/mitmproxy/console/test_flowlist.py b/test/mitmproxy/console/test_flowlist.py
index 7c442b63..d63dab1c 100644
--- a/test/mitmproxy/console/test_flowlist.py
+++ b/test/mitmproxy/console/test_flowlist.py
@@ -1,4 +1,5 @@
 from unittest import mock
+import urwid
 
 import mitmproxy.tools.console.flowlist as flowlist
 from mitmproxy.tools import console
@@ -19,3 +20,18 @@ class TestFlowlist:
         with mock.patch('mitmproxy.tools.console.signals.status_message.send') as mock_thing:
             x.new_request("nonexistent url", "GET")
         mock_thing.assert_called_once_with(message="Invalid URL: No hostname given")
+
+    def test_logbuffer_set_focus(self):
+        m = self.mkmaster()
+        b = flowlist.LogBufferBox(m)
+        e = urwid.Text("Log message")
+        m.logbuffer.append(e)
+        m.logbuffer.append(e)
+
+        assert len(m.logbuffer) == 2
+        b.set_focus(0)
+        assert m.logbuffer.focus == 0
+        b.set_focus(1)
+        assert m.logbuffer.focus == 1
+        b.set_focus(2)
+        assert m.logbuffer.focus == 1