From ba13fda10d3065a0c8dfd95d55680675b3bf08c2 Mon Sep 17 00:00:00 2001 From: Shadab Zafar Date: Mon, 20 Jun 2016 00:50:41 +0530 Subject: Add tests for PathEdit --- test/mitmproxy/console/test_pathedit.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/test/mitmproxy/console/test_pathedit.py b/test/mitmproxy/console/test_pathedit.py index 107a48ac..ff6ef846 100644 --- a/test/mitmproxy/console/test_pathedit.py +++ b/test/mitmproxy/console/test_pathedit.py @@ -2,6 +2,8 @@ import os from os.path import normpath from mitmproxy.console import pathedit +from mock import patch + from .. import tutils @@ -47,3 +49,25 @@ class TestPathCompleter: s = "thisisatotallynonexistantpathforsure" assert c.complete(s) == s assert c.final == s + + +class TestPathEdit(): + + def test_keypress(self): + + pe = pathedit.PathEdit() + + with patch('urwid.widget.Edit.get_edit_text') as get_text, \ + patch('urwid.widget.Edit.set_edit_text') as set_text: + + cd = tutils.test_data.path("completion") + get_text.return_value = os.path.join(cd, "a") + + # Pressing tab should set completed path + pe.keypress((1,), "tab") + set_text_called_with = set_text.call_args[0][0] + assert set_text_called_with.endswith(normpath("/completion/aaa")) + + # Pressing any other key should reset + pe.keypress((1,), "a") + assert pe.lookup is None -- cgit v1.2.3