@@ -13,6 +13,7 @@ import (
1313 "time"
1414
1515 "github.com/mitchellh/mapstructure"
16+ "github.com/stretchr/testify/assert"
1617 "github.com/stretchr/testify/require"
1718
1819 "github.com/99designs/gqlgen/client"
@@ -21,15 +22,16 @@ import (
2122func TestClient (t * testing.T ) {
2223 h := http .HandlerFunc (func (w http.ResponseWriter , r * http.Request ) {
2324 b , err := io .ReadAll (r .Body )
24- require .NoError (t , err )
25- require .Equal (t , `{"query":"user(id:$id){name}","variables":{"id":1}}` , string (b ))
26-
27- err = json .NewEncoder (w ).Encode (map [string ]any {
28- "data" : map [string ]any {
29- "name" : "bob" ,
30- },
31- })
32- require .NoError (t , err )
25+ if assert .NoError (t , err ) {
26+ assert .Equal (t , `{"query":"user(id:$id){name}","variables":{"id":1}}` , string (b ))
27+
28+ err = json .NewEncoder (w ).Encode (map [string ]any {
29+ "data" : map [string ]any {
30+ "name" : "bob" ,
31+ },
32+ })
33+ assert .NoError (t , err )
34+ }
3335 })
3436
3537 c := client .New (h )
@@ -46,14 +48,17 @@ func TestClient(t *testing.T) {
4648func TestClientMultipartFormData (t * testing.T ) {
4749 h := http .HandlerFunc (func (w http.ResponseWriter , r * http.Request ) {
4850 bodyBytes , err := io .ReadAll (r .Body )
49- require .NoError (t , err )
50- require .Contains (t , string (bodyBytes ), `Content-Disposition: form-data; name="operations"` )
51- require .Contains (t , string (bodyBytes ), `{"query":"mutation ($input: Input!) {}","variables":{"file":{}}` )
52- require .Contains (t , string (bodyBytes ), `Content-Disposition: form-data; name="map"` )
53- require .Contains (t , string (bodyBytes ), `{"0":["variables.file"]}` )
54- require .Contains (t , string (bodyBytes ), `Content-Disposition: form-data; name="0"; filename="example.txt"` )
55- require .Contains (t , string (bodyBytes ), `Content-Type: text/plain` )
56- require .Contains (t , string (bodyBytes ), `Hello World` )
51+ if ! assert .NoError (t , err ) {
52+ return
53+ }
54+
55+ assert .Contains (t , string (bodyBytes ), `Content-Disposition: form-data; name="operations"` )
56+ assert .Contains (t , string (bodyBytes ), `{"query":"mutation ($input: Input!) {}","variables":{"file":{}}` )
57+ assert .Contains (t , string (bodyBytes ), `Content-Disposition: form-data; name="map"` )
58+ assert .Contains (t , string (bodyBytes ), `{"0":["variables.file"]}` )
59+ assert .Contains (t , string (bodyBytes ), `Content-Disposition: form-data; name="0"; filename="example.txt"` )
60+ assert .Contains (t , string (bodyBytes ), `Content-Type: text/plain` )
61+ assert .Contains (t , string (bodyBytes ), `Hello World` )
5762
5863 w .Write ([]byte (`{}` ))
5964 })
@@ -83,7 +88,7 @@ func TestClientMultipartFormData(t *testing.T) {
8388
8489func TestAddHeader (t * testing.T ) {
8590 h := http .HandlerFunc (func (w http.ResponseWriter , r * http.Request ) {
86- require .Equal (t , "ASDF" , r .Header .Get ("Test-Key" ))
91+ assert .Equal (t , "ASDF" , r .Header .Get ("Test-Key" ))
8792
8893 w .Write ([]byte (`{}` ))
8994 })
@@ -98,7 +103,7 @@ func TestAddHeader(t *testing.T) {
98103
99104func TestAddClientHeader (t * testing.T ) {
100105 h := http .HandlerFunc (func (w http.ResponseWriter , r * http.Request ) {
101- require .Equal (t , "ASDF" , r .Header .Get ("Test-Key" ))
106+ assert .Equal (t , "ASDF" , r .Header .Get ("Test-Key" ))
102107
103108 w .Write ([]byte (`{}` ))
104109 })
@@ -112,9 +117,9 @@ func TestAddClientHeader(t *testing.T) {
112117func TestBasicAuth (t * testing.T ) {
113118 h := http .HandlerFunc (func (w http.ResponseWriter , r * http.Request ) {
114119 user , pass , ok := r .BasicAuth ()
115- require .True (t , ok )
116- require .Equal (t , "user" , user )
117- require .Equal (t , "pass" , pass )
120+ assert .True (t , ok )
121+ assert .Equal (t , "user" , user )
122+ assert .Equal (t , "pass" , pass )
118123
119124 w .Write ([]byte (`{}` ))
120125 })
@@ -130,8 +135,10 @@ func TestBasicAuth(t *testing.T) {
130135func TestAddCookie (t * testing.T ) {
131136 h := http .HandlerFunc (func (w http.ResponseWriter , r * http.Request ) {
132137 c , err := r .Cookie ("foo" )
133- require .NoError (t , err )
134- require .Equal (t , "value" , c .Value )
138+ if ! assert .NoError (t , err ) {
139+ return
140+ }
141+ assert .Equal (t , "value" , c .Value )
135142
136143 w .Write ([]byte (`{}` ))
137144 })
@@ -147,14 +154,16 @@ func TestAddCookie(t *testing.T) {
147154func TestAddExtensions (t * testing.T ) {
148155 h := http .HandlerFunc (func (w http.ResponseWriter , r * http.Request ) {
149156 b , err := io .ReadAll (r .Body )
150- require .NoError (t , err )
151- require .Equal (t , `{"query":"user(id:1){name}","extensions":{"persistedQuery":{"sha256Hash":"ceec2897e2da519612279e63f24658c3e91194cbb2974744fa9007a7e1e9f9e7","version":1}}}` , string (b ))
157+ if ! assert .NoError (t , err ) {
158+ return
159+ }
160+ assert .Equal (t , `{"query":"user(id:1){name}","extensions":{"persistedQuery":{"sha256Hash":"ceec2897e2da519612279e63f24658c3e91194cbb2974744fa9007a7e1e9f9e7","version":1}}}` , string (b ))
152161 err = json .NewEncoder (w ).Encode (map [string ]any {
153162 "data" : map [string ]any {
154163 "Name" : "Bob" ,
155164 },
156165 })
157- require .NoError (t , err )
166+ assert .NoError (t , err )
158167 })
159168
160169 c := client .New (h )
0 commit comments