| 87 | } |
| 88 | |
| 89 | func TestResponseWriterWrite(t *testing.T) { |
| 90 | testWriter := httptest.NewRecorder() |
| 91 | writer := &responseWriter{} |
| 92 | writer.reset(testWriter) |
| 93 | w := ResponseWriter(writer) |
| 94 | |
| 95 | n, err := w.Write([]byte("hola")) |
| 96 | assert.Equal(t, 4, n) |
| 97 | assert.Equal(t, 4, w.Size()) |
| 98 | assert.Equal(t, http.StatusOK, w.Status()) |
| 99 | assert.Equal(t, http.StatusOK, testWriter.Code) |
| 100 | assert.Equal(t, "hola", testWriter.Body.String()) |
| 101 | require.NoError(t, err) |
| 102 | |
| 103 | n, err = w.Write([]byte(" adios")) |
| 104 | assert.Equal(t, 6, n) |
| 105 | assert.Equal(t, 10, w.Size()) |
| 106 | assert.Equal(t, "hola adios", testWriter.Body.String()) |
| 107 | require.NoError(t, err) |
| 108 | } |
| 109 | |
| 110 | func TestResponseWriterHijack(t *testing.T) { |
| 111 | testWriter := httptest.NewRecorder() |