4. Тестирование REST API

mux.HandleFunc("GET /user/{id}", ...) // Написали "user" в единственном числе вместо "users"

package api

import (
    "encoding/json"
    "net/http"
)

// UserService изолирует бизнес-логику работы с пользователями
type UserService interface {
    GetByID(id string) (string, error)
    Delete(id string) error
}

// UsersAPI группирует эндпоинты ресурса пользователей
type UsersAPI struct {
    svc UserService
}

func NewUsersAPI(svc UserService) *UsersAPI {
    return &UsersAPI{svc: svc}
}

// RegisterRoutes регистрирует маршруты ресурса в мультиплексоре (синтаксис Go 1.22+)
func (api *UsersAPI) RegisterRoutes(mux *http.ServeMux) {
    mux.HandleFunc("GET /users/{id}", api.getUser)
    mux.HandleFunc("DELETE /users/{id}", api.deleteUser)
}

func (api *UsersAPI) getUser(w http.ResponseWriter, r *http.Request) {
    // Извлекаем wildcard-параметр из контекста маршрута
    id := r.PathValue("id")
    if id == "" {
        http.Error(w, "missing id", http.StatusBadRequest)
        return
    }

    name, err := api.svc.GetByID(id)
    if err != nil {
        http.Error(w, "not found", http.StatusNotFound)
        return
    }

    w.Header().Set("Content-Type", "application/json")
    _ = json.NewEncoder(w).Encode(map[string]string{
        "id":   id,
        "name": name,
    })
}

func (api *UsersAPI) deleteUser(w http.ResponseWriter, r *http.Request) {
    id := r.PathValue("id")
    if err := api.svc.Delete(id); err != nil {
        http.Error(w, "internal error", http.StatusInternalServerError)
        return
    }

    // 204 No Content по стандарту не содержит тела ответа
    w.WriteHeader(http.StatusNoContent)
}

package api_test

import (
    "net/http"
    "net/http/httptest"
    "testing"

    "github.com/stretchr/testify/require"
    "go.uber.org/mock/gomock"

    "yourproject/internal/api"
    "yourproject/internal/mocks"
)

// setupRouter конструирует изолированный роутер для теста
func setupRouter(t *testing.T) (*http.ServeMux, *mocks.MockUserService) {
    t.Helper()
    ctrl := gomock.NewController(t)
    mockSvc := mocks.NewMockUserService(ctrl)

    mux := http.NewServeMux()
    usersAPI := api.NewUsersAPI(mockSvc)
    usersAPI.RegisterRoutes(mux) // Регистрируем маршруты в реальном роутере!

    return mux, mockSvc
}

func TestUsersAPI_REST(t *testing.T) {
    t.Parallel()

    type testCase struct {
        name           string
        method         string
        targetURL      string
        setupMock      func(m *mocks.MockUserService)
        expectedStatus int
        expectedJSON   string
    }

    testCases := []testCase{
        {
            name:      "GET /users/{id} - Успешное получение",
            method:    http.MethodGet,
            targetURL: "/users/42", // Проверяем реальный URL с подстановкой параметра
            setupMock: func(m *mocks.MockUserService) {
                // Убеждаемся, что роутер корректно передал параметр "42" в PathValue
                m.EXPECT().GetByID("42").Return("Gopher", nil).Times(1)
            },
            expectedStatus: http.StatusOK,
            expectedJSON:   `{"id": "42", "name": "Gopher"}`,
        },
        {
            name:      "DELETE /users/{id} - Успешное удаление",
            method:    http.MethodDelete,
            targetURL: "/users/99",
            setupMock: func(m *mocks.MockUserService) {
                m.EXPECT().Delete("99").Return(nil).Times(1)
            },
            expectedStatus: http.StatusNoContent,
            expectedJSON:   "", // Для 204 No Content тело отсутствует
        },
        {
            name:           "GET /users/ - Запрос с завершающим слэшем без ID",
            method:         http.MethodGet,
            targetURL:      "/users/",
            setupMock:      func(m *mocks.MockUserService) {}, // Мок не должен вызываться
            expectedStatus: http.StatusNotFound,               // Маршрутизатор не найдет шаблон
            expectedJSON:   "",
        },
        {
            name:           "POST /users/42 - Неподдерживаемый HTTP-метод",
            method:         http.MethodPost,
            targetURL:      "/users/42",
            setupMock:      func(m *mocks.MockUserService) {},
            expectedStatus: http.StatusMethodNotAllowed, // 405 Method Not Allowed генерируется роутером
            expectedJSON:   "",
        },
    }

    for _, tc := range testCases {
        tc := tc
        t.Run(tc.name, func(t *testing.T) {
            t.Parallel()

            mux, mockSvc := setupRouter(t)
            tc.setupMock(mockSvc)

            // Создаем запрос, адресованный РОУТЕРУ, а не изолированной функции
            req := httptest.NewRequest(tc.method, tc.targetURL, nil)
            rec := httptest.NewRecorder()

            // Передаем управление в ServeMux
            mux.ServeHTTP(rec, req)

            res := rec.Result()
            defer res.Body.Close()

            require.Equal(t, tc.expectedStatus, res.StatusCode)

            if tc.expectedJSON != "" {
                require.JSONEq(t, tc.expectedJSON, rec.Body.String())
            }
        })
    }
}