fix: unit test pet

isd-sgcu · Feb 20, 2024 · cb4ff13 · cb4ff13
1 parent 23292bb
commit cb4ff13
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/app/service/pet/pet.service_test.go b/src/app/service/pet/pet.service_test.go
@@ -578,6 +578,13 @@ func (t *PetServiceTest) TestDeleteNotFound() {
 	protoResp := &petproto.DeletePetResponse{
 		Success: false,
 	}
+	imageProtoReq := &imgproto.DeleteByPetIdRequest{
+		PetId: t.Pet.Id,
+	}
+	imageProtoResp := &imgproto.DeleteByPetIdResponse{
+		Success: true,
+	}
+
 	clientErr := status.Error(codes.NotFound, constant.PetNotFoundMessage)
 
 	expected := t.NotFoundErr
@@ -586,6 +593,7 @@ func (t *PetServiceTest) TestDeleteNotFound() {
 	client.On("Delete", protoReq).Return(protoResp, clientErr)
 
 	imageClient := imagemock.ImageClientMock{}
+	imageClient.On("DeleteByPetId", imageProtoReq).Return(imageProtoResp, nil)
 
 	imageSvc := imageSvc.NewService(&imageClient)
 	svc := NewService(client, imageSvc)
@@ -602,6 +610,13 @@ func (t *PetServiceTest) TestDeleteServiceUnavailableError() {
 	protoResp := &petproto.DeletePetResponse{
 		Success: false,
 	}
+	imageProtoReq := &imgproto.DeleteByPetIdRequest{
+		PetId: t.Pet.Id,
+	}
+	imageProtoResp := &imgproto.DeleteByPetIdResponse{
+		Success: true,
+	}
+
 	clientErr := status.Error(codes.Unavailable, constant.UnavailableServiceMessage)
 
 	expected := t.UnavailableServiceErr
@@ -610,6 +625,7 @@ func (t *PetServiceTest) TestDeleteServiceUnavailableError() {
 	client.On("Delete", protoReq).Return(protoResp, clientErr)
 
 	imageClient := imagemock.ImageClientMock{}
+	imageClient.On("DeleteByPetId", imageProtoReq).Return(imageProtoResp, nil)
 
 	imageSvc := imageSvc.NewService(&imageClient)
 	svc := NewService(client, imageSvc)