package place import ( "context" "database/sql" "errors" "net/http/httptest" "strconv" "testing" "github.com/brianvoe/gofakeit/v6" "github.com/gofiber/fiber/v2" "github.com/gojuno/minimock/v3" "github.com/stretchr/testify/assert" API "git.dmitriygnatenko.ru/dima/homethings/internal/api/v1" "git.dmitriygnatenko.ru/dima/homethings/internal/api/v1/place/mocks" "git.dmitriygnatenko.ru/dima/homethings/internal/dto" "git.dmitriygnatenko.ru/dima/homethings/internal/helpers" "git.dmitriygnatenko.ru/dima/homethings/internal/models" ) func TestUpdatePlaceHandler(t *testing.T) { t.Parallel() type req struct { method string route string contentType string body *dto.UpdatePlaceRequest } var ( placeID = gofakeit.Number(1, 1000) parentID = gofakeit.Number(1, 1000) title = gofakeit.Phrase() testError = errors.New(gofakeit.Phrase()) layout = "2006-01-02 15:04:05" correctReq = req{ method: fiber.MethodPut, route: "/v1/places/" + strconv.Itoa(placeID), body: &dto.UpdatePlaceRequest{ Title: title, ParentID: &parentID, }, contentType: fiber.MIMEApplicationJSON, } repoRes = models.Place{ ID: placeID, Title: title, ParentID: sql.NullInt64{Int64: int64(parentID), Valid: true}, CreatedAt: gofakeit.Date(), UpdatedAt: gofakeit.Date(), } expectedRes = dto.PlaceResponse{ ID: placeID, ParentID: &parentID, Title: repoRes.Title, CreatedAt: repoRes.CreatedAt.Format(layout), UpdatedAt: repoRes.UpdatedAt.Format(layout), } ) tests := []struct { name string req req resCode int resBody interface{} placeRepoMock func(mc *minimock.Controller) PlaceRepository }{ { name: "positive case", req: correctReq, resCode: fiber.StatusOK, resBody: expectedRes, placeRepoMock: func(mc *minimock.Controller) PlaceRepository { mock := mocks.NewPlaceRepositoryMock(mc) mock.UpdateMock.Inspect(func(ctx context.Context, req models.UpdatePlaceRequest, tx *sql.Tx) { assert.Equal(mc, title, req.Title) assert.Equal(mc, parentID, int(req.ParentID.Int64)) }).Return(nil) mock.GetMock.Inspect(func(ctx context.Context, id int) { assert.Equal(mc, placeID, id) }).Return(&repoRes, nil) return mock }, }, { name: "negative case - bad request", req: req{ method: fiber.MethodPut, route: "/v1/places/" + gofakeit.Word(), }, resCode: fiber.StatusBadRequest, placeRepoMock: func(mc *minimock.Controller) PlaceRepository { return mocks.NewPlaceRepositoryMock(mc) }, }, { name: "negative case - body parse error", req: req{ method: fiber.MethodPut, route: "/v1/places/" + strconv.Itoa(placeID), }, resCode: fiber.StatusBadRequest, placeRepoMock: func(mc *minimock.Controller) PlaceRepository { return mocks.NewPlaceRepositoryMock(mc) }, }, { name: "negative case - request without title", req: req{ method: fiber.MethodPut, route: "/v1/places/" + strconv.Itoa(placeID), contentType: fiber.MIMEApplicationJSON, body: &dto.UpdatePlaceRequest{}, }, resCode: fiber.StatusBadRequest, resBody: []*dto.ValidateErrorResponse{ { Field: "UpdatePlaceRequest.Title", Tag: "required", }, }, placeRepoMock: func(mc *minimock.Controller) PlaceRepository { return mocks.NewPlaceRepositoryMock(mc) }, }, { name: "negative case - repository error (update place)", req: correctReq, resCode: fiber.StatusInternalServerError, placeRepoMock: func(mc *minimock.Controller) PlaceRepository { mock := mocks.NewPlaceRepositoryMock(mc) mock.UpdateMock.Inspect(func(ctx context.Context, req models.UpdatePlaceRequest, tx *sql.Tx) { assert.Equal(mc, title, req.Title) assert.Equal(mc, parentID, int(req.ParentID.Int64)) }).Return(testError) return mock }, }, { name: "negative case - repository error (get place)", req: correctReq, resCode: fiber.StatusInternalServerError, placeRepoMock: func(mc *minimock.Controller) PlaceRepository { mock := mocks.NewPlaceRepositoryMock(mc) mock.UpdateMock.Inspect(func(ctx context.Context, req models.UpdatePlaceRequest, tx *sql.Tx) { assert.Equal(mc, title, req.Title) assert.Equal(mc, parentID, int(req.ParentID.Int64)) }).Return(nil) mock.GetMock.Inspect(func(ctx context.Context, id int) { assert.Equal(mc, placeID, id) }).Return(nil, testError) return mock }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { t.Parallel() mc := minimock.NewController(t) fiberApp := fiber.New() fiberApp.Put("/v1/places/:placeId", UpdatePlaceHandler(tt.placeRepoMock(mc))) fiberReq := httptest.NewRequest(tt.req.method, tt.req.route, helpers.ConvertDataToIOReader(tt.req.body)) fiberReq.Header.Add(fiber.HeaderContentType, tt.req.contentType) fiberRes, _ := fiberApp.Test(fiberReq, API.DefaultTestTimeOut) assert.Equal(t, tt.resCode, fiberRes.StatusCode) if tt.resBody != nil { assert.Equal(t, helpers.MarshalResponse(tt.resBody), helpers.ConvertBodyToString(fiberRes.Body)) } }) } }