var abortErr error BeforeEach(func() { fakeBuildDB.GetBuildReturns(build, true, nil) }) JustBeforeEach(func() { abortErr = foundBuild.Abort(lagertest.NewTestLogger("test")) }) Context("when acquiring the lease succeeds", func() { var fakeLease *dbfakes.FakeLease BeforeEach(func() { fakeLease = new(dbfakes.FakeLease) fakeBuildDB.LeaseBuildTrackingReturns(fakeLease, true, nil) }) It("succeeds", func() { Expect(abortErr).NotTo(HaveOccurred()) }) It("marks the build as aborted", func() { Expect(fakeBuildDB.AbortBuildCallCount()).To(Equal(1)) Expect(fakeBuildDB.AbortBuildArgsForCall(0)).To(Equal(build.ID)) }) }) Context("when acquiring the lease fails", func() { var fakeLease *dbfakes.FakeLease