@@ -414,7 +414,7 @@ func TestWorkspaceApplicationAuth(t *testing.T) {
414
414
t .Log ("navigating to: " , gotLocation .String ())
415
415
req , err = http .NewRequestWithContext (ctx , "GET" , gotLocation .String (), nil )
416
416
require .NoError (t , err )
417
- resp , err = client . HTTPClient . Do ( req )
417
+ resp , err = doWithRetries ( t , client , req )
418
418
require .NoError (t , err )
419
419
resp .Body .Close ()
420
420
require .Equal (t , http .StatusTemporaryRedirect , resp .StatusCode )
@@ -471,7 +471,7 @@ func TestWorkspaceApplicationAuth(t *testing.T) {
471
471
req , err = http .NewRequestWithContext (ctx , "GET" , gotLocation .String (), nil )
472
472
require .NoError (t , err )
473
473
req .Header .Set (codersdk .SessionCustomHeader , apiKey )
474
- resp , err = client . HTTPClient . Do ( req )
474
+ resp , err = doWithRetries ( t , client , req )
475
475
require .NoError (t , err )
476
476
resp .Body .Close ()
477
477
require .Equal (t , http .StatusOK , resp .StatusCode )
@@ -1153,7 +1153,7 @@ func TestWorkspaceAppsNonCanonicalHeaders(t *testing.T) {
1153
1153
req .Header ["Sec-WebSocket-Key" ] = []string {secWebSocketKey }
1154
1154
1155
1155
req .Header .Set (codersdk .SessionCustomHeader , client .SessionToken ())
1156
- resp , err := client . HTTPClient . Do ( req )
1156
+ resp , err := doWithRetries ( t , client , req )
1157
1157
require .NoError (t , err )
1158
1158
defer resp .Body .Close ()
1159
1159
@@ -1205,7 +1205,7 @@ func TestWorkspaceAppsNonCanonicalHeaders(t *testing.T) {
1205
1205
req .Header ["Sec-WebSocket-Key" ] = []string {secWebSocketKey }
1206
1206
1207
1207
req .Header .Set (codersdk .SessionCustomHeader , client .SessionToken ())
1208
- resp , err := client . HTTPClient . Do ( req )
1208
+ resp , err := doWithRetries ( t , client , req )
1209
1209
require .NoError (t , err )
1210
1210
defer resp .Body .Close ()
1211
1211
0 commit comments