123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151 |
- package notification
- import (
- "context"
- "database/sql"
- "net/http/httptest"
- "strconv"
- "testing"
- "time"
- "github.com/brianvoe/gofakeit/v6"
- "github.com/gofiber/fiber/v2"
- "github.com/gojuno/minimock/v3"
- "github.com/stretchr/testify/assert"
- "git.dmitriygnatenko.ru/dima/homethings/internal/api/v1/notification/mocks"
- "git.dmitriygnatenko.ru/dima/homethings/internal/helpers/test"
- "git.dmitriygnatenko.ru/dima/homethings/internal/models"
- )
- func Test_DeleteThingNotificationHandler(t *testing.T) {
- t.Parallel()
- type req struct {
- method string
- route string
- contentType string
- }
- var (
- thingID = uint64(gofakeit.Number(1, 1000))
- testError = gofakeit.Error()
- correctReq = req{
- method: fiber.MethodDelete,
- route: "/v1/things/notifications/" + strconv.FormatUint(thingID, 10),
- contentType: fiber.MIMEApplicationJSON,
- }
- repoRes = models.ThingNotification{
- ThingID: uint64(thingID),
- NotificationDate: gofakeit.Date().Truncate(time.Second),
- CreatedAt: gofakeit.Date(),
- UpdatedAt: gofakeit.Date(),
- }
- )
- tests := []struct {
- name string
- req req
- resCode int
- resBody interface{}
- repoMock func(mc *minimock.Controller) ThingNotificationRepository
- }{
- {
- name: "positive case",
- req: correctReq,
- resCode: fiber.StatusOK,
- repoMock: func(mc *minimock.Controller) ThingNotificationRepository {
- mock := mocks.NewThingNotificationRepositoryMock(mc)
- mock.GetMock.Inspect(func(ctx context.Context, id uint64) {
- assert.Equal(mc, thingID, id)
- }).Return(&repoRes, nil)
- mock.DeleteMock.Inspect(func(ctx context.Context, id uint64) {
- assert.Equal(mc, thingID, id)
- }).Return(nil)
- return mock
- },
- },
- {
- name: "negative case - bad request",
- req: req{
- method: fiber.MethodDelete,
- route: "/v1/things/notifications/" + gofakeit.Word(),
- contentType: fiber.MIMEApplicationJSON,
- },
- resCode: fiber.StatusBadRequest,
- repoMock: func(mc *minimock.Controller) ThingNotificationRepository {
- return mocks.NewThingNotificationRepositoryMock(mc)
- },
- },
- {
- name: "negative case - repository error (get)",
- req: correctReq,
- resCode: fiber.StatusInternalServerError,
- repoMock: func(mc *minimock.Controller) ThingNotificationRepository {
- mock := mocks.NewThingNotificationRepositoryMock(mc)
- mock.GetMock.Inspect(func(ctx context.Context, id uint64) {
- assert.Equal(mc, thingID, id)
- }).Return(nil, testError)
- return mock
- },
- },
- {
- name: "negative case - bad request (notification not found)",
- req: correctReq,
- resCode: fiber.StatusBadRequest,
- repoMock: func(mc *minimock.Controller) ThingNotificationRepository {
- mock := mocks.NewThingNotificationRepositoryMock(mc)
- mock.GetMock.Inspect(func(ctx context.Context, id uint64) {
- assert.Equal(mc, thingID, id)
- }).Return(nil, sql.ErrNoRows)
- return mock
- },
- },
- {
- name: "negative case - repository error (delete)",
- req: correctReq,
- resCode: fiber.StatusInternalServerError,
- repoMock: func(mc *minimock.Controller) ThingNotificationRepository {
- mock := mocks.NewThingNotificationRepositoryMock(mc)
- mock.GetMock.Inspect(func(ctx context.Context, id uint64) {
- assert.Equal(mc, thingID, id)
- }).Return(&repoRes, nil)
- mock.DeleteMock.Inspect(func(ctx context.Context, id uint64) {
- assert.Equal(mc, thingID, id)
- }).Return(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.Delete("/v1/things/notifications/:thingId", DeleteThingNotificationHandler(tt.repoMock(mc)))
- fiberReq := httptest.NewRequest(tt.req.method, tt.req.route, nil)
- fiberReq.Header.Add(fiber.HeaderContentType, tt.req.contentType)
- fiberRes, _ := fiberApp.Test(fiberReq, test.TestTimeout)
- assert.Equal(t, tt.resCode, fiberRes.StatusCode)
- if tt.resBody != nil {
- assert.Equal(t, test.MarshalResponse(tt.resBody), test.ConvertBodyToString(fiberRes.Body))
- }
- })
- }
- }
|