Add preloop and postloop hook capabilities

1 files changed, 48 insertions, 0 deletions

diff --git a/tests/test_plugin.py b/tests/test_plugin.py
index c82e6bcd..a6fc8fc1 100644
--- a/tests/test_plugin.py
+++ b/tests/test_plugin.py
@@ -23,6 +23,12 @@ class Plugin:
     def reset_counters(self):
         self.called_pph = 0
 
+    def prepost_hook_one(self):
+        self.poutput("one")
+
+    def prepost_hook_two(self):
+        self.poutput("two")
+
     def pph(self, statement: cmd2.Statement) -> Tuple[bool, cmd2.Statement]:
         self.called_pph += 1
         return False, statement
@@ -53,6 +59,48 @@ class PluggedApp(Plugin, cmd2.Cmd):
 # test hooks
 #
 ###
+def test_preloop_hook(capsys):
+    app = PluggedApp()
+    app.register_preloop_hook(app.prepost_hook_one)
+    app.cmdqueue.append('say hello')
+    app.cmdqueue.append('quit')
+    app.cmdloop()
+    out, err = capsys.readouterr()
+    assert out == 'one\nhello\n'
+    assert not err
+
+def test_preloop_hooks(capsys):
+    app = PluggedApp()
+    app.register_preloop_hook(app.prepost_hook_one)
+    app.register_preloop_hook(app.prepost_hook_two)
+    app.cmdqueue.append('say hello')
+    app.cmdqueue.append('quit')
+    app.cmdloop()
+    out, err = capsys.readouterr()
+    assert out == 'one\ntwo\nhello\n'
+    assert not err
+
+def test_postloop_hook(capsys):
+    app = PluggedApp()
+    app.register_postloop_hook(app.prepost_hook_one)
+    app.cmdqueue.append('say hello')
+    app.cmdqueue.append('quit')
+    app.cmdloop()
+    out, err = capsys.readouterr()
+    assert out == 'hello\none\n'
+    assert not err
+
+def test_postloop_hooks(capsys):
+    app = PluggedApp()
+    app.register_postloop_hook(app.prepost_hook_one)
+    app.register_postloop_hook(app.prepost_hook_two)
+    app.cmdqueue.append('say hello')
+    app.cmdqueue.append('quit')
+    app.cmdloop()
+    out, err = capsys.readouterr()
+    assert out == 'hello\none\ntwo\n'
+    assert not err
+
 def test_postparsing_hook(capsys):
     app = PluggedApp()
     app.onecmd_plus_hooks('say hello')