var _ = Describe("GetStateRetryable", func() {
Describe("Attempt", func() {
var (
fakeAgentClient *fakeagentclient.FakeAgentClient
getStateRetryable boshretry.Retryable
)
BeforeEach(func() {
fakeAgentClient = fakeagentclient.NewFakeAgentClient()
getStateRetryable = NewGetStateRetryable(fakeAgentClient)
})
Context("when get_state fails", func() {
BeforeEach(func() {
fakeAgentClient.SetGetStateBehavior(AgentState{}, errors.New("fake-get-state-error"))
})
It("returns an error", func() {
isRetryable, err := getStateRetryable.Attempt()
Expect(err).To(HaveOccurred())
Expect(isRetryable).To(BeFalse())
Expect(err.Error()).To(ContainSubstring("fake-get-state-error"))
Expect(fakeAgentClient.GetStateCalledTimes).To(Equal(1))
})
})
Context("when get_state returns state as pending", func() {
BeforeEach(func() {
fakeAgentClient.SetGetStateBehavior(AgentState{JobState: "pending"}, nil)
})
