updated parser unit test

Azure · Jan 28, 2025 · 1d26db3 · 1d26db3
1 parent 5a955ae
commit 1d26db3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/aks-node-controller/parser/parser_test.go b/aks-node-controller/parser/parser_test.go
@@ -47,12 +47,12 @@ func TestBuildCSECmd(t *testing.T) {
 			validator: func(cmd *exec.Cmd) {
 				vars := environToMap(cmd.Env)
 				assert.Equal(t, "false", vars["GPU_NODE"])
-				assert.NotEmpty(t, vars["CONTAINERD_CONFIG_CONTENT"])
+				assert.NotEmpty(t, vars["CONTAINERD_CONFIG_NO_GPU_CONTENT"])
 				// Ensure the containerd config does not use the
 				// nvidia container runtime when skipping the
 				// GPU driver install, since it will fail to run even non-GPU
 				// pods, as it will not be installed.
-				containerdConfigFileContent, err := getBase64DecodedValue([]byte(vars["CONTAINERD_CONFIG_CONTENT"]))
+				containerdConfigFileContent, err := getBase64DecodedValue([]byte(vars["CONTAINERD_CONFIG_NO_GPU_CONTENT"]))
 				require.NoError(t, err)
 				expectedShimConfig := `version = 2
 oom_score = 0

diff --git a/aks-node-controller/parser/templates/containerd_no_GPU.toml.gtpl b/aks-node-controller/parser/templates/containerd_no_GPU.toml.gtpl
@@ -102,4 +102,4 @@ root = "{{.KubeletConfig.GetContainerDataDir}}"{{- end}}
   pod_annotations = ["io.katacontainers.*"]
   [plugins."io.containerd.grpc.v1.cri".containerd.runtimes.kata-cc.options]
     ConfigPath = "/opt/confidential-containers/share/defaults/kata-containers/configuration-clh-snp.toml"
-{{- end}}
+{{- end}}