add output to studiomode enable/disable comands

upd studiomode unit tests
This commit is contained in:
onyx-and-iris 2025-05-08 01:15:52 +01:00
parent 4f0a3816ba
commit e8b699cba6
2 changed files with 4 additions and 0 deletions

View File

@ -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')

View File

@ -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