3. Middleware тестирование

type Middleware func(http.Handler) http.Handler

package api

import (
    "context"
    "net/http"
    "strings"
)

// Кастомный строковый тип для ключа контекста исключает коллизии в сторонних пакетах
type contextKey string

const UserIDKey contextKey = "userID"

// AuthMiddleware проверяет авторизационный заголовок и внедряет UserID в контекст
func AuthMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        authHeader := r.Header.Get("Authorization")
        if authHeader == "" {
            http.Error(w, "missing token", http.StatusUnauthorized)
            return // КРИТИЧЕСКИ ВАЖНО: немедленный возврат!
        }

        parts := strings.Split(authHeader, " ")
        if len(parts) != 2 || parts[0] != "Bearer" || parts[1] != "secret-token" {
            http.Error(w, "invalid token", http.StatusUnauthorized)
            return // Прерываем цепочку
        }

        // Создаем новый контекст, содержащий извлеченный идентификатор
        ctx := context.WithValue(r.Context(), UserIDKey, 42)

        // Делегируем выполнение дальше, передавая запрос с обновленным контекстом
        next.ServeHTTP(w, r.WithContext(ctx))
    })
}

package api_test

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

    "github.com/stretchr/testify/require"
    "yourproject/internal/api"
)

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

    // 1. Описание таблицы тестовых сценариев
    type testCase struct {
        name           string
        authHeader     string
        expectedStatus int
        expectedNext   bool // Ожидается ли передача управления следующему звену?
    }

    testCases := []testCase{
        {
            name:           "Успешная авторизация по валидному Bearer токену",
            authHeader:     "Bearer secret-token",
            expectedStatus: http.StatusOK,
            expectedNext:   true,
        },
        {
            name:           "Отказ: Заголовок Authorization полностью отсутствует",
            authHeader:     "",
            expectedStatus: http.StatusUnauthorized,
            expectedNext:   false,
        },
        {
            name:           "Отказ: Передан некорректный секретный токен",
            authHeader:     "Bearer hacker-token",
            expectedStatus: http.StatusUnauthorized,
            expectedNext:   false,
        },
        {
            name:           "Отказ: Неверная схема авторизации (Basic вместо Bearer)",
            authHeader:     "Basic secret",
            expectedStatus: http.StatusUnauthorized,
            expectedNext:   false,
        },
    }

    for _, tc := range testCases {
        tc := tc // Захват переменной цикла для параллельного запуска
        t.Run(tc.name, func(t *testing.T) {
            t.Parallel()

            // 1. Инициализируем Dummy Handler-шпион
            var nextCalled bool
            var contextUserID any

            nextHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
                nextCalled = true
                // Извлекаем значение из контекста для строгой верификации
                contextUserID = r.Context().Value(api.UserIDKey)
                w.WriteHeader(http.StatusOK)
            })

            // 2. Оборачиваем шпиона в тестируемый мидлварь
            handlerToTest := api.AuthMiddleware(nextHandler)

            // 3. Формируем входящий запрос и рекордер
            req := httptest.NewRequest(http.MethodGet, "/secure-resource", nil)
            if tc.authHeader != "" {
                req.Header.Set("Authorization", tc.authHeader)
            }
            rec := httptest.NewRecorder()

            // 4. Act: Запускаем исполнение цепочки в памяти
            handlerToTest.ServeHTTP(rec, req)

            // 5. Assert: Верифицируем сетевой контракт
            res := rec.Result()
            defer res.Body.Close()

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

            // КЛЮЧЕВАЯ ПРОВЕРКА: сработал ли Short-circuit?
            require.Equal(t, tc.expectedNext, nextCalled,
                "Поведение передачи управления в next.ServeHTTP не соответствует спецификации")

            // Проверяем корректность инъекции контекста при успешном сценарии
            if tc.expectedNext {
                require.NotNil(t, contextUserID, "UserID обязан присутствовать в контексте")
                require.Equal(t, 42, contextUserID)
            }
        })
    }
}