MachineTest#addOssMachineTest 测试方法将 oss bucket 作为机器添加到系统中, machineId 手动指定并以 oss 为前缀

mgr/src/test/java/cn/reghao/devops/mgr/devops/MachineTest.java

@@ -0,0 +1,71 @@
+package cn.reghao.devops.mgr.devops;
+
+import cn.reghao.devops.common.machine.Machine;
+import cn.reghao.devops.common.msg.constant.NodeStatus;
+import cn.reghao.devops.common.msg.event.EvtAgentStart;
+import cn.reghao.devops.mgr.MgrApplication;
+import cn.reghao.devops.mgr.mgr.machine.db.query.MachineQuery;
+import cn.reghao.devops.mgr.mgr.machine.db.repository.MachineInfoRepository;
+import cn.reghao.devops.mgr.mgr.machine.model.po.MachineInfo;
+import cn.reghao.devops.mgr.mgr.machine.service.MachineService;
+import cn.reghao.jutil.jdk.machine.data.detail.NetworkDetail;
+import lombok.extern.slf4j.Slf4j;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.ActiveProfiles;
+import org.springframework.test.context.junit4.SpringRunner;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.stream.Collectors;
+
+/**
+ * @author reghao
+ * @date 2025-06-05 10:53:40
+ */
+@Slf4j
+@ActiveProfiles("dev")
+@SpringBootTest(classes = MgrApplication.class)
+@RunWith(SpringRunner.class)
+public class MachineTest {
+    @Autowired
+    MachineService machineService;
+    @Test
+    public void addOssMachineTest() {
+        Map<String, String> map = new HashMap<>();
+        map.put("oss-dnkt-admin", "110.110.110.1");
+        map.put("oss-dnkt-app", "110.110.110.2");
+        map.put("oss-dnkt-cloud", "110.110.110.3");
+        map.put("oss-dnkt-timuke", "110.110.110.4");
+        map.put("oss-dnkt-nztp", "110.110.110.5");
+
+        map.forEach((machineId, ipv4) -> {
+            Machine machine = new Machine();
+            EvtAgentStart evtAgentStart = machine.detail();
+            NetworkDetail networkDetail = evtAgentStart.getNetworkDetails().get(0);
+            networkDetail.setPubicIpv4(ipv4);
+            networkDetail.setIpv4(ipv4);
+            evtAgentStart.setMachineId(machineId);
+            evtAgentStart.setNetworkDetails(List.of(networkDetail));
+            evtAgentStart.setDiskDetails(new ArrayList<>());
+            machineService.addOrUpdateMachine(evtAgentStart);
+        });
+    }
+
+    @Autowired
+    MachineQuery machineQuery;
+    @Autowired
+    MachineInfoRepository machineInfoRepository;
+    @Test
+    public void shutdownOssMachineTest() {
+        List<MachineInfo> list = machineQuery.getMachineInfos(NodeStatus.Online.getCode()).stream()
+                .filter(machineInfo -> machineInfo.getMachineId().startsWith("oss"))
+                .peek(machineInfo -> machineInfo.setStat(NodeStatus.Offline.getCode()))
+                .collect(Collectors.toList());
+        machineInfoRepository.saveAll(list);
+    }
+}