李建琦
6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
26 lines
787 B
Go
26 lines
787 B
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
middleware2 "github.com/Wei-Shaw/sub2api/internal/server/middleware"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestOpsCaptureWriterDoesNotCopyIngressRejectBody(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
context, _ := gin.CreateTestContext(httptest.NewRecorder())
|
|
writer := acquireOpsCaptureWriter(context.Writer)
|
|
defer releaseOpsCaptureWriter(writer)
|
|
writer.ctx = context
|
|
context.Writer = writer
|
|
middleware2.MarkIngressRejected(context, middleware2.IngressRejectInvalidAPIKey)
|
|
context.Status(http.StatusUnauthorized)
|
|
_, err := context.Writer.WriteString(`{"code":"INVALID_API_KEY","message":"Invalid API key"}`)
|
|
require.NoError(t, err)
|
|
require.Zero(t, writer.buf.Len())
|
|
}
|