mirror of
https://github.com/facefusion/facefusion.git
synced 2026-09-15 12:05:27 +02:00
* refactor state manager * refactor state manager * refactor state manager * refactor state manager
33 lines
603 B
Python
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
|