Ω(fakeLockHandler.AddLockCallCount()).Should(Equal(1))
lockName, lockContents, initiallyClaimed := fakeLockHandler.AddLockArgsForCall(0)
Ω(lockName).Should(Equal("some-lock"))
Ω(string(lockContents)).Should(Equal("lock-contents"))
Ω(initiallyClaimed).Should(BeFalse())
})
Context("when adding the lock fails", func() {
BeforeEach(func() {
called := false
fakeLockHandler.AddLockStub = func(_ string, _ []byte, _ bool) (string, error) {
// succeed on second call
if !called {
called = true
return "", errors.New("disaster")
} else {
return "some-ref", nil
}
}
})
It("does not return an error as it retries", func() {
_, _, err := lockPool.AddUnclaimedLock(lockDir)
Ω(err).ShouldNot(HaveOccurred())
Ω(fakeLockHandler.AddLockCallCount()).Should(Equal(2))
})
})
Context("when adding the lock succeeds", func() {
Ω(fakeLockHandler.AddLockCallCount()).Should(Equal(1))
lockName, lockContents := fakeLockHandler.AddLockArgsForCall(0)
Ω(lockName).Should(Equal("some-lock"))
Ω(string(lockContents)).Should(Equal("lock-contents"))
})
Context("when adding the lock fails", func() {
BeforeEach(func() {
called := false
fakeLockHandler.AddLockStub = func(lockName string, lockContents []byte) (string, error) {
// succeed on second call
if !called {
called = true
return "", errors.New("disaster")
} else {
return "some-ref", nil
}
}
})
It("does not return an error as it retries", func() {
_, _, err := lockPool.AddLock(lockDir)
Ω(err).ShouldNot(HaveOccurred())
Ω(fakeLockHandler.AddLockCallCount()).Should(Equal(2))
})
})
Context("when adding the lock succeeds", func() {