From bc69e7ebc8b90e4c135a8131e3a3eae93ece3aa3 Mon Sep 17 00:00:00 2001 From: mlycore Date: Tue, 14 Nov 2023 17:50:07 +0800 Subject: [PATCH] chore: update backup test Signed-off-by: mlycore --- pitr/cli/internal/cmd/backup_test.go | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pitr/cli/internal/cmd/backup_test.go b/pitr/cli/internal/cmd/backup_test.go index 2ddab5c5..8844c18e 100644 --- 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()