7
7
"testing"
8
8
9
9
"github.com/docker/docker/api/server/httputils"
10
- "github.com/stretchr/testify/assert"
10
+ "github.com/gotestyourself/gotestyourself/assert"
11
+ is "github.com/gotestyourself/gotestyourself/assert/cmp"
11
12
"golang.org/x/net/context"
12
13
)
13
14
@@ -17,7 +18,7 @@ func TestVersionMiddlewareVersion(t *testing.T) {
17
18
expectedVersion := defaultVersion
18
19
handler := func (ctx context.Context , w http.ResponseWriter , r * http.Request , vars map [string ]string ) error {
19
20
v := httputils .VersionFromContext (ctx )
20
- assert .Equal (t , expectedVersion , v )
21
+ assert .Check (t , is . Equal ( expectedVersion , v ) )
21
22
return nil
22
23
}
23
24
@@ -56,17 +57,17 @@ func TestVersionMiddlewareVersion(t *testing.T) {
56
57
err := h (ctx , resp , req , map [string ]string {"version" : test .reqVersion })
57
58
58
59
if test .errString != "" {
59
- assert .EqualError (t , err , test .errString )
60
+ assert .Check (t , is . Error ( err , test .errString ) )
60
61
} else {
61
- assert .NoError (t , err )
62
+ assert .Check (t , err )
62
63
}
63
64
}
64
65
}
65
66
66
67
func TestVersionMiddlewareWithErrorsReturnsHeaders (t * testing.T ) {
67
68
handler := func (ctx context.Context , w http.ResponseWriter , r * http.Request , vars map [string ]string ) error {
68
69
v := httputils .VersionFromContext (ctx )
69
- assert .NotEmpty (t , v )
70
+ assert .Check (t , len ( v ) != 0 )
70
71
return nil
71
72
}
72
73
@@ -81,11 +82,11 @@ func TestVersionMiddlewareWithErrorsReturnsHeaders(t *testing.T) {
81
82
82
83
vars := map [string ]string {"version" : "0.1" }
83
84
err := h (ctx , resp , req , vars )
84
- assert .Error (t , err )
85
+ assert .Check (t , is . ErrorContains ( err , "" ) )
85
86
86
87
hdr := resp .Result ().Header
87
- assert .Contains (t , hdr .Get ("Server" ), "Docker/" + defaultVersion )
88
- assert .Contains (t , hdr .Get ("Server" ), runtime .GOOS )
89
- assert .Equal (t , hdr .Get ("API-Version" ), defaultVersion )
90
- assert .Equal (t , hdr .Get ("OSType" ), runtime .GOOS )
88
+ assert .Check (t , is . Contains ( hdr .Get ("Server" ), "Docker/" + defaultVersion ) )
89
+ assert .Check (t , is . Contains ( hdr .Get ("Server" ), runtime .GOOS ) )
90
+ assert .Check (t , is . Equal ( hdr .Get ("API-Version" ), defaultVersion ) )
91
+ assert .Check (t , is . Equal ( hdr .Get ("OSType" ), runtime .GOOS ) )
91
92
}
0 commit comments