chore: update backup test

Signed-off-by: mlycore <[email protected]>
apache · Nov 14, 2023 · bc69e7e · bc69e7e
1 parent 8526edc
commit bc69e7e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pitr/cli/internal/cmd/backup_test.go b/pitr/cli/internal/cmd/backup_test.go
@@ -382,18 +382,21 @@ var _ = Describe("test backup mock", func() {
 		})
 
 		It("agent server is not running", func() {
+			mockIreq.EXPECT().Body(gomock.Any())
 			mockIreq.EXPECT().Send(gomock.Any()).Return(errors.New("error")).AnyTimes()
 			mockIreq.EXPECT().Header(gomock.Any()).AnyTimes()
 			Expect(checkAgentServerStatus(ls)).To(BeFalse())
 		})
 
 		It("agent server are running", func() {
+			mockIreq.EXPECT().Body(gomock.Any())
 			mockIreq.EXPECT().Send(gomock.Any()).Return(nil).AnyTimes()
 			mockIreq.EXPECT().Header(gomock.Any()).AnyTimes()
 			Expect(checkAgentServerStatus(ls)).To(BeTrue())
 		})
 
 		It("one agent server is not running", func() {
+			mockIreq.EXPECT().Body(gomock.Any())
 			mockIreq.EXPECT().Send(gomock.Any()).Return(errors.New("failed"))
 			mockIreq.EXPECT().Send(gomock.Any()).Return(nil).AnyTimes()
 			mockIreq.EXPECT().Header(gomock.Any()).AnyTimes()