From 5112a15564b1725b724f89190bdccaef590e7c07 Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Tue, 17 Jun 2025 10:08:30 +0200
Subject: [PATCH] Basic state manager test

---
 tests/test_state_manager.py | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 tests/test_state_manager.py

diff --git a/tests/test_state_manager.py b/tests/test_state_manager.py
new file mode 100644
index 0000000..21b16d6
--- /dev/null
+++ b/tests/test_state_manager.py
@@ -0,0 +1,35 @@
+from typing import Union
+
+import pytest
+
+from facefusion.processors.types import ProcessorState
+from facefusion.state_manager import STATE_SET, get_item, init_item, set_item
+from facefusion.types import AppContext, State
+
+
+def get_state(app_context : AppContext) -> Union[State, ProcessorState]:
+	return STATE_SET.get(app_context)
+
+
+def clear_state(app_context : AppContext) -> None:
+	STATE_SET[app_context] = {}
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> None:
+	clear_state('cli')
+	clear_state('ui')
+
+
+def test_init_item() -> None:
+	init_item('video_memory_strategy', 'tolerant')
+
+	assert get_state('cli').get('video_memory_strategy') == 'tolerant'
+	assert get_state('ui').get('video_memory_strategy') == 'tolerant'
+
+
+def test_get_item_and_set_item() -> None:
+	set_item('video_memory_strategy', 'tolerant')
+
+	assert get_item('video_memory_strategy') == 'tolerant'
+	assert get_state('ui').get('video_memory_strategy') is None