bbs test

Author: kart2bc

controllers/evacuation_controller_test.go

@@ -261,7 +261,7 @@ var _ = Describe("Evacuation Controller", func() {
 
 		It("unclaims and reauctions the lrp", func() {
 			Expect(fakeActualLRPDB.UnclaimActualLRPCallCount()).To(Equal(1))
-			_, _, key := fakeActualLRPDB.UnclaimActualLRPArgsForCall(0)
+			_, _, _, key := fakeActualLRPDB.UnclaimActualLRPArgsForCall(0)
 			Expect(key).To(Equal(lrpKey))
 
 			Expect(fakeDesiredLRPDB.DesiredLRPSchedulingInfoByProcessGuidCallCount()).To(Equal(1))
@@ -1143,7 +1143,7 @@ var _ = Describe("Evacuation Controller", func() {
 
 				It("unclaims the lrp and requests an auction", func() {
 					Expect(fakeActualLRPDB.UnclaimActualLRPCallCount()).To(Equal(1))
-					_, _, lrpKey := fakeActualLRPDB.UnclaimActualLRPArgsForCall(0)
+					_, _, _, lrpKey := fakeActualLRPDB.UnclaimActualLRPArgsForCall(0)
 					Expect(lrpKey.ProcessGuid).To(Equal(actual.ProcessGuid))
 					Expect(lrpKey.Index).To(Equal(actual.Index))
 

controllers/lrp_convergence_controller_test.go

@@ -262,8 +262,9 @@ var _ = Describe("LRP Convergence Controllers", func() {
 
 		It("transition the LRP to UNCLAIMED state", func() {
 			Eventually(fakeLRPDB.UnclaimActualLRPCallCount).Should(Equal(1))
-			_, _, actualKey := fakeLRPDB.UnclaimActualLRPArgsForCall(0)
+			_, _, isStale, actualKey := fakeLRPDB.UnclaimActualLRPArgsForCall(0)
 			Expect(actualKey).To(Equal(key))
+			Expect(isStale).To(BeTrue())
 		})
 
 		It("emits an LRPChanged event", func() {
@@ -541,6 +542,8 @@ var _ = Describe("LRP Convergence Controllers", func() {
 
 			It("unclaims the suspect replacement lrp", func() {
 				Expect(fakeLRPDB.UnclaimActualLRPCallCount()).To(Equal(1))
+				_, _, isStale, _ := fakeLRPDB.UnclaimActualLRPArgsForCall(0)
+				Expect(isStale).To(BeFalse())
 			})
 
 			It("does not emit change events", func() {