public/liteFlow
1
0
Fork 0
mirror of https://gitee.com/dromara/liteFlow.git synced 2026-06-11 06:06:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Test: SetUtil 测试

Browse Source
This commit is contained in:
LuanY77
2025-07-31 18:12:48 +08:00
parent c3254275da
commit 8cbfe6be51
10 changed files with 174 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/ProxyTest.java → liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/ai/proxy/ProxyTest.java
View File

@@ -1,8 +1,9 @@
package com.yomahub.liteflow.test;
package com.yomahub.liteflow.test.ai.proxy;
import com.yomahub.liteflow.ai.context.ChatContext;
import com.yomahub.liteflow.core.FlowExecutor;
import com.yomahub.liteflow.flow.LiteflowResponse;
import com.yomahub.liteflow.slot.DefaultContext;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
@@ -19,10 +20,10 @@ import javax.annotation.Resource;
* @since TODO
*/
@TestPropertySource(value = "classpath:application.yaml")
@TestPropertySource("classpath:/proxy/application.yaml")
@SpringBootTest(classes = {ProxyTest.class})
@EnableAutoConfiguration
@ComponentScan({"com.yomahub.liteflow.test.cmp"})
@ComponentScan({"com.yomahub.liteflow.test.ai.proxy.cmp"})
public class ProxyTest {
@Resource
@@ -30,7 +31,7 @@ public class ProxyTest {
@Test
public void testProxy() {
LiteflowResponse liteflowResponse = flowExecutor.execute2Resp("chain1", null, ChatContext.class);
LiteflowResponse liteflowResponse = flowExecutor.execute2Resp("chain1", null, ChatContext.class, DefaultContext.class);
Assertions.assertTrue(liteflowResponse.isSuccess());
}
}

2
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/cmp/ACmp.java → liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/ai/proxy/cmp/ACmp.java
View File

@@ -1,4 +1,4 @@
package com.yomahub.liteflow.test.cmp;
package com.yomahub.liteflow.test.ai.proxy.cmp;
import com.yomahub.liteflow.core.NodeComponent;
import org.springframework.stereotype.Component;

4
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/cmp/AICmp.java → liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/ai/proxy/cmp/AICmp.java
View File

@@ -1,4 +1,4 @@
package com.yomahub.liteflow.test.cmp;
package com.yomahub.liteflow.test.ai.proxy.cmp;
import com.yomahub.liteflow.ai.annotation.*;
import com.yomahub.liteflow.ai.domain.enums.ResponseType;
@@ -18,7 +18,7 @@ import com.yomahub.liteflow.ai.domain.enums.ResponseType;
model = "qwen3:32b"
)
@AIChat(
systemPrompt = "classpath:system_prompt.txt",
systemPrompt = "classpath:proxy/system_prompt.txt",
userPrompt = "{{question}}, {{answer}}"
)
@AIInput(

2
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/cmp/BCmp.java → liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/ai/proxy/cmp/BCmp.java
View File

@@ -1,4 +1,4 @@
package com.yomahub.liteflow.test.cmp;
package com.yomahub.liteflow.test.ai.proxy.cmp;
import com.yomahub.liteflow.core.NodeComponent;
import org.springframework.stereotype.Component;

2
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/cmp/Output.java → liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/ai/proxy/cmp/Output.java
View File

@@ -1,4 +1,4 @@
package com.yomahub.liteflow.test.cmp;
package com.yomahub.liteflow.test.ai.proxy.cmp;
/**
* 测试结构化输出使用

158
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/ai/util/SetUtilTest.java Normal file
View File

@@ -0,0 +1,158 @@
package com.yomahub.liteflow.test.ai.util;
import com.yomahub.liteflow.ai.util.SetUtil;
import com.yomahub.liteflow.ai.util.TriState;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Nested;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.Arguments;
import org.junit.jupiter.params.provider.MethodSource;
import java.util.*;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.stream.Stream;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.junit.jupiter.params.provider.Arguments.arguments;
/**
* SetUtil 工具类测试
*
* @author 苍镜月
* @since TODO
*/
@DisplayName("SetUtil 工具类测试")
class SetUtilTest {
// --- 为参数化测试提供数据 ---
private static final String NOT_PRESENT_METHOD_SOURCE = "com.yomahub.liteflow.test.ai.util.SetUtilTest#provideNotPresentValues";
private static final String PRESENT_METHOD_SOURCE = "com.yomahub.liteflow.test.ai.util.SetUtilTest#providePresentValues";
/**
* 提供被认为是 "not present" (不存在/为空/为默认值) 的各种值
*/
static Stream<Arguments> provideNotPresentValues() {
return Stream.of(
arguments((Object) null),
arguments(""),
arguments(" "),
arguments(new ArrayList<>()),
arguments(Collections.emptyMap()),
arguments((Object) new String[0]),
arguments((Object) new int[0]),
arguments(TriState.UNSET),
arguments(-1),
arguments(-1L),
arguments(-1.0),
arguments(-1.0f)
);
}
/**
* 提供被认为是 "present" (存在/有效) 的各种值
*/
static Stream<Arguments> providePresentValues() {
Map<String, Integer> presentMap = new HashMap<>();
presentMap.put("key", 1);
return Stream.of(
arguments("hello"),
arguments(" a "),
arguments(Arrays.asList(1, 2)),
arguments(presentMap),
arguments((Object) new String[]{"a"}),
arguments((Object) new int[]{1}),
arguments(new Object()),
arguments(TriState.TRUE),
arguments(0),
arguments(1),
arguments(0L),
arguments(0.0),
arguments(0.0f)
);
}
@Nested
@DisplayName("核心判断方法：isPresent 和 isNotPresent")
class IsPresentAndNotPresentTests {
@ParameterizedTest
@MethodSource(NOT_PRESENT_METHOD_SOURCE)
@DisplayName("isNotPresent: 对于 null、空集合或默认值应返回 true")
void testIsNotPresent_shouldReturnTrue_forEmptyOrDefaultValues(Object value) {
assertTrue(SetUtil.isNotPresent(value), "值 " + value + " 应被视为 not present");
}
@ParameterizedTest
@MethodSource(PRESENT_METHOD_SOURCE)
@DisplayName("isNotPresent: 对于有效值应返回 false")
void testIsNotPresent_shouldReturnFalse_forPresentValues(Object value) {
assertFalse(SetUtil.isNotPresent(value), "值 " + value + " 应被视为 present");
}
@ParameterizedTest
@MethodSource(PRESENT_METHOD_SOURCE)
@DisplayName("isPresent: 对于有效值应返回 true")
void testIsPresent_shouldReturnTrue_forPresentValues(Object value) {
assertTrue(SetUtil.isPresent(value), "值 " + value + " 应被视为 present");
}
@ParameterizedTest
@MethodSource(NOT_PRESENT_METHOD_SOURCE)
@DisplayName("isPresent: 对于 null、空集合或默认值应返回 false")
void testIsPresent_shouldReturnFalse_forEmptyOrDefaultValues(Object value) {
assertFalse(SetUtil.isPresent(value), "值 " + value + " 应被视为 not present");
}
}
@Nested
@DisplayName("消费方法: setIfPresent")
class SetIfPresentTests {
@ParameterizedTest
@MethodSource(PRESENT_METHOD_SOURCE)
@DisplayName("当值存在时，应执行 Consumer")
void shouldExecuteConsumer_whenValueIsPresent(Object presentValue) {
AtomicBoolean executed = new AtomicBoolean(false);
SetUtil.setIfPresent(v -> executed.set(true), presentValue);
assertTrue(executed.get(), "当值为 " + presentValue + " 时，Consumer 应该被执行");
}
@ParameterizedTest
@MethodSource(NOT_PRESENT_METHOD_SOURCE)
@DisplayName("当值不存在时，不应执行 Consumer")
void shouldNotExecuteConsumer_whenValueIsNotPresent(Object notPresentValue) {
AtomicBoolean executed = new AtomicBoolean(false);
SetUtil.setIfPresent(v -> executed.set(true), notPresentValue);
assertFalse(executed.get(), "当值为 " + notPresentValue + " 时，Consumer 不应该被执行");
}
}
@Nested
@DisplayName("消费方法: setIfNotPresent")
class SetIfNotPresentTests {
@ParameterizedTest
@MethodSource(NOT_PRESENT_METHOD_SOURCE)
@DisplayName("当值不存在时，应执行 Consumer")
void shouldExecuteConsumer_whenValueIsNotPresent(Object notPresentValue) {
AtomicBoolean executed = new AtomicBoolean(false);
SetUtil.setIfNotPresent(v -> executed.set(true), notPresentValue);
assertTrue(executed.get(), "当值为 " + notPresentValue + " 时，Consumer 应该被执行");
}
@ParameterizedTest
@MethodSource(PRESENT_METHOD_SOURCE)
@DisplayName("当值存在时，不应执行 Consumer")
void shouldNotExecuteConsumer_whenValueIsPresent(Object presentValue) {
AtomicBoolean executed = new AtomicBoolean(false);
SetUtil.setIfNotPresent(v -> executed.set(true), presentValue);
assertFalse(executed.get(), "当值为 " + presentValue + " 时，Consumer 不应该被执行");
}
}
}

6
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/resources/application.yaml
View File

@@ -1,6 +0,0 @@
liteflow:
rule-source: flow.el.xml
ai:
base-packages:
- com.yomahub.liteflow.test.cmp
enable: true

6
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/resources/proxy/application.yaml Normal file
View File

@@ -0,0 +1,6 @@
liteflow:
rule-source: proxy/flow.el.xml
ai:
base-packages:
- com.yomahub.liteflow.test.ai.proxy.cmp
enable: true

0
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/resources/flow.el.xml → liteflow-testcase-el/liteflow-testcase-el-ai/src/test/resources/proxy/flow.el.xml
View File

0
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/resources/system_prompt.txt → liteflow-testcase-el/liteflow-testcase-el-ai/src/test/resources/proxy/system_prompt.txt
View File