[Agent][Deploy] - add zjyd

agent-operator/ImageSyncOperator_test.go

@@ -0,0 +1,121 @@
+package main
+
+import (
+	"testing"
+	image2 "wdd.io/agent-common/image"
+	"wdd.io/agent-common/utils"
+)
+
+/* 拉取
+ */
+
+func TestFetchDependencyRepos_Middle(t *testing.T) {
+
+	errorPullImageList, errorGzipImageList, realCmiiImageName, allGzipFileNameList := C_DownloadCompressUploadDependency(true, true, false, false)
+
+	utils.BeautifulPrintListWithTitle(errorPullImageList, "errorPullImageList")
+	utils.BeautifulPrintListWithTitle(errorGzipImageList, "errorGzipImageList")
+	utils.BeautifulPrintListWithTitle(realCmiiImageName, "realCmiiImageName")
+	utils.BeautifulPrintListWithTitle(allGzipFileNameList, "allGzipFileNameList")
+
+}
+
+func TestFetchDependencyRepos_RKE(t *testing.T) {
+
+	errorPullImageList, errorGzipImageList, realCmiiImageName, allGzipFileNameList := C_DownloadCompressUploadDependency(true, true, false, true)
+
+	utils.BeautifulPrintListWithTitle(errorPullImageList, "errorPullImageList")
+	utils.BeautifulPrintListWithTitle(errorGzipImageList, "errorGzipImageList")
+	utils.BeautifulPrintListWithTitle(realCmiiImageName, "realCmiiImageName")
+	utils.BeautifulPrintListWithTitle(allGzipFileNameList, "allGzipFileNameList")
+
+}
+
+func TestLoadSplitGzipImageToTargetHarbor(t *testing.T) {
+	errorLoadImageNameList, errorPushImageNameList := LoadSplitCmiiGzipImageToTargetHarbor("xmyd", DirectPushDeployHarborHost)
+
+	utils.BeautifulPrintListWithTitle(errorLoadImageNameList, "errorLoadImageNameList")
+	utils.BeautifulPrintListWithTitle(errorPushImageNameList, "errorPushImageNameList")
+}
+
+func TestLoadSplitDepGzipImageToTargetHarbor(t *testing.T) {
+	errorLoadImageNameList, errorPushImageNameList := LoadSplitDepGzipImageToTargetHarbor(DirectPushDeployHarborHost)
+
+	utils.BeautifulPrintListWithTitle(errorLoadImageNameList, "errorLoadImageNameList")
+	utils.BeautifulPrintListWithTitle(errorPushImageNameList, "errorPushImageNameList")
+}
+
+func TestPullFromEntityAndSyncConditionally(t *testing.T) {
+	// 创建一个模拟的sync对象，用于测试函数的行为。这里需要根据你的实际需求来设置mock数据和预期结果。
+	sync := ImageSyncEntity{
+		CmiiNameTagList: []string{
+			//"cmii-uav-mqtthandler:5.4.0-bjdyt-052102",
+		},
+		FullNameImageList:         nil,
+		ProjectVersion:            "",
+		ProjectName:               "bjtg",
+		DirectHarborHost:          "harbor.wdd.io:8033",
+		ShouldDownloadImage:       false,
+		ShouldCompressImageToGzip: false,
+		ShouldUploadToDemoMinio:   false,
+		ShouldDirectPushToHarbor:  true,
+	}
+
+	// 调用函数并获取结果。这里需要根据你的实际需求来验证返回的结果是否符合预期。
+	result := sync.PullFromEntityAndSyncConditionally()
+
+	utils.BeautifulPrint(result)
+
+	// 添加断言以检查函数的输出，例如：
+	if len(result.ErrorPullImageList) != 0 {
+		t.Errorf("Expected no error pulling images, got %v", result.ErrorPullImageList)
+	}
+	// ...其他验证逻辑...
+}
+
+func TestDownloadLoadTagUpload(t *testing.T) {
+
+	localGzipFileList := []string{
+		"/root/octopus_image/middle",
+		"/root/octopus_image/rke",
+		"/root/octopus_image/xjyd",
+	}
+
+	for _, s := range localGzipFileList {
+		log.InfoF("start to sync => %s", s)
+
+		targetImageFullNameList := A_DownloadLoadTagUpload(false, "", "", s, "172.28.0.251:8033")
+
+		utils.BeautifulPrintListWithTitle(targetImageFullNameList, "targetImageFullNameList")
+
+	}
+
+}
+
+func TestConcatAndUniformCmiiImage(t *testing.T) {
+	// 创建一个模拟的fullImageList和cmiiImageList，用于测试函数的行为。这里需要根据你的实际需求来设置mock数据和预期结果。
+	fullImageList := []string{"image3", "image4"}
+	cmiiImageList := []string{"image1", "image2"}
+
+	// 调用函数并获取结果。这里需要根据你的实际需求来验证返回的结果是否符合预期。
+	result := concatAndUniformCmiiImage(fullImageList, cmiiImageList)
+
+	// 添加断言以检查函数的输出，例如：
+	expectedResult := []string{"image3", "image4", image2.CmiiHarborPrefix + "image1", image2.CmiiHarborPrefix + "image2"}
+	if len(result) != len(expectedResult) {
+		t.Errorf("Expected %v, got %v", expectedResult, result)
+	}
+	// ...其他验证逻辑...
+}
+
+func TestImageSyncEntity_PullFromEntityAndSyncConditionally(t *testing.T) {
+	imageSyncEntity := ImageSyncEntity{
+		ProjectVersion:   "5.4.0",
+		DirectHarborHost: "36.134.71.138",
+	}
+
+	imageSyncResult := imageSyncEntity.PullFromEntityAndSyncConditionally()
+
+	utils.BeautifulPrint(imageSyncResult)
+
+}