fixup! USE Matchers

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

Author: elezar

tests/e2e/nvidia-ctk_containerd_test.go

@@ -614,18 +614,8 @@ func getPluginConfig(tree *toml.Tree, version int64) (*toml.Tree, error) {
 }
 
 // getRuntimesConfig gets the runtimes configuration from the plugin config
-func getRuntimesConfig(pluginConfig *toml.Tree, _ int64) (map[string]interface{}, error) {
-	containerdSection := pluginConfig.Get("containerd")
-	if containerdSection == nil {
-		return nil, fmt.Errorf("containerd section not found")
-	}
-
-	containerdTree, ok := containerdSection.(*toml.Tree)
-	if !ok {
-		return nil, fmt.Errorf("containerd section is not a TOML tree")
-	}
-
-	runtimes := containerdTree.Get("runtimes")
+func getRuntimesConfig(pluginConfig *toml.Tree) (map[string]interface{}, error) {
+	runtimes := pluginConfig.GetPath([]string{"containerd", "runtimes"})
 	if runtimes == nil {
 		return nil, fmt.Errorf("runtimes section not found")
 	}