From e999efafbc75133b03211dbecaf9bc424bf478ff Mon Sep 17 00:00:00 2001 From: 苏向夜 Date: Fri, 26 Jan 2024 23:28:29 +0800 Subject: :sparkles: feat(ci): add new generator tests --- tests/test_generator.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) (limited to 'tests/test_generator.py') diff --git a/tests/test_generator.py b/tests/test_generator.py index 32f1d626..bcab31ed 100644 --- a/tests/test_generator.py +++ b/tests/test_generator.py @@ -9,3 +9,30 @@ def test_generator(): } generator.match(Output("text", "test.event1")) assert generator.output(Output("text", "test.event1")) == "Event1 文本" + + +def test_generator_with_var(): + generator = Generator() + generator.events = { + "test.event1": "Event1 文本: {{ var }}", + } + + assert ( + generator.output(Output("text", "test.event1", variables={"var": "变量测试"})) + == "Event1 文本: 变量测试" + ) + + +def test_generator_with_function(): + def add(a, b): + return a + b + + generator = Generator() + generator.events = { + "test.event": "{{ func(1, 2) }}", + } + + assert ( + generator.output(Output("text", "test.event", variables={"func": add})) + == "3" + ) -- cgit v1.2.3-70-g09d2