mirror of
https://github.com/onyx-and-iris/slobs-cli.git
synced 2025-06-27 15:20:24 +01:00
47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
import anyio
|
|
import asyncclick as click
|
|
import pytest
|
|
from asyncclick.testing import CliRunner
|
|
|
|
from slobs_cli import cli
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_stream_start():
|
|
runner = CliRunner()
|
|
result = await runner.invoke(cli, ["stream", "status"])
|
|
assert result.exit_code == 0
|
|
active = "Stream is currently active." in result.output
|
|
|
|
if not active:
|
|
result = await runner.invoke(cli, ["stream", "start"])
|
|
assert result.exit_code == 0
|
|
assert "Stream started" in result.output
|
|
await anyio.sleep(1) # Allow some time for the stream to start
|
|
else:
|
|
with pytest.raises(ExceptionGroup) as exc_info:
|
|
result = await runner.invoke(
|
|
cli, ["stream", "start"], catch_exceptions=False
|
|
)
|
|
assert exc_info.group_contains(click.Abort, match="Stream is already active.")
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_stream_stop():
|
|
runner = CliRunner()
|
|
result = await runner.invoke(cli, ["stream", "status"])
|
|
assert result.exit_code == 0
|
|
active = "Stream is currently active." in result.output
|
|
|
|
if active:
|
|
result = await runner.invoke(cli, ["stream", "stop"])
|
|
assert result.exit_code == 0
|
|
assert "Stream stopped" in result.output
|
|
await anyio.sleep(1) # Allow some time for the stream to stop
|
|
else:
|
|
with pytest.raises(ExceptionGroup) as exc_info:
|
|
result = await runner.invoke(
|
|
cli, ["stream", "stop"], catch_exceptions=False
|
|
)
|
|
assert exc_info.group_contains(click.Abort, match="Stream is already inactive.")
|