[ Cmii ] [ Operator ] - agent operator for DCU part

agent-operator/CmiiOperator_test.go

@@ -2,32 +2,36 @@ package main
 
 import (
 	"testing"
+	image2 "wdd.io/agent-common/image"
 	"wdd.io/agent-common/utils"
 )
 
+/* 拉取
+ */
+
 func TestFetchDemoImages(t *testing.T) {
 
-	errorPullImageList, errorGzipImageList, allCmiiImageName := FetchDemoImages("shls", true)
-
-	utils.BeautifulPrintListWithTitle(errorPullImageList, "cmii errorPullImageList")
-	utils.BeautifulPrintListWithTitle(errorGzipImageList, "cmii errorGzipImageList")
-	utils.BeautifulPrintListWithTitle(allCmiiImageName, "cmii allCmiiImageName")
+	//errorPullImageList, errorGzipImageList, allCmiiImageName := DownloadCompressUploadFromDemo("shls", true)
+	//
+	//utils.BeautifulPrintListWithTitle(errorPullImageList, "cmii errorPullImageList")
+	//utils.BeautifulPrintListWithTitle(errorGzipImageList, "cmii errorGzipImageList")
+	//utils.BeautifulPrintListWithTitle(allCmiiImageName, "cmii allCmiiImageName")
 }
 
 func TestFetchVersionImages(t *testing.T) {
-	errorPullImageList, errorGzipImageList, allCmiiImageName := FetchVersionImages("5.4.0", true)
-
-	utils.BeautifulPrintListWithTitle(errorPullImageList, "cmii errorPullImageList")
-	utils.BeautifulPrintListWithTitle(errorGzipImageList, "cmii errorGzipImageList")
-	utils.BeautifulPrintListWithTitle(allCmiiImageName, "cmii allCmiiImageName")
+	//errorPullImageList, errorGzipImageList, allCmiiImageName := DownloadCompressUploadFromVersion("5.4.0", true)
+	//
+	//utils.BeautifulPrintListWithTitle(errorPullImageList, "cmii errorPullImageList")
+	//utils.BeautifulPrintListWithTitle(errorGzipImageList, "cmii errorGzipImageList")
+	//utils.BeautifulPrintListWithTitle(allCmiiImageName, "cmii allCmiiImageName")
 }
 
 func TestFetchDependencyRepos(t *testing.T) {
 
-	errorPullImageList, errorGzipImageList := FetchDependencyRepos(true)
-
-	utils.BeautifulPrintListWithTitle(errorPullImageList, "dep errorPullImageList")
-	utils.BeautifulPrintListWithTitle(errorGzipImageList, "dep errorGzipImageList")
+	//errorPullImageList, errorGzipImageList := FetchDependencyRepos(true)
+	//
+	//utils.BeautifulPrintListWithTitle(errorPullImageList, "dep errorPullImageList")
+	//utils.BeautifulPrintListWithTitle(errorGzipImageList, "dep errorGzipImageList")
 }
 
 func TestLoadSplitGzipImageToTargetHarbor(t *testing.T) {
@@ -44,6 +48,48 @@ func TestLoadSplitDepGzipImageToTargetHarbor(t *testing.T) {
 	utils.BeautifulPrintListWithTitle(errorPushImageNameList, "errorPushImageNameList")
 }
 
+func TestPullFromEntityAndSyncConditionally(t *testing.T) {
+	// 创建一个模拟的sync对象，用于测试函数的行为。这里需要根据你的实际需求来设置mock数据和预期结果。
+	sync := ImageSyncEntity{
+		CmiiNameTagList: []string{
+			"cmii-uav-gateway:5.4.0",
+		},
+		FullNameImageList:        nil,
+		ProjectVersion:           "",
+		DirectHarborHost:         "",
+		CompressImageToGzip:      false,
+		UploadToDemoMinio:        false,
+		ShouldDirectPushToHarbor: false,
+	}
+
+	// 调用函数并获取结果。这里需要根据你的实际需求来验证返回的结果是否符合预期。
+	result := sync.PullFromEntityAndSyncConditionally()
+
+	utils.BeautifulPrint(sync)
+
+	// 添加断言以检查函数的输出，例如：
+	if len(result.ErrorPullImageList) != 0 {
+		t.Errorf("Expected no error pulling images, got %v", result.ErrorPullImageList)
+	}
+	// ...其他验证逻辑...
+}
+
+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",