77 "testing"
88
99 "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"
1112 "golang.org/x/net/context"
1213)
1314
@@ -17,7 +18,7 @@ func TestVersionMiddlewareVersion(t *testing.T) {
1718 expectedVersion := defaultVersion
1819 handler := func (ctx context.Context , w http.ResponseWriter , r * http.Request , vars map [string ]string ) error {
1920 v := httputils .VersionFromContext (ctx )
20- assert .Equal (t , expectedVersion , v )
21+ assert .Check (t , is . Equal ( expectedVersion , v ) )
2122 return nil
2223 }
2324
@@ -56,17 +57,17 @@ func TestVersionMiddlewareVersion(t *testing.T) {
5657 err := h (ctx , resp , req , map [string ]string {"version" : test .reqVersion })
5758
5859 if test .errString != "" {
59- assert .EqualError (t , err , test .errString )
60+ assert .Check (t , is . Error ( err , test .errString ) )
6061 } else {
61- assert .NoError (t , err )
62+ assert .Check (t , err )
6263 }
6364 }
6465}
6566
6667func TestVersionMiddlewareWithErrorsReturnsHeaders (t * testing.T ) {
6768 handler := func (ctx context.Context , w http.ResponseWriter , r * http.Request , vars map [string ]string ) error {
6869 v := httputils .VersionFromContext (ctx )
69- assert .NotEmpty (t , v )
70+ assert .Check (t , len ( v ) != 0 )
7071 return nil
7172 }
7273
@@ -81,11 +82,11 @@ func TestVersionMiddlewareWithErrorsReturnsHeaders(t *testing.T) {
8182
8283 vars := map [string ]string {"version" : "0.1" }
8384 err := h (ctx , resp , req , vars )
84- assert .Error (t , err )
85+ assert .Check (t , is . ErrorContains ( err , "" ) )
8586
8687 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 ) )
9192}
0 commit comments