ensure studio mode is disabled on test cleanup

This commit is contained in:
onyx-and-iris 2025-06-22 02:54:08 +01:00
parent 864751ecc9
commit b14d9b7610

View File

@ -9,7 +9,13 @@ import os
import anyio import anyio
from anyio import create_task_group from anyio import create_task_group
from pyslobs import ConnectionConfig, ScenesService, SlobsConnection, StreamingService from pyslobs import (
ConnectionConfig,
ScenesService,
SlobsConnection,
StreamingService,
TransitionsService,
)
async def cleanup(conn: SlobsConnection): async def cleanup(conn: SlobsConnection):
@ -29,6 +35,11 @@ async def cleanup(conn: SlobsConnection):
if model.recording_status != 'offline': if model.recording_status != 'offline':
await ss.toggle_recording() await ss.toggle_recording()
ts = TransitionsService(conn)
model = await ts.get_model()
if model.studio_mode:
await ts.disable_studio_mode()
conn.close() conn.close()