testutil: use testing.TB instead of assert.TestingT

Signed-off-by: Sebastiaan van Stijn <github@gone.nl>

testutil/daemon/config.go

@@ -2,10 +2,10 @@ package daemon
 
 import (
 	"context"
+	"testing"
 
 	"github.com/docker/docker/api/types"
 	"github.com/docker/docker/api/types/swarm"
-	"github.com/docker/docker/testutil"
 	"gotest.tools/assert"
 )
 
@@ -13,10 +13,8 @@ import (
 type ConfigConstructor func(*swarm.Config)
 
 // CreateConfig creates a config given the specified spec
-func (d *Daemon) CreateConfig(t assert.TestingT, configSpec swarm.ConfigSpec) string {
+func (d *Daemon) CreateConfig(t testing.TB, configSpec swarm.ConfigSpec) string {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -26,10 +24,8 @@ func (d *Daemon) CreateConfig(t assert.TestingT, configSpec swarm.ConfigSpec) st
 }
 
 // ListConfigs returns the list of the current swarm configs
-func (d *Daemon) ListConfigs(t assert.TestingT) []swarm.Config {
+func (d *Daemon) ListConfigs(t testing.TB) []swarm.Config {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -39,10 +35,8 @@ func (d *Daemon) ListConfigs(t assert.TestingT) []swarm.Config {
 }
 
 // GetConfig returns a swarm config identified by the specified id
-func (d *Daemon) GetConfig(t assert.TestingT, id string) *swarm.Config {
+func (d *Daemon) GetConfig(t testing.TB, id string) *swarm.Config {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -52,10 +46,8 @@ func (d *Daemon) GetConfig(t assert.TestingT, id string) *swarm.Config {
 }
 
 // DeleteConfig removes the swarm config identified by the specified id
-func (d *Daemon) DeleteConfig(t assert.TestingT, id string) {
+func (d *Daemon) DeleteConfig(t testing.TB, id string) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -65,10 +57,8 @@ func (d *Daemon) DeleteConfig(t assert.TestingT, id string) {
 
 // UpdateConfig updates the swarm config identified by the specified id
 // Currently, only label update is supported.
-func (d *Daemon) UpdateConfig(t assert.TestingT, id string, f ...ConfigConstructor) {
+func (d *Daemon) UpdateConfig(t testing.TB, id string, f ...ConfigConstructor) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 

testutil/daemon/container.go

@@ -2,17 +2,15 @@ package daemon
 
 import (
 	"context"
+	"testing"
 
 	"github.com/docker/docker/api/types"
-	"github.com/docker/docker/testutil"
 	"gotest.tools/assert"
 )
 
 // ActiveContainers returns the list of ids of the currently running containers
-func (d *Daemon) ActiveContainers(t assert.TestingT) []string {
+func (d *Daemon) ActiveContainers(t testing.TB) []string {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -27,10 +25,8 @@ func (d *Daemon) ActiveContainers(t assert.TestingT) []string {
 }
 
 // FindContainerIP returns the ip of the specified container
-func (d *Daemon) FindContainerIP(t assert.TestingT, id string) string {
+func (d *Daemon) FindContainerIP(t testing.TB, id string) string {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 

testutil/daemon/daemon.go

@@ -21,7 +21,6 @@ import (
 	"github.com/docker/docker/opts"
 	"github.com/docker/docker/pkg/ioutils"
 	"github.com/docker/docker/pkg/stringid"
-	"github.com/docker/docker/testutil"
 	"github.com/docker/docker/testutil/request"
 	"github.com/docker/go-connections/sockets"
 	"github.com/docker/go-connections/tlsconfig"
@@ -34,7 +33,7 @@ type logT interface {
 }
 
 // nopLog is a no-op implementation of logT that is used in daemons created by
-// NewDaemon (where no testingT is available).
+// NewDaemon (where no testing.TB is available).
 type nopLog struct{}
 
 func (nopLog) Logf(string, ...interface{}) {}
@@ -194,10 +193,8 @@ func (d *Daemon) ReadLogFile() ([]byte, error) {
 }
 
 // NewClientT creates new client based on daemon's socket path
-func (d *Daemon) NewClientT(t assert.TestingT, extraOpts ...client.Opt) *client.Client {
+func (d *Daemon) NewClientT(t testing.TB, extraOpts ...client.Opt) *client.Client {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 
 	c, err := d.NewClient(extraOpts...)
 	assert.NilError(t, err, "cannot create daemon client")
@@ -595,10 +592,8 @@ func (d *Daemon) ReloadConfig() error {
 }
 
 // LoadBusybox image into the daemon
-func (d *Daemon) LoadBusybox(t assert.TestingT) {
+func (d *Daemon) LoadBusybox(t testing.TB) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	clientHost, err := client.NewClientWithOpts(client.FromEnv)
 	assert.NilError(t, err, "failed to create client")
 	defer clientHost.Close()
@@ -709,10 +704,8 @@ func (d *Daemon) queryRootDir() (string, error) {
 }
 
 // Info returns the info struct for this daemon
-func (d *Daemon) Info(t assert.TestingT) types.Info {
+func (d *Daemon) Info(t testing.TB) types.Info {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	c := d.NewClientT(t)
 	info, err := c.Info(context.Background())
 	assert.NilError(t, err)

testutil/daemon/node.go

@@ -3,11 +3,11 @@ package daemon
 import (
 	"context"
 	"strings"
+	"testing"
 	"time"
 
 	"github.com/docker/docker/api/types"
 	"github.com/docker/docker/api/types/swarm"
-	"github.com/docker/docker/testutil"
 	"gotest.tools/assert"
 )
 
@@ -15,10 +15,8 @@ import (
 type NodeConstructor func(*swarm.Node)
 
 // GetNode returns a swarm node identified by the specified id
-func (d *Daemon) GetNode(t assert.TestingT, id string, errCheck ...func(error) bool) *swarm.Node {
+func (d *Daemon) GetNode(t testing.TB, id string, errCheck ...func(error) bool) *swarm.Node {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -36,10 +34,8 @@ func (d *Daemon) GetNode(t assert.TestingT, id string, errCheck ...func(error) b
 }
 
 // RemoveNode removes the specified node
-func (d *Daemon) RemoveNode(t assert.TestingT, id string, force bool) {
+func (d *Daemon) RemoveNode(t testing.TB, id string, force bool) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -51,10 +47,8 @@ func (d *Daemon) RemoveNode(t assert.TestingT, id string, force bool) {
 }
 
 // UpdateNode updates a swarm node with the specified node constructor
-func (d *Daemon) UpdateNode(t assert.TestingT, id string, f ...NodeConstructor) {
+func (d *Daemon) UpdateNode(t testing.TB, id string, f ...NodeConstructor) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -75,10 +69,8 @@ func (d *Daemon) UpdateNode(t assert.TestingT, id string, f ...NodeConstructor)
 }
 
 // ListNodes returns the list of the current swarm nodes
-func (d *Daemon) ListNodes(t assert.TestingT) []swarm.Node {
+func (d *Daemon) ListNodes(t testing.TB) []swarm.Node {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 

testutil/daemon/plugin.go

@@ -2,15 +2,15 @@ package daemon
 
 import (
 	"context"
+	"testing"
 
 	"github.com/docker/docker/api/types"
 	"github.com/docker/docker/client"
-	"gotest.tools/assert"
 	"gotest.tools/poll"
 )
 
 // PluginIsRunning provides a poller to check if the specified plugin is running
-func (d *Daemon) PluginIsRunning(t assert.TestingT, name string) func(poll.LogT) poll.Result {
+func (d *Daemon) PluginIsRunning(t testing.TB, name string) func(poll.LogT) poll.Result {
 	return withClient(t, d, withPluginInspect(name, func(plugin *types.Plugin, t poll.LogT) poll.Result {
 		if plugin.Enabled {
 			return poll.Success()
@@ -20,7 +20,7 @@ func (d *Daemon) PluginIsRunning(t assert.TestingT, name string) func(poll.LogT)
 }
 
 // PluginIsNotRunning provides a poller to check if the specified plugin is not running
-func (d *Daemon) PluginIsNotRunning(t assert.TestingT, name string) func(poll.LogT) poll.Result {
+func (d *Daemon) PluginIsNotRunning(t testing.TB, name string) func(poll.LogT) poll.Result {
 	return withClient(t, d, withPluginInspect(name, func(plugin *types.Plugin, t poll.LogT) poll.Result {
 		if !plugin.Enabled {
 			return poll.Success()
@@ -30,7 +30,7 @@ func (d *Daemon) PluginIsNotRunning(t assert.TestingT, name string) func(poll.Lo
 }
 
 // PluginIsNotPresent provides a poller to check if the specified plugin is not present
-func (d *Daemon) PluginIsNotPresent(t assert.TestingT, name string) func(poll.LogT) poll.Result {
+func (d *Daemon) PluginIsNotPresent(t testing.TB, name string) func(poll.LogT) poll.Result {
 	return withClient(t, d, func(c client.APIClient, t poll.LogT) poll.Result {
 		_, _, err := c.PluginInspectWithRaw(context.Background(), name)
 		if client.IsErrNotFound(err) {
@@ -44,7 +44,7 @@ func (d *Daemon) PluginIsNotPresent(t assert.TestingT, name string) func(poll.Lo
 }
 
 // PluginReferenceIs provides a poller to check if the specified plugin has the specified reference
-func (d *Daemon) PluginReferenceIs(t assert.TestingT, name, expectedRef string) func(poll.LogT) poll.Result {
+func (d *Daemon) PluginReferenceIs(t testing.TB, name, expectedRef string) func(poll.LogT) poll.Result {
 	return withClient(t, d, withPluginInspect(name, func(plugin *types.Plugin, t poll.LogT) poll.Result {
 		if plugin.PluginReference == expectedRef {
 			return poll.Success()
@@ -67,7 +67,7 @@ func withPluginInspect(name string, f func(*types.Plugin, poll.LogT) poll.Result
 
 }
 
-func withClient(t assert.TestingT, d *Daemon, f func(client.APIClient, poll.LogT) poll.Result) func(poll.LogT) poll.Result {
+func withClient(t testing.TB, d *Daemon, f func(client.APIClient, poll.LogT) poll.Result) func(poll.LogT) poll.Result {
 	return func(pt poll.LogT) poll.Result {
 		c := d.NewClientT(t)
 		return f(c, pt)

testutil/daemon/secret.go

@@ -2,10 +2,10 @@ package daemon
 
 import (
 	"context"
+	"testing"
 
 	"github.com/docker/docker/api/types"
 	"github.com/docker/docker/api/types/swarm"
-	"github.com/docker/docker/testutil"
 	"gotest.tools/assert"
 )
 
@@ -13,10 +13,8 @@ import (
 type SecretConstructor func(*swarm.Secret)
 
 // CreateSecret creates a secret given the specified spec
-func (d *Daemon) CreateSecret(t assert.TestingT, secretSpec swarm.SecretSpec) string {
+func (d *Daemon) CreateSecret(t testing.TB, secretSpec swarm.SecretSpec) string {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -27,10 +25,8 @@ func (d *Daemon) CreateSecret(t assert.TestingT, secretSpec swarm.SecretSpec) st
 }
 
 // ListSecrets returns the list of the current swarm secrets
-func (d *Daemon) ListSecrets(t assert.TestingT) []swarm.Secret {
+func (d *Daemon) ListSecrets(t testing.TB) []swarm.Secret {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -40,10 +36,8 @@ func (d *Daemon) ListSecrets(t assert.TestingT) []swarm.Secret {
 }
 
 // GetSecret returns a swarm secret identified by the specified id
-func (d *Daemon) GetSecret(t assert.TestingT, id string) *swarm.Secret {
+func (d *Daemon) GetSecret(t testing.TB, id string) *swarm.Secret {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -53,10 +47,8 @@ func (d *Daemon) GetSecret(t assert.TestingT, id string) *swarm.Secret {
 }
 
 // DeleteSecret removes the swarm secret identified by the specified id
-func (d *Daemon) DeleteSecret(t assert.TestingT, id string) {
+func (d *Daemon) DeleteSecret(t testing.TB, id string) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -66,10 +58,8 @@ func (d *Daemon) DeleteSecret(t assert.TestingT, id string) {
 
 // UpdateSecret updates the swarm secret identified by the specified id
 // Currently, only label update is supported.
-func (d *Daemon) UpdateSecret(t assert.TestingT, id string, f ...SecretConstructor) {
+func (d *Daemon) UpdateSecret(t testing.TB, id string, f ...SecretConstructor) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 

testutil/daemon/service.go

@@ -2,22 +2,20 @@ package daemon
 
 import (
 	"context"
+	"testing"
 	"time"
 
 	"github.com/docker/docker/api/types"
 	"github.com/docker/docker/api/types/filters"
 	"github.com/docker/docker/api/types/swarm"
-	"github.com/docker/docker/testutil"
 	"gotest.tools/assert"
 )
 
 // ServiceConstructor defines a swarm service constructor function
 type ServiceConstructor func(*swarm.Service)
 
-func (d *Daemon) createServiceWithOptions(t assert.TestingT, opts types.ServiceCreateOptions, f ...ServiceConstructor) string {
+func (d *Daemon) createServiceWithOptions(t testing.TB, opts types.ServiceCreateOptions, f ...ServiceConstructor) string {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	var service swarm.Service
 	for _, fn := range f {
 		fn(&service)
@@ -35,18 +33,14 @@ func (d *Daemon) createServiceWithOptions(t assert.TestingT, opts types.ServiceC
 }
 
 // CreateService creates a swarm service given the specified service constructor
-func (d *Daemon) CreateService(t assert.TestingT, f ...ServiceConstructor) string {
+func (d *Daemon) CreateService(t testing.TB, f ...ServiceConstructor) string {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	return d.createServiceWithOptions(t, types.ServiceCreateOptions{}, f...)
 }
 
 // GetService returns the swarm service corresponding to the specified id
-func (d *Daemon) GetService(t assert.TestingT, id string) *swarm.Service {
+func (d *Daemon) GetService(t testing.TB, id string) *swarm.Service {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -56,10 +50,8 @@ func (d *Daemon) GetService(t assert.TestingT, id string) *swarm.Service {
 }
 
 // GetServiceTasks returns the swarm tasks for the specified service
-func (d *Daemon) GetServiceTasks(t assert.TestingT, service string, additionalFilters ...filters.KeyValuePair) []swarm.Task {
+func (d *Daemon) GetServiceTasks(t testing.TB, service string, additionalFilters ...filters.KeyValuePair) []swarm.Task {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -80,10 +72,8 @@ func (d *Daemon) GetServiceTasks(t assert.TestingT, service string, additionalFi
 }
 
 // UpdateService updates a swarm service with the specified service constructor
-func (d *Daemon) UpdateService(t assert.TestingT, service *swarm.Service, f ...ServiceConstructor) {
+func (d *Daemon) UpdateService(t testing.TB, service *swarm.Service, f ...ServiceConstructor) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -96,10 +86,8 @@ func (d *Daemon) UpdateService(t assert.TestingT, service *swarm.Service, f ...S
 }
 
 // RemoveService removes the specified service
-func (d *Daemon) RemoveService(t assert.TestingT, id string) {
+func (d *Daemon) RemoveService(t testing.TB, id string) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -108,10 +96,8 @@ func (d *Daemon) RemoveService(t assert.TestingT, id string) {
 }
 
 // ListServices returns the list of the current swarm services
-func (d *Daemon) ListServices(t assert.TestingT) []swarm.Service {
+func (d *Daemon) ListServices(t testing.TB) []swarm.Service {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -121,10 +107,8 @@ func (d *Daemon) ListServices(t assert.TestingT) []swarm.Service {
 }
 
 // GetTask returns the swarm task identified by the specified id
-func (d *Daemon) GetTask(t assert.TestingT, id string) swarm.Task {
+func (d *Daemon) GetTask(t testing.TB, id string) swarm.Task {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 

testutil/daemon/swarm.go

@@ -6,7 +6,6 @@ import (
 	"testing"
 
 	"github.com/docker/docker/api/types/swarm"
-	"github.com/docker/docker/testutil"
 	"github.com/pkg/errors"
 	"gotest.tools/assert"
 )
@@ -78,10 +77,8 @@ func (d *Daemon) NodeID() string {
 }
 
 // SwarmInit initializes a new swarm cluster.
-func (d *Daemon) SwarmInit(t assert.TestingT, req swarm.InitRequest) {
+func (d *Daemon) SwarmInit(t testing.TB, req swarm.InitRequest) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	if req.ListenAddr == "" {
 		req.ListenAddr = fmt.Sprintf("%s:%d", d.swarmListenAddr, d.SwarmPort)
 	}
@@ -100,10 +97,8 @@ func (d *Daemon) SwarmInit(t assert.TestingT, req swarm.InitRequest) {
 }
 
 // SwarmJoin joins a daemon to an existing cluster.
-func (d *Daemon) SwarmJoin(t assert.TestingT, req swarm.JoinRequest) {
+func (d *Daemon) SwarmJoin(t testing.TB, req swarm.JoinRequest) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	if req.ListenAddr == "" {
 		req.ListenAddr = fmt.Sprintf("%s:%d", d.swarmListenAddr, d.SwarmPort)
 	}
@@ -116,20 +111,18 @@ func (d *Daemon) SwarmJoin(t assert.TestingT, req swarm.JoinRequest) {
 
 // SwarmLeave forces daemon to leave current cluster.
 //
-// The passed in TestingT is only used to validate that the client was successfully created
+// The passed in testing.TB is only used to validate that the client was successfully created
 // Some tests rely on error checking the result of the actual unlock, so allow
 // the error to be returned.
-func (d *Daemon) SwarmLeave(t assert.TestingT, force bool) error {
+func (d *Daemon) SwarmLeave(t testing.TB, force bool) error {
 	cli := d.NewClientT(t)
 	defer cli.Close()
 	return cli.SwarmLeave(context.Background(), force)
 }
 
 // SwarmInfo returns the swarm information of the daemon
-func (d *Daemon) SwarmInfo(t assert.TestingT) swarm.Info {
+func (d *Daemon) SwarmInfo(t testing.TB) swarm.Info {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	info, err := cli.Info(context.Background())
 	assert.NilError(t, err, "get swarm info")
@@ -138,10 +131,10 @@ func (d *Daemon) SwarmInfo(t assert.TestingT) swarm.Info {
 
 // SwarmUnlock tries to unlock a locked swarm
 //
-// The passed in TestingT is only used to validate that the client was successfully created
+// The passed in testing.TB is only used to validate that the client was successfully created
 // Some tests rely on error checking the result of the actual unlock, so allow
 // the error to be returned.
-func (d *Daemon) SwarmUnlock(t assert.TestingT, req swarm.UnlockRequest) error {
+func (d *Daemon) SwarmUnlock(t testing.TB, req swarm.UnlockRequest) error {
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -153,10 +146,8 @@ func (d *Daemon) SwarmUnlock(t assert.TestingT, req swarm.UnlockRequest) error {
 }
 
 // GetSwarm returns the current swarm object
-func (d *Daemon) GetSwarm(t assert.TestingT) swarm.Swarm {
+func (d *Daemon) GetSwarm(t testing.TB) swarm.Swarm {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -166,10 +157,8 @@ func (d *Daemon) GetSwarm(t assert.TestingT) swarm.Swarm {
 }
 
 // UpdateSwarm updates the current swarm object with the specified spec constructors
-func (d *Daemon) UpdateSwarm(t assert.TestingT, f ...SpecConstructor) {
+func (d *Daemon) UpdateSwarm(t testing.TB, f ...SpecConstructor) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -183,10 +172,8 @@ func (d *Daemon) UpdateSwarm(t assert.TestingT, f ...SpecConstructor) {
 }
 
 // RotateTokens update the swarm to rotate tokens
-func (d *Daemon) RotateTokens(t assert.TestingT) {
+func (d *Daemon) RotateTokens(t testing.TB) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 
@@ -203,10 +190,8 @@ func (d *Daemon) RotateTokens(t assert.TestingT) {
 }
 
 // JoinTokens returns the current swarm join tokens
-func (d *Daemon) JoinTokens(t assert.TestingT) swarm.JoinTokens {
+func (d *Daemon) JoinTokens(t testing.TB) swarm.JoinTokens {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	cli := d.NewClientT(t)
 	defer cli.Close()
 

testutil/environment/clean.go

@@ -4,21 +4,19 @@ import (
 	"context"
 	"regexp"
 	"strings"
+	"testing"
 
 	"github.com/docker/docker/api/types"
 	"github.com/docker/docker/api/types/filters"
 	"github.com/docker/docker/client"
-	"github.com/docker/docker/testutil"
 	"gotest.tools/assert"
 )
 
 // Clean the environment, preserving protected objects (images, containers, ...)
 // and removing everything else. It's meant to run after any tests so that they don't
 // depend on each others.
-func (e *Execution) Clean(t assert.TestingT) {
+func (e *Execution) Clean(t testing.TB) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	client := e.APIClient()
 
 	platform := e.OSType
@@ -34,10 +32,8 @@ func (e *Execution) Clean(t assert.TestingT) {
 	}
 }
 
-func unpauseAllContainers(t assert.TestingT, client client.ContainerAPIClient) {
+func unpauseAllContainers(t testing.TB, client client.ContainerAPIClient) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	ctx := context.Background()
 	containers := getPausedContainers(ctx, t, client)
 	if len(containers) > 0 {
@@ -48,10 +44,8 @@ func unpauseAllContainers(t assert.TestingT, client client.ContainerAPIClient) {
 	}
 }
 
-func getPausedContainers(ctx context.Context, t assert.TestingT, client client.ContainerAPIClient) []types.Container {
+func getPausedContainers(ctx context.Context, t testing.TB, client client.ContainerAPIClient) []types.Container {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	filter := filters.NewArgs()
 	filter.Add("status", "paused")
 	containers, err := client.ContainerList(ctx, types.ContainerListOptions{
@@ -65,10 +59,8 @@ func getPausedContainers(ctx context.Context, t assert.TestingT, client client.C
 
 var alreadyExists = regexp.MustCompile(`Error response from daemon: removal of container (\w+) is already in progress`)
 
-func deleteAllContainers(t assert.TestingT, apiclient client.ContainerAPIClient, protectedContainers map[string]struct{}) {
+func deleteAllContainers(t testing.TB, apiclient client.ContainerAPIClient, protectedContainers map[string]struct{}) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	ctx := context.Background()
 	containers := getAllContainers(ctx, t, apiclient)
 	if len(containers) == 0 {
@@ -90,10 +82,8 @@ func deleteAllContainers(t assert.TestingT, apiclient client.ContainerAPIClient,
 	}
 }
 
-func getAllContainers(ctx context.Context, t assert.TestingT, client client.ContainerAPIClient) []types.Container {
+func getAllContainers(ctx context.Context, t testing.TB, client client.ContainerAPIClient) []types.Container {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	containers, err := client.ContainerList(ctx, types.ContainerListOptions{
 		Quiet: true,
 		All:   true,
@@ -102,10 +92,8 @@ func getAllContainers(ctx context.Context, t assert.TestingT, client client.Cont
 	return containers
 }
 
-func deleteAllImages(t assert.TestingT, apiclient client.ImageAPIClient, protectedImages map[string]struct{}) {
+func deleteAllImages(t testing.TB, apiclient client.ImageAPIClient, protectedImages map[string]struct{}) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	images, err := apiclient.ImageList(context.Background(), types.ImageListOptions{})
 	assert.Check(t, err, "failed to list images")
 
@@ -124,10 +112,8 @@ func deleteAllImages(t assert.TestingT, apiclient client.ImageAPIClient, protect
 	}
 }
 
-func removeImage(ctx context.Context, t assert.TestingT, apiclient client.ImageAPIClient, ref string) {
+func removeImage(ctx context.Context, t testing.TB, apiclient client.ImageAPIClient, ref string) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	_, err := apiclient.ImageRemove(ctx, ref, types.ImageRemoveOptions{
 		Force: true,
 	})
@@ -137,10 +123,8 @@ func removeImage(ctx context.Context, t assert.TestingT, apiclient client.ImageA
 	assert.Check(t, err, "failed to remove image %s", ref)
 }
 
-func deleteAllVolumes(t assert.TestingT, c client.VolumeAPIClient, protectedVolumes map[string]struct{}) {
+func deleteAllVolumes(t testing.TB, c client.VolumeAPIClient, protectedVolumes map[string]struct{}) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	volumes, err := c.VolumeList(context.Background(), filters.Args{})
 	assert.Check(t, err, "failed to list volumes")
 
@@ -157,10 +141,8 @@ func deleteAllVolumes(t assert.TestingT, c client.VolumeAPIClient, protectedVolu
 	}
 }
 
-func deleteAllNetworks(t assert.TestingT, c client.NetworkAPIClient, daemonPlatform string, protectedNetworks map[string]struct{}) {
+func deleteAllNetworks(t testing.TB, c client.NetworkAPIClient, daemonPlatform string, protectedNetworks map[string]struct{}) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	networks, err := c.NetworkList(context.Background(), types.NetworkListOptions{})
 	assert.Check(t, err, "failed to list networks")
 
@@ -180,10 +162,8 @@ func deleteAllNetworks(t assert.TestingT, c client.NetworkAPIClient, daemonPlatf
 	}
 }
 
-func deleteAllPlugins(t assert.TestingT, c client.PluginAPIClient, protectedPlugins map[string]struct{}) {
+func deleteAllPlugins(t testing.TB, c client.PluginAPIClient, protectedPlugins map[string]struct{}) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	plugins, err := c.PluginList(context.Background(), filters.Args{})
 	// Docker EE does not allow cluster-wide plugin management.
 	if client.IsErrNotImplemented(err) {

testutil/environment/environment.go

@@ -6,11 +6,11 @@ import (
 	"os"
 	"path/filepath"
 	"strings"
+	"testing"
 
 	"github.com/docker/docker/api/types"
 	"github.com/docker/docker/api/types/filters"
 	"github.com/docker/docker/client"
-	"github.com/docker/docker/testutil"
 	"github.com/docker/docker/testutil/fixtures/load"
 	"github.com/pkg/errors"
 	"gotest.tools/assert"
@@ -165,10 +165,7 @@ func (e *Execution) IsUserNamespace() bool {
 // HasExistingImage checks whether there is an image with the given reference.
 // Note that this is done by filtering and then checking whether there were any
 // results -- so ambiguous references might result in false-positives.
-func (e *Execution) HasExistingImage(t assert.TestingT, reference string) bool {
+func (e *Execution) HasExistingImage(t testing.TB, reference string) bool {
-	if ht, ok := t.(testutil.HelperT); ok {
-		ht.Helper()
-	}
 	client := e.APIClient()
 	filter := filters.NewArgs()
 	filter.Add("dangling", "false")

testutil/environment/protect.go

@@ -2,11 +2,11 @@ package environment
 
 import (
 	"context"
+	"testing"
 
 	"github.com/docker/docker/api/types"
 	"github.com/docker/docker/api/types/filters"
 	dclient "github.com/docker/docker/client"
-	"github.com/docker/docker/testutil"
 	"gotest.tools/assert"
 )
 
@@ -33,10 +33,8 @@ func newProtectedElements() protectedElements {
 // ProtectAll protects the existing environment (containers, images, networks,
 // volumes, and, on Linux, plugins) from being cleaned up at the end of test
 // runs
-func ProtectAll(t assert.TestingT, testEnv *Execution) {
+func ProtectAll(t testing.TB, testEnv *Execution) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	ProtectContainers(t, testEnv)
 	ProtectImages(t, testEnv)
 	ProtectNetworks(t, testEnv)
@@ -48,10 +46,8 @@ func ProtectAll(t assert.TestingT, testEnv *Execution) {
 
 // ProtectContainer adds the specified container(s) to be protected in case of
 // clean
-func (e *Execution) ProtectContainer(t assert.TestingT, containers ...string) {
+func (e *Execution) ProtectContainer(t testing.TB, containers ...string) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	for _, container := range containers {
 		e.protectedElements.containers[container] = struct{}{}
 	}
@@ -59,18 +55,14 @@ func (e *Execution) ProtectContainer(t assert.TestingT, containers ...string) {
 
 // ProtectContainers protects existing containers from being cleaned up at the
 // end of test runs
-func ProtectContainers(t assert.TestingT, testEnv *Execution) {
+func ProtectContainers(t testing.TB, testEnv *Execution) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	containers := getExistingContainers(t, testEnv)
 	testEnv.ProtectContainer(t, containers...)
 }
 
-func getExistingContainers(t assert.TestingT, testEnv *Execution) []string {
+func getExistingContainers(t testing.TB, testEnv *Execution) []string {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	client := testEnv.APIClient()
 	containerList, err := client.ContainerList(context.Background(), types.ContainerListOptions{
 		All: true,
@@ -85,10 +77,8 @@ func getExistingContainers(t assert.TestingT, testEnv *Execution) []string {
 }
 
 // ProtectImage adds the specified image(s) to be protected in case of clean
-func (e *Execution) ProtectImage(t assert.TestingT, images ...string) {
+func (e *Execution) ProtectImage(t testing.TB, images ...string) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	for _, image := range images {
 		e.protectedElements.images[image] = struct{}{}
 	}
@@ -96,10 +86,8 @@ func (e *Execution) ProtectImage(t assert.TestingT, images ...string) {
 
 // ProtectImages protects existing images and on linux frozen images from being
 // cleaned up at the end of test runs
-func ProtectImages(t assert.TestingT, testEnv *Execution) {
+func ProtectImages(t testing.TB, testEnv *Execution) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	images := getExistingImages(t, testEnv)
 
 	if testEnv.OSType == "linux" {
@@ -108,10 +96,8 @@ func ProtectImages(t assert.TestingT, testEnv *Execution) {
 	testEnv.ProtectImage(t, images...)
 }
 
-func getExistingImages(t assert.TestingT, testEnv *Execution) []string {
+func getExistingImages(t testing.TB, testEnv *Execution) []string {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	client := testEnv.APIClient()
 	filter := filters.NewArgs()
 	filter.Add("dangling", "false")
@@ -145,10 +131,8 @@ func tagsFromImageSummary(image types.ImageSummary) []string {
 
 // ProtectNetwork adds the specified network(s) to be protected in case of
 // clean
-func (e *Execution) ProtectNetwork(t assert.TestingT, networks ...string) {
+func (e *Execution) ProtectNetwork(t testing.TB, networks ...string) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	for _, network := range networks {
 		e.protectedElements.networks[network] = struct{}{}
 	}
@@ -156,18 +140,14 @@ func (e *Execution) ProtectNetwork(t assert.TestingT, networks ...string) {
 
 // ProtectNetworks protects existing networks from being cleaned up at the end
 // of test runs
-func ProtectNetworks(t assert.TestingT, testEnv *Execution) {
+func ProtectNetworks(t testing.TB, testEnv *Execution) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	networks := getExistingNetworks(t, testEnv)
 	testEnv.ProtectNetwork(t, networks...)
 }
 
-func getExistingNetworks(t assert.TestingT, testEnv *Execution) []string {
+func getExistingNetworks(t testing.TB, testEnv *Execution) []string {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	client := testEnv.APIClient()
 	networkList, err := client.NetworkList(context.Background(), types.NetworkListOptions{})
 	assert.NilError(t, err, "failed to list networks")
@@ -180,10 +160,8 @@ func getExistingNetworks(t assert.TestingT, testEnv *Execution) []string {
 }
 
 // ProtectPlugin adds the specified plugin(s) to be protected in case of clean
-func (e *Execution) ProtectPlugin(t assert.TestingT, plugins ...string) {
+func (e *Execution) ProtectPlugin(t testing.TB, plugins ...string) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	for _, plugin := range plugins {
 		e.protectedElements.plugins[plugin] = struct{}{}
 	}
@@ -191,18 +169,14 @@ func (e *Execution) ProtectPlugin(t assert.TestingT, plugins ...string) {
 
 // ProtectPlugins protects existing plugins from being cleaned up at the end of
 // test runs
-func ProtectPlugins(t assert.TestingT, testEnv *Execution) {
+func ProtectPlugins(t testing.TB, testEnv *Execution) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	plugins := getExistingPlugins(t, testEnv)
 	testEnv.ProtectPlugin(t, plugins...)
 }
 
-func getExistingPlugins(t assert.TestingT, testEnv *Execution) []string {
+func getExistingPlugins(t testing.TB, testEnv *Execution) []string {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	client := testEnv.APIClient()
 	pluginList, err := client.PluginList(context.Background(), filters.Args{})
 	// Docker EE does not allow cluster-wide plugin management.
@@ -219,10 +193,8 @@ func getExistingPlugins(t assert.TestingT, testEnv *Execution) []string {
 }
 
 // ProtectVolume adds the specified volume(s) to be protected in case of clean
-func (e *Execution) ProtectVolume(t assert.TestingT, volumes ...string) {
+func (e *Execution) ProtectVolume(t testing.TB, volumes ...string) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	for _, volume := range volumes {
 		e.protectedElements.volumes[volume] = struct{}{}
 	}
@@ -230,18 +202,14 @@ func (e *Execution) ProtectVolume(t assert.TestingT, volumes ...string) {
 
 // ProtectVolumes protects existing volumes from being cleaned up at the end of
 // test runs
-func ProtectVolumes(t assert.TestingT, testEnv *Execution) {
+func ProtectVolumes(t testing.TB, testEnv *Execution) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	volumes := getExistingVolumes(t, testEnv)
 	testEnv.ProtectVolume(t, volumes...)
 }
 
-func getExistingVolumes(t assert.TestingT, testEnv *Execution) []string {
+func getExistingVolumes(t testing.TB, testEnv *Execution) []string {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	client := testEnv.APIClient()
 	volumeList, err := client.VolumeList(context.Background(), filters.Args{})
 	assert.NilError(t, err, "failed to list volumes")

testutil/registry/registry.go

@@ -10,8 +10,7 @@ import (
 	"testing"
 	"time"
 
-	"github.com/docker/docker/testutil"
+	"github.com/opencontainers/go-digest"
-	digest "github.com/opencontainers/go-digest"
 	"gotest.tools/assert"
 )
 
@@ -176,10 +175,8 @@ func (r *V2) getBlobFilename(blobDigest digest.Digest) string {
 }
 
 // ReadBlobContents read the file corresponding to the specified digest
-func (r *V2) ReadBlobContents(t assert.TestingT, blobDigest digest.Digest) []byte {
+func (r *V2) ReadBlobContents(t testing.TB, blobDigest digest.Digest) []byte {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	// Load the target manifest blob.
 	manifestBlob, err := ioutil.ReadFile(r.getBlobFilename(blobDigest))
 	assert.NilError(t, err, "unable to read blob")
@@ -187,10 +184,8 @@ func (r *V2) ReadBlobContents(t assert.TestingT, blobDigest digest.Digest) []byt
 }
 
 // WriteBlobContents write the file corresponding to the specified digest with the given content
-func (r *V2) WriteBlobContents(t assert.TestingT, blobDigest digest.Digest, data []byte) {
+func (r *V2) WriteBlobContents(t testing.TB, blobDigest digest.Digest, data []byte) {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	err := ioutil.WriteFile(r.getBlobFilename(blobDigest), data, os.FileMode(0644))
 	assert.NilError(t, err, "unable to write malicious data blob")
 }

testutil/request/request.go

@@ -11,12 +11,12 @@ import (
 	"net/url"
 	"os"
 	"path/filepath"
+	"testing"
 	"time"
 
 	"github.com/docker/docker/client"
 	"github.com/docker/docker/opts"
 	"github.com/docker/docker/pkg/ioutils"
-	"github.com/docker/docker/testutil"
 	"github.com/docker/docker/testutil/environment"
 	"github.com/docker/go-connections/sockets"
 	"github.com/docker/go-connections/tlsconfig"
@@ -25,10 +25,8 @@ import (
 )
 
 // NewAPIClient returns a docker API client configured from environment variables
-func NewAPIClient(t assert.TestingT, ops ...client.Opt) client.APIClient {
+func NewAPIClient(t testing.TB, ops ...client.Opt) client.APIClient {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	ops = append([]client.Opt{client.FromEnv}, ops...)
 	clt, err := client.NewClientWithOpts(ops...)
 	assert.NilError(t, err)
@@ -36,10 +34,8 @@ func NewAPIClient(t assert.TestingT, ops ...client.Opt) client.APIClient {
 }
 
 // DaemonTime provides the current time on the daemon host
-func DaemonTime(ctx context.Context, t assert.TestingT, client client.APIClient, testEnv *environment.Execution) time.Time {
+func DaemonTime(ctx context.Context, t testing.TB, client client.APIClient, testEnv *environment.Execution) time.Time {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	if testEnv.IsLocalDaemon() {
 		return time.Now()
 	}
@@ -54,10 +50,8 @@ func DaemonTime(ctx context.Context, t assert.TestingT, client client.APIClient,
 
 // DaemonUnixTime returns the current time on the daemon host with nanoseconds precision.
 // It return the time formatted how the client sends timestamps to the server.
-func DaemonUnixTime(ctx context.Context, t assert.TestingT, client client.APIClient, testEnv *environment.Execution) string {
+func DaemonUnixTime(ctx context.Context, t testing.TB, client client.APIClient, testEnv *environment.Execution) string {
-	if ht, ok := t.(testutil.HelperT); ok {
+	t.Helper()
-		ht.Helper()
-	}
 	dt := DaemonTime(ctx, t, client, testEnv)
 	return fmt.Sprintf("%d.%09d", dt.Unix(), int64(dt.Nanosecond()))
 }