feat: image svc add assignpet

isd-sgcu · Jan 27, 2024 · a56c4aa · a56c4aa
1 parent 1160b70
commit a56c4aa
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 3 deletions.
diff --git a/src/app/service/image/image.service.go b/src/app/service/image/image.service.go
@@ -32,3 +32,20 @@ func (s *Service) FindByPetId(petId string) ([]*proto.Image, error) {
 	return res.Images, nil
 
 }
+
+func (s *Service) AssignPet(petId string, imageIds []string) error {
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	_, err := s.client.AssignPet(ctx, &proto.AssignPetRequest{PetId: petId, Ids: imageIds})
+	if err != nil {
+		log.Error().
+			Err(err).
+			Str("service", "image").
+			Str("module", "assign pet").
+			Msg("Error while connecting to service")
+		return err
+	}
+
+	return nil
+}
diff --git a/src/app/service/image/image.service_test.go b/src/app/service/image/image.service_test.go
@@ -15,8 +15,9 @@ import (
 
 type ImageServiceTest struct {
 	suite.Suite
-	petId  string
-	images []*proto.Image
+	petId    string
+	images   []*proto.Image
+	imageIds []string
 }
 
 func TestImageService(t *testing.T) {
@@ -37,6 +38,7 @@ func (t *ImageServiceTest) SetupTest() {
 			ImageUrl: faker.URL(),
 		},
 	}
+	t.imageIds = []string{t.images[0].Id, t.images[1].Id}
 }
 
 func (t *ImageServiceTest) TestFindByPetIdSuccess() {
@@ -56,7 +58,7 @@ func (t *ImageServiceTest) TestFindByPetIdSuccess() {
 func (t *ImageServiceTest) TestFindByPetIdError() {
 	c := mock.ClientMock{}
 	c.On("FindByPetId", &proto.FindImageByPetIdRequest{PetId: t.petId}).
-		Return(&proto.FindImageByPetIdResponse{Images: t.images}, status.Error(codes.Unavailable, "Connection Timeout"))
+		Return(nil, status.Error(codes.Unavailable, "Connection Timeout"))
 
 	srv := NewService(&c)
 	actual, err := srv.FindByPetId(t.petId)
@@ -66,3 +68,27 @@ func (t *ImageServiceTest) TestFindByPetIdError() {
 	assert.Nil(t.T(), actual)
 	assert.Equal(t.T(), codes.Unavailable, st.Code())
 }
+
+func (t *ImageServiceTest) TestAssignPetSuccess() {
+	c := mock.ClientMock{}
+	c.On("AssignPet", &proto.AssignPetRequest{PetId: t.petId, Ids: t.imageIds}).
+		Return(&proto.AssignPetResponse{Success: true}, nil)
+
+	srv := NewService(&c)
+	err := srv.AssignPet(t.petId, t.imageIds)
+
+	assert.Nil(t.T(), err)
+}
+
+func (t *ImageServiceTest) TestAssignPetError() {
+	c := mock.ClientMock{}
+	c.On("AssignPet", &proto.AssignPetRequest{PetId: t.petId, Ids: t.imageIds}).
+		Return(nil, status.Error(codes.Unavailable, "Connection Timeout"))
+
+	srv := NewService(&c)
+	err := srv.AssignPet(t.petId, t.imageIds)
+
+	st, ok := status.FromError(err)
+	assert.True(t.T(), ok)
+	assert.Equal(t.T(), codes.Unavailable, st.Code())
+}