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)