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 }
logErrBuffer = bytes.NewBufferString("") logger = boshlog.NewWriterLogger(boshlog.LevelDebug, logOutBuffer, logErrBuffer) ui = NewWriterUI(uiOut, uiErr, logger) fakeTimeService = fakeclock.NewFakeClock(time.Now()) stage = NewStage(ui, fakeTimeService, logger) }) Describe("Perform", func() { It("prints a single-line stage", func() { actionsPerformed := []string{} err := stage.Perform("Simple stage 1", func() error { actionsPerformed = append(actionsPerformed, "1") fakeTimeService.Increment(time.Minute) return nil }) Expect(err).To(BeNil()) expectedOutput := "Simple stage 1... Finished (00:01:00)\n" Expect(uiOut.String()).To(Equal(expectedOutput)) Expect(actionsPerformed).To(Equal([]string{"1"})) }) It("fails on error", func() { actionsPerformed := []string{} stageError := bosherr.Error("fake-stage-1-error") err := stage.Perform("Simple stage 1", func() error {
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("provides a channel that receives the time at each interval", func() { ticker := fakeClock.NewTicker(10 * time.Second) timeChan := ticker.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) Eventually(timeChan).Should(Receive(Equal(initialTime.Add(20 * time.Second)))) fakeClock.Increment(10 * time.Second) Eventually(timeChan).Should(Receive(Equal(initialTime.Add(30 * time.Second)))) }) })
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)
sshRetryStrategy = &SSHRetryStrategy{ ConnectionRefusedTimeout: connectionRefusedTimeout, AuthFailureTimeout: authFailureTimeout, TimeService: fakeTimeService, } }) Describe("IsRetryable", func() { refusedErr := errors.New("connection refused") authErr := errors.New("unable to authenticate") Context("when err is connection refused", func() { It("retries for connectionRefusedTimeout", func() { Expect(sshRetryStrategy.IsRetryable(refusedErr)).To(BeTrue()) fakeTimeService.Increment(connectionRefusedTimeout - time.Second) Expect(sshRetryStrategy.IsRetryable(refusedErr)).To(BeTrue()) fakeTimeService.Increment(time.Second) Expect(sshRetryStrategy.IsRetryable(refusedErr)).To(BeFalse()) }) }) Context("when err is unable to authenticate", func() { It("retries for authFailureTimeout", func() { Expect(sshRetryStrategy.IsRetryable(authErr)).To(BeTrue()) fakeTimeService.Increment(authFailureTimeout - time.Second) Expect(sshRetryStrategy.IsRetryable(authErr)).To(BeTrue()) fakeTimeService.Increment(time.Second)