func (c *FakeClient) Do(req *http.Request) (*http.Response, error) {
c.CallCount++
if req.Body != nil {
content, err := boshhttp.ReadAndClose(req.Body)
if err != nil {
return nil, err
}
c.RequestBodies = append(c.RequestBodies, string(content))
}
c.Requests = append(c.Requests, req)
if len(c.doBehavior) > 0 {
output := c.doBehavior[0]
c.doBehavior = c.doBehavior[1:]
return output.resp, output.err
}
var resp *http.Response
if !c.returnNilResponse {
resp = &http.Response{
Body: boshhttp.NewStringReadCloser(c.responseMessage),
StatusCode: c.StatusCode,
}
}
err := c.Error
return resp, err
}
uint(maxOtherAttempts),
delay,
timeService,
)
})
Describe("Try", func() {
var (
lastError error
unavailable *http.Response
notFound *http.Response
)
BeforeEach(func() {
lastError = errors.New("last-error")
unavailable = &http.Response{StatusCode: 503, Body: boshhttp.NewStringReadCloser("")}
notFound = &http.Response{StatusCode: 404, Body: boshhttp.NewStringReadCloser("")}
})
Context("when all responses are only 503s", func() {
It("retries until maxUnavailableAttempts + maxOtherAttempts are exhausted", func() {
for i := 0; i < maxUnavailableAttempts+maxOtherAttempts-1; i++ {
retryable.AddAttemptBehavior(unavailable, true, errors.New("fake-error"))
}
retryable.AddAttemptBehavior(unavailable, true, lastError)
errChan := tryInBackground(monitRetryStrategy)
Eventually(errChan).Should(Receive(Equal(lastError)))
Expect(retryable.AttemptCalled).To(Equal(maxUnavailableAttempts + maxOtherAttempts))
})