diff --git a/aks-node-controller/parser/helper_test.go b/aks-node-controller/parser/helper_test.go
index c47c520af63..74a468e25be 100644
--- 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) {