diff --git a/bigtop-manager-ai/bigtop-manager-ai-core/src/test/java/org/apache/bigtop/manager/ai/core/enums/PlatformTypeTest.java b/bigtop-manager-ai/bigtop-manager-ai-core/src/test/java/org/apache/bigtop/manager/ai/core/enums/PlatformTypeTest.java index 863242b9..cb2bc5ea 100644 --- a/bigtop-manager-ai/bigtop-manager-ai-core/src/test/java/org/apache/bigtop/manager/ai/core/enums/PlatformTypeTest.java +++ b/bigtop-manager-ai/bigtop-manager-ai-core/src/test/java/org/apache/bigtop/manager/ai/core/enums/PlatformTypeTest.java @@ -31,10 +31,11 @@ public class PlatformTypeTest { @Test public void testGetPlatforms() { List senders = PlatformType.getPlatforms(); - assertEquals(3, senders.size()); + assertEquals(4, senders.size()); assertTrue(senders.contains("openai")); assertTrue(senders.contains("dashscope")); assertTrue(senders.contains("qianfan")); + assertTrue(senders.contains("deepseek")); } @Test @@ -42,6 +43,7 @@ public void testGetPlatformType() { assertEquals(PlatformType.OPENAI, PlatformType.getPlatformType("openai")); assertEquals(PlatformType.DASH_SCOPE, PlatformType.getPlatformType("dashscope")); assertEquals(PlatformType.QIANFAN, PlatformType.getPlatformType("qianfan")); + assertEquals(PlatformType.DEEPSEEK, PlatformType.getPlatformType("deepseek")); assertNull(PlatformType.getPlatformType("")); assertNull(PlatformType.getPlatformType(null)); assertNull(PlatformType.getPlatformType("unknown"));