318 lines
9.3 KiB
Go
318 lines
9.3 KiB
Go
package main
|
||
|
||
import (
|
||
"fmt"
|
||
"strconv"
|
||
"testing"
|
||
"time"
|
||
"wdd.io/agent-common/assert"
|
||
"wdd.io/agent-common/pusher"
|
||
"wdd.io/agent-common/utils"
|
||
"wdd.io/agent-operator/config"
|
||
)
|
||
|
||
var CmiiDevNamespaceList = []string{
|
||
"uavcloud-dev",
|
||
"uavcloud-devflight",
|
||
"uavcloud-devoperation",
|
||
}
|
||
|
||
func TestFindAppNotHealthyOrRestartCountGreaterThanN(t *testing.T) {
|
||
|
||
deploymentRestartCountGreaterThanN := FindAppNotHealthyOrRestartCountGreaterThanN("devflight", 10)
|
||
deploymentRestartCountGreaterThanN = FilterAllCmiiAppSoft(deploymentRestartCountGreaterThanN)
|
||
|
||
for _, deploymentInterface := range deploymentRestartCountGreaterThanN {
|
||
println()
|
||
utils.BeautifulPrint(deploymentInterface)
|
||
println()
|
||
}
|
||
|
||
}
|
||
|
||
func TestFindDeploymentReplicasSmallerThanN(t *testing.T) {
|
||
|
||
deploymentReplicasSmallerThanN := FindDeploymentReplicasSmallerThanN("devflight", 0)
|
||
|
||
for _, deploymentInterface := range deploymentReplicasSmallerThanN {
|
||
println()
|
||
utils.BeautifulPrint(deploymentInterface)
|
||
println()
|
||
}
|
||
|
||
}
|
||
|
||
func TestFindCmiiMiddlewarePodInterface(t *testing.T) {
|
||
middlewarePodInterface := FindCmiiMiddlewarePodInterface(config.Integration)
|
||
|
||
for _, middlePod := range middlewarePodInterface {
|
||
println()
|
||
utils.BeautifulPrint(middlePod)
|
||
println()
|
||
}
|
||
}
|
||
|
||
func TestRollBackCmiiDeploymentFromUpdateLog(t *testing.T) {
|
||
updateLog := RollBackCmiiDeploymentFromUpdateLog("2024-01-10-14-37-07 uavcloud-devflight cmii-uav-depotautoreturn 12345678 123sdsa45678")
|
||
|
||
assert.Equal(t, updateLog, true, "roll back from update log failed !")
|
||
}
|
||
|
||
func TestRestartCmiiBackendDeployment(t *testing.T) {
|
||
|
||
RestartCmiiBackendDeployment("dev")
|
||
}
|
||
|
||
func TestRestartCmiiFrontendDeployment(t *testing.T) {
|
||
RestartCmiiFrontendDeployment("dev")
|
||
}
|
||
|
||
func TestFindDeploymentNotHealthy(t *testing.T) {
|
||
|
||
for _, devNamespace := range CmiiDevNamespaceList {
|
||
notHealthy := FindDeploymentNotHealthy(devNamespace)
|
||
notHealthy = FilterAllCmiiAppSoft(notHealthy)
|
||
for _, deploymentInterface := range notHealthy {
|
||
utils.BeautifulPrint(deploymentInterface)
|
||
}
|
||
}
|
||
}
|
||
|
||
func TestFindAllNodeNotHealthy(t *testing.T) {
|
||
start := time.Now()
|
||
allNodeNotHealthy := FindAllNodeNotHealthy()
|
||
elapsed := time.Since(start).Milliseconds()
|
||
fmt.Printf("执行耗时： %d ms\n", elapsed)
|
||
|
||
allNodeNotHealthy = FilterAllCmiiNodeSoft(allNodeNotHealthy)
|
||
|
||
assert.Equal(t, len(allNodeNotHealthy), 0, "have unhealthy pod !")
|
||
|
||
for _, nodeInterface := range allNodeNotHealthy {
|
||
println()
|
||
utils.BeautifulPrint(nodeInterface)
|
||
println()
|
||
}
|
||
|
||
}
|
||
|
||
func TestFindPodNotHealthy(t *testing.T) {
|
||
|
||
podNotHealthy := FindPodNotHealthy("devfl")
|
||
podNotHealthy = FilterAllCmiiPodSoft(podNotHealthy)
|
||
|
||
for _, podInterface := range podNotHealthy {
|
||
t.Logf("[%s] [%s]", podInterface.Name, podInterface.PodPhase)
|
||
}
|
||
}
|
||
|
||
func TestFindPodNotHealthy_And_Delete(t *testing.T) {
|
||
podNotHealthy := FindPodNotHealthy("devf")
|
||
podNotHealthy = FilterAllCmiiPodSoft(podNotHealthy)
|
||
|
||
for _, podInterface := range podNotHealthy {
|
||
t.Logf("[%s] [%s]", podInterface.Name, podInterface.PodPhase)
|
||
podDelete := DefaultCmiiOperator.PodDelete(podInterface.Namespace, podInterface.Name)
|
||
|
||
assert.Equal(t, podDelete, true, "delete of ", podInterface.Namespace, podInterface.Name, " failed !")
|
||
}
|
||
|
||
}
|
||
|
||
func TestFilterAllCmiiAppStrict(t *testing.T) {
|
||
allInterface := DefaultCmiiOperator.DeploymentAllInterface("devflight")
|
||
FilterAllCmiiAppStrict(allInterface)
|
||
}
|
||
|
||
func TestRestartDeploymentFromList(t *testing.T) {
|
||
allInterface := DefaultCmiiOperator.DeploymentAllInterface("devflight")
|
||
allInterface = FilterAllCmiiAppSoft(allInterface)
|
||
|
||
RestartDeploymentFromList(allInterface)
|
||
|
||
}
|
||
|
||
func TestBackupAllCmiiDeploymentToMap(t *testing.T) {
|
||
|
||
backendMap, frontendMap, srsMap := BackupAllCmiiDeploymentToMap(config.Demo)
|
||
|
||
utils.BeautifulPrint(backendMap)
|
||
utils.BeautifulPrint(frontendMap)
|
||
utils.BeautifulPrint(srsMap)
|
||
}
|
||
|
||
func TestBackupAllCmiiDeploymentToList(t *testing.T) {
|
||
allCmiiImageList := BackupAllCmiiDeploymentToList(config.Demo, true)
|
||
|
||
utils.BeautifulPrint(allCmiiImageList)
|
||
}
|
||
|
||
// 直接更新DEMO的Tag，实现DEMO更新
|
||
func TestUpdateCmiiImageTagFromNameTagMap(t *testing.T) {
|
||
|
||
backendMap := map[string]string{
|
||
"cmii-admin-data": "6.0.0",
|
||
"cmii-admin-gateway": "6.0.0",
|
||
"cmii-admin-user": "6.0.0",
|
||
"cmii-open-gateway": "6.0.0",
|
||
"cmii-suav-supervision": "6.0.0",
|
||
"cmii-uav-airspace": "6.0.0",
|
||
"cmii-uav-alarm": "6.0.0",
|
||
"cmii-uav-brain": "6.0.0",
|
||
"cmii-uav-cloud-live": "6.0.0",
|
||
"cmii-uav-cms": "6.0.0",
|
||
"cmii-uav-data-post-process": "6.0.0",
|
||
"cmii-uav-developer": "6.0.0",
|
||
"cmii-uav-device": "6.0.0",
|
||
"cmii-uav-emergency": "6.0.0",
|
||
"cmii-uav-gateway": "6.0.0",
|
||
"cmii-uav-gis-server": "6.0.0",
|
||
"cmii-uav-industrial-portfolio": "6.0.0",
|
||
"cmii-uav-integration": "6.0.0",
|
||
"cmii-uav-logger": "6.0.0",
|
||
"cmii-uav-material-warehouse": "6.0.0",
|
||
"cmii-uav-mission": "6.0.0",
|
||
"cmii-uav-mqtthandler": "6.0.0",
|
||
"cmii-uav-notice": "6.0.0",
|
||
"cmii-uav-oauth": "6.0.0",
|
||
"cmii-uav-process": "6.0.0",
|
||
"cmii-uav-surveillance": "6.0.0",
|
||
"cmii-uav-threedsimulation": "6.0.0",
|
||
"cmii-uav-tower": "6.0.0",
|
||
"cmii-uav-user": "6.0.0",
|
||
"cmii-uav-waypoint": "6.0.0",
|
||
"cmii-uav-sense-adapter": "6.0.0",
|
||
"cmii-uas-lifecycle": "6.0.0",
|
||
"cmii-uas-gateway": "6.0.0",
|
||
}
|
||
|
||
frontendMap := map[string]string{
|
||
"cmii-suav-platform-supervision": "6.0.0",
|
||
"cmii-suav-platform-supervisionh5": "6.0.0",
|
||
"cmii-uav-platform": "6.0.0",
|
||
"cmii-uav-platform-media": "6.0.0",
|
||
"cmii-uav-platform-ai-brain": "6.0.0",
|
||
"cmii-uav-platform-cms-portal": "6.0.0",
|
||
"cmii-uav-platform-armypeople": "6.0.0",
|
||
"cmii-uav-platform-mws": "6.0.0",
|
||
"cmii-uav-platform-oms": "6.0.0",
|
||
"cmii-uav-platform-open": "6.0.0",
|
||
"cmii-uav-platform-securityh5": "6.0.0",
|
||
"cmii-uav-platform-share": "6.0.0",
|
||
"cmii-uav-platform-uasms": "6.0.0",
|
||
"cmii-uav-platform-uas": "6.0.0",
|
||
//"cmii-uav-platform-base": "5.7.0",
|
||
//"cmii-uav-platform-detection": "5.7.0",
|
||
//"cmii-uav-platform-emergency-rescue": "5.7.0",
|
||
//"cmii-uav-platform-logistics": "5.7.0",
|
||
//"cmii-uav-platform-multiterminal": "5.7.0",
|
||
//"cmii-uav-platform-seniclive": "5.7.0",
|
||
//"cmii-uav-platform-splice": "5.7.0",
|
||
//"cmii-uav-platform-threedsimulation": "5.7.0",
|
||
//"cmii-uav-platform-visualization": "5.7.0",
|
||
//"cmii-uav-platform-security": "4.1.6",
|
||
}
|
||
|
||
result := UpdateCmiiImageTagFromNameTagMap(config.Demo, backendMap)
|
||
utils.BeautifulPrint(result)
|
||
|
||
result = UpdateCmiiImageTagFromNameTagMap(config.Demo, frontendMap)
|
||
utils.BeautifulPrint(result)
|
||
|
||
}
|
||
|
||
func TestScaleCmiiFrontendDeploymentToDesiredReplicas(t *testing.T) {
|
||
ScaleCmiiFrontendDeploymentToDesiredReplicas(config.Dev, 0)
|
||
}
|
||
|
||
func TestScaleCmiiBackendDeploymentToDesiredReplicas(t *testing.T) {
|
||
ScaleCmiiBackendDeploymentToDesiredReplicas(config.Dev, 0)
|
||
}
|
||
|
||
func TestBackupAllDeploymentFromEnv(t *testing.T) {
|
||
|
||
BackupAllDeploymentFromEnvToFile(config.Demo)
|
||
|
||
}
|
||
|
||
func TestBackUpAllCmiiAppImageNameFromEnv(t *testing.T) {
|
||
|
||
BackUpAllCmiiAppImageNameFromEnv(config.Demo)
|
||
|
||
}
|
||
|
||
func TestRestartCmiiDeployment(t *testing.T) {
|
||
cmiiEnv := config.Demo
|
||
appName := "cmii-uav-notice"
|
||
|
||
kill := DefaultCmiiOperator.DeploymentRestartByKill(cmiiEnv, appName)
|
||
assert.Equal(t, kill, true, "have unhealthy pod !")
|
||
|
||
check := DefaultCmiiOperator.DeploymentStatusCheck(cmiiEnv, appName, 180)
|
||
assert.Equal(t, check, true, "DeploymentStatusCheck failed !")
|
||
}
|
||
|
||
func TestUpdateCmiiDeploymentImageTag(t *testing.T) {
|
||
|
||
//计算20:00的时间
|
||
|
||
now := time.Now()
|
||
|
||
targetTime := time.Date(now.Year(), now.Month(), now.Day(), 13, 45, 00, 0, now.Location())
|
||
|
||
duration := time.Duration(0)
|
||
|
||
if !now.After(targetTime) {
|
||
// 计算到20:00的剩余时间
|
||
duration = targetTime.Sub(now)
|
||
} else {
|
||
// 如果当前时间已经过了20:00，那么就跳到下一天
|
||
//targetTime = targetTime.Add(24 * time.Hour)
|
||
}
|
||
|
||
fmt.Println(duration)
|
||
|
||
// 等待到20:00
|
||
//time.Sleep(duration)
|
||
|
||
cmiiEnv := config.Demo
|
||
|
||
appNameTagMap := map[string]string{
|
||
//"cmii-app-release": "6.1.0-122001",
|
||
"cmii-uav-device": "6.1.0-1125-123001",
|
||
//"cmii-uav-platform-pilot2-to-cloud": "6.1.0-32030-123101",
|
||
//"cmii-uav-industrial-portfolio": "6.1.0-20241125-121702",
|
||
}
|
||
|
||
for appName, newTag := range appNameTagMap {
|
||
ok, oldImageTag, newImageTag := UpdateCmiiDeploymentImageTag(cmiiEnv, appName, newTag)
|
||
|
||
if !ok {
|
||
log.ErrorF("update image tag failed !")
|
||
UpdateCmiiDeploymentImageTag(cmiiEnv, appName, oldImageTag)
|
||
}
|
||
utils.SplitLinePrint()
|
||
|
||
check := DefaultCmiiOperator.DeploymentStatusCheck(cmiiEnv, appName, 300)
|
||
assert.Equal(t, check, true, "deployment run failed!")
|
||
if !check {
|
||
UpdateCmiiDeploymentImageTag(cmiiEnv, appName, oldImageTag)
|
||
}
|
||
|
||
deploy := DefaultCmiiOperator.DeploymentOneInterface(cmiiEnv, appName)
|
||
|
||
// push message
|
||
message := pusher.CmiiUpdateMessage{
|
||
Namespace: cmiiEnv,
|
||
AppName: appName,
|
||
FromTag: oldImageTag,
|
||
ToTag: newImageTag,
|
||
Replicas: strconv.FormatInt(int64(deploy.Replicas), 10),
|
||
DeployStatus: check,
|
||
}
|
||
message.SendMessage()
|
||
|
||
}
|
||
|
||
}
|