@@ -30,11 +30,12 @@ var _ = Describe("Uploader", func() {
3030 serverRequests []* http.Request
3131 serverRequestBody []string
3232
33- logger * lagertest.TestLogger
34- url * url.URL
35- file * os.File
36- expectedBytes int
37- expectedMD5 string
33+ logger * lagertest.TestLogger
34+ url * url.URL
35+ file * os.File
36+ expectedBytes int
37+ expectedMD5 string
38+ defaultTimeout time.Duration
3839 )
3940
4041 BeforeEach (func () {
@@ -49,6 +50,7 @@ var _ = Describe("Uploader", func() {
4950 rawMD5 := md5 .Sum ([]byte (contentString ))
5051 expectedMD5 = base64 .StdEncoding .EncodeToString (rawMD5 [:])
5152 file .Close ()
53+ defaultTimeout = 500 * time .Millisecond
5254 })
5355
5456 AfterEach (func () {
@@ -61,7 +63,7 @@ var _ = Describe("Uploader", func() {
6163
6264 Describe ("Insecure Upload" , func () {
6365 BeforeEach (func () {
64- upldr = uploader .New (logger , 100 * time . Millisecond , nil )
66+ upldr = uploader .New (logger , defaultTimeout , nil )
6567 })
6668
6769 Context ("when the upload is successful" , func () {
@@ -160,7 +162,7 @@ var _ = Describe("Uploader", func() {
160162 testServer = httptest .NewServer (http .HandlerFunc (func (w http.ResponseWriter , r * http.Request ) {
161163 requestInitiated <- struct {}{}
162164
163- time .Sleep (300 * time . Millisecond )
165+ time .Sleep (3 * defaultTimeout )
164166 fmt .Fprintln (w , "Hello, client" )
165167 }))
166168
@@ -269,7 +271,7 @@ var _ = Describe("Uploader", func() {
269271 })
270272
271273 It ("uploads the file to the url" , func () {
272- upldr = uploader .New (logger , 1 * time . Second , tlsConfig )
274+ upldr = uploader .New (logger , defaultTimeout , tlsConfig )
273275 numBytes , err = upldr .Upload (file .Name (), url , nil )
274276 Expect (err ).NotTo (HaveOccurred ())
275277
@@ -286,7 +288,7 @@ var _ = Describe("Uploader", func() {
286288 })
287289
288290 It ("returns the number of bytes written" , func () {
289- upldr = uploader .New (logger , 100 * time . Millisecond , tlsConfig )
291+ upldr = uploader .New (logger , defaultTimeout , tlsConfig )
290292 numBytes , err = upldr .Upload (file .Name (), url , nil )
291293 Expect (err ).NotTo (HaveOccurred ())
292294
@@ -307,15 +309,15 @@ var _ = Describe("Uploader", func() {
307309 })
308310
309311 It ("can communicate with the fileserver via one-sided TLS" , func () {
310- upldr = uploader .New (logger , 100 * time . Millisecond , tlsConfig )
312+ upldr = uploader .New (logger , defaultTimeout , tlsConfig )
311313 numBytes , err = upldr .Upload (file .Name (), url , nil )
312314 Expect (err ).NotTo (HaveOccurred ())
313315 })
314316 })
315317
316318 Context ("when the client has incorrect certs" , func () {
317319 It ("fails when no certs are provided" , func () {
318- upldr = uploader .New (logger , 100 * time . Millisecond , nil )
320+ upldr = uploader .New (logger , defaultTimeout , nil )
319321 numBytes , err = upldr .Upload (file .Name (), url , nil )
320322 Expect (err ).To (HaveOccurred ())
321323 })
@@ -328,7 +330,7 @@ var _ = Describe("Uploader", func() {
328330 tlsconfig .WithAuthorityFromFile ("fixtures/correct/server-ca.crt" ),
329331 )
330332 Expect (err ).NotTo (HaveOccurred ())
331- upldr = uploader .New (logger , 100 * time . Millisecond , tlsConfig )
333+ upldr = uploader .New (logger , defaultTimeout , tlsConfig )
332334 numBytes , err = upldr .Upload (file .Name (), url , nil )
333335 Expect (err ).To (HaveOccurred ())
334336 })
@@ -341,7 +343,7 @@ var _ = Describe("Uploader", func() {
341343 tlsconfig .WithAuthorityFromFile ("fixtures/incorrect/server-ca.crt" ),
342344 )
343345 Expect (err ).NotTo (HaveOccurred ())
344- upldr = uploader .New (logger , 100 * time . Millisecond , tlsConfig )
346+ upldr = uploader .New (logger , defaultTimeout , tlsConfig )
345347 numBytes , err = upldr .Upload (file .Name (), url , nil )
346348 Expect (err ).To (HaveOccurred ())
347349 })
0 commit comments