test(helper): add coverage on getTargetEnvironment method (#5639)

Co-authored-by: Jessie Wang <[email protected]>
Azure · Jan 23, 2025 · c16abcc · c16abcc
1 parent cfe184d
commit c16abcc
Showing 1 changed file with 36 additions and 0 deletions.
diff --git a/aks-node-controller/parser/helper_test.go b/aks-node-controller/parser/helper_test.go
@@ -884,6 +884,42 @@ func Test_getTargetEnvironment(t *testing.T) {
 			},
 			want: helpers.AksCustomCloudName,
 		},
+		{
+			name: "China location cluster config",
+			args: args{
+				v: &aksnodeconfigv1.Configuration{
+					CustomCloudConfig: &aksnodeconfigv1.CustomCloudConfig{},
+					ClusterConfig: &aksnodeconfigv1.ClusterConfig{
+						Location: "china",
+					},
+				},
+			},
+			want: "AzureChinaCloud",
+		},
+		{
+			name: "Germany location cluster config",
+			args: args{
+				v: &aksnodeconfigv1.Configuration{
+					CustomCloudConfig: &aksnodeconfigv1.CustomCloudConfig{},
+					ClusterConfig: &aksnodeconfigv1.ClusterConfig{
+						Location: "germanynortheast",
+					},
+				},
+			},
+			want: "AzureGermanCloud",
+		},
+		{
+			name: "usgov location cluster config",
+			args: args{
+				v: &aksnodeconfigv1.Configuration{
+					CustomCloudConfig: &aksnodeconfigv1.CustomCloudConfig{},
+					ClusterConfig: &aksnodeconfigv1.ClusterConfig{
+						Location: "usdod",
+					},
+				},
+			},
+			want: "AzureUSGovernmentCloud",
+		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {