func sleepForIncrements(timeService *fakeclock.FakeClock, attempts int, delay time.Duration) {
for i := 0; i < attempts; i++ {
Eventually(timeService.WatcherCount).Should(Equal(1))
timeService.Increment(delay)
Eventually(timeService.WatcherCount).Should(Equal(0))
}
}
func incrementSleepInBackground(fakeTimeService *fakeclock.FakeClock, delay time.Duration) chan struct{} {
doneChan := make(chan struct{})
go func() {
for {
select {
case <-doneChan:
return
default:
if fakeTimeService.WatcherCount() > 0 {
fakeTimeService.Increment(delay)
Eventually(fakeTimeService.WatcherCount).Should(Equal(0))
}
}
}
}()
return doneChan
}
var _ = Describe("FakeClock", func() {
const Δ time.Duration = 10 * time.Millisecond
var (
fakeClock *fakeclock.FakeClock
initialTime time.Time
)
BeforeEach(func() {
initialTime = time.Date(2014, 1, 1, 3, 0, 30, 0, time.UTC)
fakeClock = fakeclock.NewFakeClock(initialTime)
})
Describe("Now", func() {
It("returns the current time, w/o race conditions", func() {
go fakeClock.Increment(time.Minute)
Eventually(fakeClock.Now).Should(Equal(initialTime.Add(time.Minute)))
})
})
Describe("Sleep", func() {
It("blocks until the given interval elapses", func() {
doneSleeping := make(chan struct{})
go func() {
fakeClock.Sleep(10 * time.Second)
close(doneSleeping)
}()
Consistently(doneSleeping, Δ).ShouldNot(BeClosed())
fakeClock.Increment(5 * time.Second)
const Δ = 10 * time.Millisecond
var (
fakeClock *fakeclock.FakeClock
initialTime time.Time
)
BeforeEach(func() {
initialTime = time.Date(2014, 1, 1, 3, 0, 30, 0, time.UTC)
fakeClock = fakeclock.NewFakeClock(initialTime)
})
It("proivdes a channel that receives after the given interval has elapsed", func() {
timer := fakeClock.NewTimer(10 * time.Second)
timeChan := timer.C()
Consistently(timeChan, Δ).ShouldNot(Receive())
fakeClock.Increment(5 * time.Second)
Consistently(timeChan, Δ).ShouldNot(Receive())
fakeClock.Increment(4 * time.Second)
Consistently(timeChan, Δ).ShouldNot(Receive())
fakeClock.Increment(1 * time.Second)
Eventually(timeChan).Should(Receive(Equal(initialTime.Add(10 * time.Second))))
fakeClock.Increment(10 * time.Second)
Consistently(timeChan, Δ).ShouldNot(Receive())
})
})
Expect(retryable.Response().Body.(ClosedChecker).Closed()).To(BeTrue())
})
})
})
It("waits for retry delay between retries", func() {
for i := 0; i < maxUnavailableAttempts+maxOtherAttempts; i++ {
retryable.AddAttemptBehavior(unavailable, true, nil)
}
go func() {
monitRetryStrategy.Try()
}()
Eventually(retryable.Attempts).Should(Equal(1))
timeService.Increment(delay - time.Millisecond)
Consistently(retryable.Attempts).Should(Equal(1))
timeService.Increment(2 * time.Millisecond)
Eventually(retryable.Attempts).Should(Equal(2))
})
Context("when error is not due to failed response", func() {
It("retries until maxOtherAttempts are exhausted", func() {
for i := 0; i < maxOtherAttempts-1; i++ {
retryable.AddAttemptBehavior(nil, true, errors.New("request error"))
}
retryable.AddAttemptBehavior(nil, true, lastError)
errChan := tryInBackground(monitRetryStrategy)
sleepForIncrements(timeService, maxOtherAttempts, delay)