From e8b699cba6f7e2e225281374d6dabf615e87303a Mon Sep 17 00:00:00 2001 From: onyx-and-iris Date: Thu, 8 May 2025 01:15:52 +0100 Subject: [PATCH] add output to studiomode enable/disable comands upd studiomode unit tests --- obsws_cli/studiomode.py | 2 ++ tests/test_studiomode.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/obsws_cli/studiomode.py b/obsws_cli/studiomode.py index 91b8ba9..235b106 100644 --- a/obsws_cli/studiomode.py +++ b/obsws_cli/studiomode.py @@ -16,12 +16,14 @@ def main(): def enable(ctx: typer.Context): """Enable studio mode.""" ctx.obj.set_studio_mode_enabled(True) + typer.echo('Studio mode has been enabled.') @app.command('disable | off') def disable(ctx: typer.Context): """Disable studio mode.""" ctx.obj.set_studio_mode_enabled(False) + typer.echo('Studio mode has been disabled.') @app.command('toggle | tg') diff --git a/tests/test_studiomode.py b/tests/test_studiomode.py index 94fa1fc..c151055 100644 --- a/tests/test_studiomode.py +++ b/tests/test_studiomode.py @@ -11,6 +11,7 @@ def test_studio_enable(): """Test the studio enable command.""" result = runner.invoke(app, ['studiomode', 'enable']) assert result.exit_code == 0 + assert 'Studio mode has been enabled.' in result.stdout result = runner.invoke(app, ['studiomode', 'status']) assert result.exit_code == 0 @@ -21,6 +22,7 @@ def test_studio_disable(): """Test the studio disable command.""" result = runner.invoke(app, ['studiomode', 'disable']) assert result.exit_code == 0 + assert 'Studio mode has been disabled.' in result.stdout result = runner.invoke(app, ['studiomode', 'status']) assert result.exit_code == 0