Fixed key binding toggle

Author: joguSD

awsshell/app.py

@@ -17,6 +17,7 @@
 from prompt_toolkit.interface import AbortAction, AcceptAction
 from prompt_toolkit.auto_suggest import AutoSuggestFromHistory
 from prompt_toolkit.history import InMemoryHistory, FileHistory
+from prompt_toolkit.enums import EditingMode
 
 from awsshell.ui import create_default_layout
 from awsshell.config import Config
@@ -417,7 +418,13 @@ def create_application(self, completer, history,
             'clidocs': Buffer(read_only=True)
         }
 
+        if self.enable_vi_bindings:
+            editing_mode = EditingMode.VI
+        else:
+            editing_mode = EditingMode.EMACS
+
         return Application(
+            editing_mode=editing_mode,
             layout=self.create_layout(display_completions_in_columns, toolbar),
             mouse_support=False,
             style=style_factory.style,

tests/integration/test_keys.py

@@ -46,20 +46,20 @@ def test_F3(self):
         enable_vi_bindings = self.aws_shell.enable_vi_bindings
         with self.assertRaises(InputInterrupt):
             self.feed_key(Keys.F3)
-            assert enable_vi_bindings != self.aws_shell.enable_vi_bindings
+        assert enable_vi_bindings != self.aws_shell.enable_vi_bindings
 
     def test_F4(self):
         show_completion_columns = self.aws_shell.show_completion_columns
         with self.assertRaises(InputInterrupt):
             self.feed_key(Keys.F4)
-            assert show_completion_columns != \
-                self.aws_shell.show_completion_columns
+        assert show_completion_columns != \
+            self.aws_shell.show_completion_columns
 
     def test_F5(self):
         show_help = self.aws_shell.show_help
         with self.assertRaises(InputInterrupt):
             self.feed_key(Keys.F5)
-            assert show_help != self.aws_shell.show_help
+        assert show_help != self.aws_shell.show_help
 
     def test_F10(self):
         self.feed_key(Keys.F10)