Add SetupCleanup test for containerd

Signed-off-by: Evan Lezar <[email protected]>

Author: elezar

cmd/nvidia-ctk-installer/container/container.go

@@ -35,8 +35,9 @@ const (
 
 // Options defines the shared options for the CLIs to configure containers runtimes.
 type Options struct {
-	Config string
-	Socket string
+	DropInConfig string
+	Config       string
+	Socket       string
 	// ExecutablePath specifies the path to the container runtime executable.
 	// This is used to extract the current config, for example.
 	// If a HostRootMount is specified, this path is relative to the host root
@@ -71,8 +72,12 @@ func (o Options) Unconfigure(cfg engine.Interface) error {
 
 // flush flushes the specified config to disk
 func (o Options) flush(cfg engine.Interface) error {
-	logrus.Infof("Flushing config to %v", o.Config)
-	n, err := cfg.Save(o.Config)
+	filepath := o.DropInConfig
+	if filepath == "" {
+		filepath = o.Config
+	}
+	logrus.Infof("Flushing config to %v", filepath)
+	n, err := cfg.Save(filepath)
 	if err != nil {
 		return fmt.Errorf("unable to flush config: %v", err)
 	}

cmd/nvidia-ctk-installer/container/runtime/containerd/containerd.go

@@ -170,7 +170,7 @@ func GetLowlevelRuntimePaths(o *container.Options, co *Options) ([]string, error
 
 func getRuntimeConfig(o *container.Options, co *Options) (engine.Interface, error) {
 	return containerd.New(
-		containerd.WithPath(o.Config),
+		containerd.WithTopLevelConfigPath(o.Config),
 		containerd.WithConfigSource(
 			toml.LoadFirst(
 				containerd.CommandLineSource(o.HostRootMount, o.ExecutablePath),