Expect(err).To(Equal(deleteErr)) Expect(fakeStemcellRepo.ClearCurrentCalled).To(BeTrue()) }) }) }) Describe("MigrateDisk", func() { It("sends migrate_disk to the agent", func() { err := vm.MigrateDisk() Expect(err).ToNot(HaveOccurred()) Expect(fakeAgentClient.MigrateDiskCallCount()).To(Equal(1)) }) Context("when migrating disk fails", func() { BeforeEach(func() { fakeAgentClient.MigrateDiskReturns(errors.New("fake-migrate-error")) }) It("returns an error", func() { err := vm.MigrateDisk() Expect(err).To(HaveOccurred()) Expect(err.Error()).To(ContainSubstring("fake-migrate-error")) }) }) }) Describe("GetState", func() { BeforeEach(func() { fakeAgentClient.GetStateReturns(biagentclient.AgentState{JobState: "testing"}, nil) })