Function test_cmdfinalization_hook_keyboard_interrupt

tests/test_plugin.py:924–944

()

def test_cmdfinalization_hook_keyboard_interrupt() -> None:
    app = PluggedApp()
    app.register_cmdfinalization_hook(app.cmdfinalization_hook_keyboard_interrupt)

    # First make sure KeyboardInterrupt isn't raised unless told to
    stop = app.onecmd_plus_hooks("say hello", raise_keyboard_interrupt=False)
    assert not stop
    assert app.called_cmdfinalization == 1

    # Now enable raising the KeyboardInterrupt
    app.reset_counters()
    with pytest.raises(KeyboardInterrupt):
        stop = app.onecmd_plus_hooks("say hello", raise_keyboard_interrupt=True)
    assert not stop
    assert app.called_cmdfinalization == 1

    # Now make sure KeyboardInterrupt isn't raised if stop is already True
    app.reset_counters()
    stop = app.onecmd_plus_hooks("quit", raise_keyboard_interrupt=True)
    assert stop
    assert app.called_cmdfinalization == 1


def test_cmdfinalization_hook_passthrough_exception() -> None: