validation/util/container: Remove bundle even if delete fails

When the container failed in creation, delete will fail, but we still
want to remove the test bundle directory.

Signed-off-by: W. Trevor King <[email protected]>

Author: wking

validation/create.go

@@ -13,6 +13,7 @@ import (
 
 func main() {
 	t := tap.New()
+	t.Header(0)
 
 	g := generate.New()
 	g.SetRootPath(".")
@@ -27,7 +28,7 @@ func main() {
 	if err != nil {
 		util.Fatal(err)
 	}
-	defer r.Clean(true)
+	defer r.Clean(true, true)
 
 	err = r.SetConfig(&g)
 	if err != nil {

validation/util/container.go

@@ -151,16 +151,19 @@ func (r *Runtime) Delete() error {
 	return cmd.Run()
 }
 
-// Clean deletes the container and removes the bundle file according to the input parameter
-func (r *Runtime) Clean(removeBundle bool) error {
+// Clean deletes the container.  If removeBundle is set, the bundle
+// directory is removed after the container is deleted succesfully or, if
+// forceRemoveBundle is true, after the deletion attempt regardless of
+// whether it was successful or not.
+func (r *Runtime) Clean(removeBundle bool, forceRemoveBundle bool) error {
 	err := r.Delete()
-	if err != nil {
-		return err
-	}
 
-	if removeBundle {
-		os.RemoveAll(r.BundleDir)
+	if removeBundle && (err == nil || forceRemoveBundle) {
+		err2 := os.RemoveAll(r.BundleDir)
+		if err2 != nil && err == nil {
+			err = err2
+		}
 	}
 
-	return nil
+	return err
 }

validation/util/test.go

@@ -92,7 +92,7 @@ func RuntimeInsideValidate(g *generate.Generator, f PreFunc) (err error) {
 		os.RemoveAll(bundleDir)
 		return err
 	}
-	defer r.Clean(true)
+	defer r.Clean(true, true)
 	err = r.SetConfig(g)
 	if err != nil {
 		return err