Files
facefusion/tests/test_session_context.py
T
Henry Ruhs fb21d1d1b5 refactor state manager (#1237)
* refactor state manager

* refactor state manager

* refactor state manager

* refactor state manager
2026-09-12 11:02:14 +02:00

33 lines
603 B
Python

from typing import Iterator
import pytest
from facefusion.session_context import get_session_id, resolve_local_id, set_session_id
@pytest.fixture(scope = 'function', autouse = True)
def before_each() -> Iterator[None]:
local_id = resolve_local_id()
set_session_id(local_id)
yield
set_session_id(local_id)
def test_get_session_id() -> None:
local_id = resolve_local_id()
assert get_session_id() == local_id
set_session_id('session-a')
assert get_session_id() == 'session-a'
def test_resolve_local_id() -> None:
local_id = resolve_local_id()
assert resolve_local_id() == local_id