From 0e8be9de58454de079d772dec6c0ef9c1774a775 Mon Sep 17 00:00:00 2001 From: 苏向夜 Date: Fri, 26 Jan 2024 13:56:24 +0800 Subject: :recycle: refactor(infini): refacted infini as version 2.0.0-alpha.1 --- tests/test_input.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 tests/test_input.py (limited to 'tests/test_input.py') 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" -- cgit v1.2.3-70-g09d2