| package handler |
|
|
| import ( |
| "encoding/json" |
| "net/http" |
| "net/http/httptest" |
| "testing" |
|
|
| "github.com/gin-gonic/gin" |
| "github.com/stretchr/testify/assert" |
| "github.com/stretchr/testify/require" |
| ) |
|
|
| func TestGatewayEnsureForwardErrorResponse_WritesFallbackWhenNotWritten(t *testing.T) { |
| gin.SetMode(gin.TestMode) |
| w := httptest.NewRecorder() |
| c, _ := gin.CreateTestContext(w) |
| c.Request = httptest.NewRequest(http.MethodGet, "/", nil) |
|
|
| h := &GatewayHandler{} |
| wrote := h.ensureForwardErrorResponse(c, false) |
|
|
| require.True(t, wrote) |
| require.Equal(t, http.StatusBadGateway, w.Code) |
|
|
| var parsed map[string]any |
| err := json.Unmarshal(w.Body.Bytes(), &parsed) |
| require.NoError(t, err) |
| assert.Equal(t, "error", parsed["type"]) |
| errorObj, ok := parsed["error"].(map[string]any) |
| require.True(t, ok) |
| assert.Equal(t, "upstream_error", errorObj["type"]) |
| assert.Equal(t, "Upstream request failed", errorObj["message"]) |
| } |
|
|
| |
| |
| func TestGatewayEnsureForwardErrorResponse_AppendsSSEAfterWritten(t *testing.T) { |
| gin.SetMode(gin.TestMode) |
| w := httptest.NewRecorder() |
| c, _ := gin.CreateTestContext(w) |
| c.Request = httptest.NewRequest(http.MethodGet, "/", nil) |
| c.String(http.StatusTeapot, "already written") |
|
|
| h := &GatewayHandler{} |
| wrote := h.ensureForwardErrorResponse(c, false) |
|
|
| require.True(t, wrote) |
| require.Equal(t, http.StatusTeapot, w.Code) |
| assert.Contains(t, w.Body.String(), "already written") |
| assert.Contains(t, w.Body.String(), `data: {"type":"error"`) |
| } |
|
|
| |
| |
| func TestGatewayEnsureForwardErrorResponse_ResponsesRouteAfterWrittenEmitsResponseFailed(t *testing.T) { |
| gin.SetMode(gin.TestMode) |
| w := httptest.NewRecorder() |
| c, _ := gin.CreateTestContext(w) |
| c.Request = httptest.NewRequest(http.MethodPost, EndpointResponses, nil) |
| _, _ = c.Writer.WriteString(":\n\n") |
|
|
| h := &GatewayHandler{} |
| wrote := h.ensureForwardErrorResponse(c, false) |
|
|
| require.True(t, wrote) |
| body := w.Body.String() |
| assert.Contains(t, body, ":\n\n") |
| assert.Contains(t, body, "event: response.failed\n") |
| assert.Contains(t, body, `"type":"response.failed"`) |
| } |
|
|
