From 7e760e292afafc4bca5634bb4c07dd3a3d306800 Mon Sep 17 00:00:00 2001
From: lhpqaq <657407891@qq.com>
Date: Sun, 26 Jan 2025 13:51:14 +0800
Subject: [PATCH] add ut

---
 .../apache/bigtop/manager/ai/core/enums/PlatformTypeTest.java | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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<String> 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"));