EventLoggerStage: fakeStage,
},
}))
})
})
Describe("Stop", func() {
It("stops agent services", func() {
err := vm.Stop()
Expect(err).ToNot(HaveOccurred())
Expect(fakeAgentClient.StopCallCount()).To(Equal(1))
})
Context("when stopping an agent fails", func() {
BeforeEach(func() {
fakeAgentClient.StopReturns(errors.New("fake-stop-error"))
})
It("returns an error", func() {
err := vm.Stop()
Expect(err).To(HaveOccurred())
Expect(err.Error()).To(ContainSubstring("fake-stop-error"))
})
})
})
Describe("Apply", func() {
It("sends apply spec to the agent", func() {
err := vm.Apply(applySpec)
Expect(err).ToNot(HaveOccurred())
Expect(fakeAgentClient.ApplyArgsForCall(0)).To(Equal(applySpec))
