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)