#3911 Support keybinding for different contexts with different commands

1 files changed, 18 insertions, 4 deletions

diff --git a/test/mitmproxy/tools/console/test_keymap.py b/test/mitmproxy/tools/console/test_keymap.py
index 0d6f9e88..f2c97ded 100644
--- a/test/mitmproxy/tools/console/test_keymap.py
+++ b/test/mitmproxy/tools/console/test_keymap.py
@@ -117,6 +117,21 @@ def test_load_path(tmpdir):
         kmc.load_path(km, dst)
         assert(km.get("chooser", "key1"))
 
+        with open(dst, 'w') as f:
+            f.write(
+                """
+                    -   key: key2
+                        ctx: [flowlist]
+                        cmd: foo
+                    -   key: key2
+                        ctx: [flowview]
+                        cmd: bar
+                """
+            )
+        kmc.load_path(km, dst)
+        assert(km.get("flowlist", "key2"))
+        assert(km.get("flowview", "key2"))
+
         km.add("key123", "str", ["flowlist", "flowview"])
         with open(dst, 'w') as f:
             f.write(
@@ -127,10 +142,9 @@ def test_load_path(tmpdir):
                 """
             )
         kmc.load_path(km, dst)
-        for b in km.bindings:
-            if b.key == "key123":
-                assert b.contexts == ["options"]
-                break
+        assert(km.get("flowlist", "key123"))
+        assert(km.get("flowview", "key123"))
+        assert(km.get("options", "key123"))
 
 
 def test_parse():