1 files changed, 16 insertions, 0 deletions
diff --git a/tests/test_input.py b/tests/test_input.py
new file mode 100644
index 00000000..5f44ca1e
--- /dev/null
+++ b/tests/test_input.py
@@ -0,0 +1,16 @@
+from infini.input import Input
+
+
+def test_new_input_without_vars():
+    assert Input("test plain_str").plain_data == "test plain_str"
+    assert Input("test plain_str").get_plain_text() == "test plain_str"
+
+
+def test_new_input_with_session_id():
+    input = Input("test plain_str", variables={"session_id": "test"})
+    assert input.get_session_id() == "test"
+
+
+def test_new_input_without_session_id():
+    input = Input("test plain_str", variables={"user_id": "test"})
+    assert input.get_session_id() == "session_unknown_test"