44 lines
1.0 KiB
Go
44 lines
1.0 KiB
Go
|
package log
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"golang.org/x/net/websocket"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestGetIP(t *testing.T) {
|
||
|
|
assert := assert.New(t)
|
||
|
|
|
||
|
|
req, _ := http.NewRequest("GET", "https://google.com/lola/duda?q=wasd", nil)
|
||
|
|
req.RemoteAddr = "127.0.0.1"
|
||
|
|
assert.Equal("127.0.0.1", getIP(req))
|
||
|
|
req.Header.Set("X-Forwarded-For", "8.8.8.8")
|
||
|
|
assert.Equal("8.8.8.8", getIP(req))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestHTTP(t *testing.T) {
|
||
|
|
assert := assert.New(t)
|
||
|
|
|
||
|
|
req, _ := http.NewRequest("GET", "https://google.com/lola/duda?q=wasd", nil)
|
||
|
|
entry := HTTP(req)
|
||
|
|
_, ok := entry.Fields["remote"]
|
||
|
|
|
||
|
|
assert.NotNil(ok, "remote address not set in logger")
|
||
|
|
assert.Equal("GET", entry.Fields["method"], "method not set in logger")
|
||
|
|
assert.Equal("/lola/duda?q=wasd", entry.Fields["url"], "path not set in logger")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestWebsocketX(t *testing.T) {
|
||
|
|
assert := assert.New(t)
|
||
|
|
ws := &websocket.Conn{}
|
||
|
|
entry := WebsocketX(ws)
|
||
|
|
_, ok := entry.Fields["remote"]
|
||
|
|
|
||
|
|
assert.NotNil(ok, "remote address not set in logger")
|
||
|
|
assert.True(entry.Fields["websocket"].(bool))
|
||
|
|
}
|