Skip to content

Commit 0b0afbf

Browse files
authored
Executor: Increase uploader default timeout (#78)
* Increase the default timeout * Use a multiple of the new default timeout to sleep for tests when the timeout needs to be exceeded
1 parent f898bc9 commit 0b0afbf

File tree

1 file changed

+15
-13
lines changed

1 file changed

+15
-13
lines changed

depot/uploader/uploader_test.go

Lines changed: 15 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)