Describe("UnmountDisk", func() { var disk *fakebidisk.FakeDisk BeforeEach(func() { disk = fakebidisk.NewFakeDisk("fake-disk-cid") }) It("sends unmount disk to the agent", func() { err := vm.UnmountDisk(disk) Expect(err).ToNot(HaveOccurred()) Expect(fakeAgentClient.UnmountDiskArgsForCall(0)).To(Equal("fake-disk-cid")) }) Context("when unmounting disk fails", func() { BeforeEach(func() { fakeAgentClient.UnmountDiskReturns(errors.New("fake-unmount-error")) }) It("returns an error", func() { err := vm.UnmountDisk(disk) Expect(err).To(HaveOccurred()) Expect(err.Error()).To(ContainSubstring("fake-unmount-error")) }) }) }) Describe("Disks", func() { BeforeEach(func() { fakeAgentClient.ListDiskReturns([]string{"fake-disk-cid-1", "fake-disk-cid-2"}, nil) })