From 6cf4460c49854d5813a9da082a4c5e7160c471f2 Mon Sep 17 00:00:00 2001 From: huangguojie2024 <503601315@qq.com> Date: Fri, 24 Jan 2025 19:36:01 +0800 Subject: [PATCH] Add ut cases for enums classes in ai core module --- .../ai/core/enums/MessageTypeTest.java | 49 +++++++++++++++++++ .../ai/core/enums/PlatformTypeTest.java | 49 +++++++++++++++++++ 2 files changed, 98 insertions(+) create mode 100644 bigtop-manager-ai/bigtop-manager-ai-core/src/test/java/org/apache/bigtop/manager/ai/core/enums/MessageTypeTest.java create mode 100644 bigtop-manager-ai/bigtop-manager-ai-core/src/test/java/org/apache/bigtop/manager/ai/core/enums/PlatformTypeTest.java diff --git a/bigtop-manager-ai/bigtop-manager-ai-core/src/test/java/org/apache/bigtop/manager/ai/core/enums/MessageTypeTest.java b/bigtop-manager-ai/bigtop-manager-ai-core/src/test/java/org/apache/bigtop/manager/ai/core/enums/MessageTypeTest.java new file mode 100644 index 00000000..050160ae --- /dev/null +++ b/bigtop-manager-ai/bigtop-manager-ai-core/src/test/java/org/apache/bigtop/manager/ai/core/enums/MessageTypeTest.java @@ -0,0 +1,49 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * https://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, + * software distributed under the License is distributed on an + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY + * KIND, either express or implied. See the License for the + * specific language governing permissions and limitations + * under the License. + */ +package org.apache.bigtop.manager.ai.core.enums; + +import org.junit.jupiter.api.Test; + +import java.util.List; + +import static org.junit.jupiter.api.Assertions.assertEquals; +import static org.junit.jupiter.api.Assertions.assertNull; +import static org.junit.jupiter.api.Assertions.assertTrue; + +public class MessageTypeTest { + + @Test + public void testGetSenders() { + List senders = MessageType.getSenders(); + assertEquals(3, senders.size()); + assertTrue(senders.contains("user")); + assertTrue(senders.contains("ai")); + assertTrue(senders.contains("system")); + } + + @Test + public void testGetMessageSender() { + assertEquals(MessageType.USER, MessageType.getMessageSender("user")); + assertEquals(MessageType.AI, MessageType.getMessageSender("ai")); + assertEquals(MessageType.SYSTEM, MessageType.getMessageSender("system")); + assertNull(MessageType.getMessageSender("")); + assertNull(MessageType.getMessageSender(null)); + assertNull(MessageType.getMessageSender("unknown")); + } +} 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 new file mode 100644 index 00000000..863242b9 --- /dev/null +++ b/bigtop-manager-ai/bigtop-manager-ai-core/src/test/java/org/apache/bigtop/manager/ai/core/enums/PlatformTypeTest.java @@ -0,0 +1,49 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * https://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, + * software distributed under the License is distributed on an + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY + * KIND, either express or implied. See the License for the + * specific language governing permissions and limitations + * under the License. + */ +package org.apache.bigtop.manager.ai.core.enums; + +import org.junit.jupiter.api.Test; + +import java.util.List; + +import static org.junit.jupiter.api.Assertions.assertEquals; +import static org.junit.jupiter.api.Assertions.assertNull; +import static org.junit.jupiter.api.Assertions.assertTrue; + +public class PlatformTypeTest { + + @Test + public void testGetPlatforms() { + List senders = PlatformType.getPlatforms(); + assertEquals(3, senders.size()); + assertTrue(senders.contains("openai")); + assertTrue(senders.contains("dashscope")); + assertTrue(senders.contains("qianfan")); + } + + @Test + public void testGetPlatformType() { + assertEquals(PlatformType.OPENAI, PlatformType.getPlatformType("openai")); + assertEquals(PlatformType.DASH_SCOPE, PlatformType.getPlatformType("dashscope")); + assertEquals(PlatformType.QIANFAN, PlatformType.getPlatformType("qianfan")); + assertNull(PlatformType.getPlatformType("")); + assertNull(PlatformType.getPlatformType(null)); + assertNull(PlatformType.getPlatformType("unknown")); + } +}