minor fixes for tests to run

NVIDIA · Nov 18, 2024 · 91269ac · 91269ac
1 parent 4f23c97
commit 91269ac
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/pkg/config/engine/containerd/containerd.go b/pkg/config/engine/containerd/containerd.go
@@ -72,15 +72,17 @@ func New(opts ...Option) (engine.Interface, error) {
 		return nil, fmt.Errorf("failed to load config: %v", err)
 	}
 
+	useLegacyConfig := b.configVersion == 1
+
 	cfg := &Config{
 		Tree:                  tomlConfig,
 		Logger:                b.logger,
 		RuntimeType:           b.runtimeType,
-		UseDefaultRuntimeName: b.useLegacyConfig,
+		UseDefaultRuntimeName: useLegacyConfig,
 		ContainerAnnotations:  b.containerAnnotations,
 	}
 
-	version, err := cfg.parseVersion(b.useLegacyConfig)
+	version, err := cfg.parseVersion(useLegacyConfig)
 	if err != nil {
 		return nil, fmt.Errorf("failed to parse config version: %v", err)
 	}

diff --git a/tools/container/runtime/containerd/config_v3_test.go b/tools/container/runtime/containerd/config_v3_test.go
@@ -29,7 +29,7 @@ import (
 )
 
 const (
-	runtimeType = "runtime_type"
+	runtimeTypeV3 = "runtime_type"
 )
 
 func TestUpdateV3ConfigDefaultRuntime(t *testing.T) {
@@ -78,7 +78,7 @@ func TestUpdateV3ConfigDefaultRuntime(t *testing.T) {
 			v3 := &containerd.ConfigV3{
 				Logger:      logger,
 				Tree:        cfg,
-				RuntimeType: runtimeType,
+				RuntimeType: runtimeTypeV3,
 			}
 
 			err = o.UpdateConfig(v3)
@@ -201,7 +201,7 @@ func TestUpdateV3Config(t *testing.T) {
 			v3 := &containerd.ConfigV3{
 				Logger:               logger,
 				Tree:                 cfg,
-				RuntimeType:          runtimeType,
+				RuntimeType:          runtimeTypeV3,
 				ContainerAnnotations: []string{"cdi.k8s.io/*"},
 			}
 
@@ -354,7 +354,7 @@ func TestUpdateV3ConfigWithRuncPresent(t *testing.T) {
 			v3 := &containerd.ConfigV3{
 				Logger:               logger,
 				Tree:                 cfg,
-				RuntimeType:          runtimeType,
+				RuntimeType:          runtimeTypeV3,
 				ContainerAnnotations: []string{"cdi.k8s.io/*"},
 			}
 
@@ -426,7 +426,7 @@ func TestRevertV3Config(t *testing.T) {
 
 			v3 := &containerd.ConfigV3{
 				Tree:        cfg,
-				RuntimeType: runtimeType,
+				RuntimeType: runtimeTypeV3,
 			}
 
 			err = o.RevertConfig(v3)
@@ -442,7 +442,7 @@ func TestRevertV3Config(t *testing.T) {
 
 func runtimeMapV3(binary string) map[string]interface{} {
 	return map[string]interface{}{
-		"runtime_type":                    runtimeType,
+		"runtime_type":                    runtimeTypeV3,
 		"runtime_root":                    "",
 		"runtime_engine":                  "",
 		"privileged_without_host_devices": false,