integration-cli: fix pollCheck

Signed-off-by: Tibor Vass <tibor@docker.com>

integration-cli/docker_utils_test.go

@@ -20,6 +20,7 @@ import (
 	"github.com/docker/docker/integration-cli/cli"
 	"github.com/docker/docker/integration-cli/daemon"
 	"gotest.tools/assert"
+	"gotest.tools/assert/cmp"
 	"gotest.tools/icmd"
 	"gotest.tools/poll"
 )
@@ -441,8 +442,18 @@ func pollCheck(t *testing.T, f checkF, compare func(x interface{}) assert.BoolOr
 	return func(poll.LogT) poll.Result {
 		t.Helper()
 		v, comment := f(t)
-		if assert.Check(t, compare(v)) {
+		r := compare(v)
-			return poll.Success()
+		switch r := r.(type) {
+		case bool:
+			if r {
+				return poll.Success()
+			}
+		case cmp.Comparison:
+			if r().Success() {
+				return poll.Success()
+			}
+		default:
+			panic(fmt.Errorf("pollCheck: type %T not implemented", r))
 		}
 		return poll.Continue(comment)
 	}