mirror of
https://github.com/labstack/echo.git
synced 2024-12-22 20:06:21 +02:00
39 lines
910 B
Go
39 lines
910 B
Go
package standard
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/labstack/gommon/log"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestResponseWriteHeader(t *testing.T) {
|
|
rec := httptest.NewRecorder()
|
|
res := NewResponse(rec, log.New("test"))
|
|
res.WriteHeader(http.StatusOK)
|
|
assert.True(t, res.Committed())
|
|
assert.Equal(t, http.StatusOK, res.Status())
|
|
}
|
|
|
|
func TestResponseWrite(t *testing.T) {
|
|
rec := httptest.NewRecorder()
|
|
res := NewResponse(rec, log.New("test"))
|
|
res.Write([]byte("test"))
|
|
assert.Equal(t, int64(4), res.Size())
|
|
assert.Equal(t, "test", rec.Body.String())
|
|
res.Flush()
|
|
assert.True(t, rec.Flushed)
|
|
}
|
|
|
|
func TestResponseSetCookie(t *testing.T) {
|
|
rec := httptest.NewRecorder()
|
|
res := NewResponse(rec, log.New("test"))
|
|
res.SetCookie(&Cookie{&http.Cookie{
|
|
Name: "name",
|
|
Value: "Jon Snow",
|
|
}})
|
|
assert.Equal(t, "name=Jon Snow", rec.Header().Get("Set-Cookie"))
|
|
}
|