Add dktesting.Cleanup() method

Author: dhui

dktesting/dktesting.go

@@ -1,11 +1,15 @@
 package dktesting
 
 import (
+	"context"
+	"fmt"
 	"testing"
 )
 
 import (
 	"github.com/dhui/dktest"
+	"github.com/docker/docker/api/types"
+	"github.com/docker/docker/client"
 )
 
 // ContainerSpec holds Docker testing setup specifications
@@ -14,6 +18,26 @@ type ContainerSpec struct {
 	Options   dktest.Options
 }
 
+// Cleanup cleanups the ContainerSpec after a test run by removing the ContainerSpec's image
+func (s *ContainerSpec) Cleanup() (retErr error) {
+	// copied from dktest.RunContext()
+	dc, err := client.NewClientWithOpts(client.FromEnv, client.WithVersion("1.41"))
+	if err != nil {
+		return err
+	}
+	defer func() {
+		if err := dc.Close(); err != nil && retErr == nil {
+			retErr = fmt.Errorf("error closing Docker client: %w", err)
+		}
+	}()
+	ctx, timeoutCancelFunc := context.WithTimeout(context.Background(), s.Options.CleanupTimeout)
+	defer timeoutCancelFunc()
+	if _, err := dc.ImageRemove(ctx, s.ImageName, types.ImageRemoveOptions{Force: true, PruneChildren: true}); err != nil {
+		return err
+	}
+	return nil
+}
+
 // ParallelTest runs Docker tests in parallel
 func ParallelTest(t *testing.T, specs []ContainerSpec,
 	testFunc func(*testing.T, dktest.ContainerInfo)) {