mirror of https://github.com/status-im/fathom.git
51 lines
1.3 KiB
Go
51 lines
1.3 KiB
Go
|
package api
|
||
|
|
|
||
|
|
import(
|
||
|
|
"testing"
|
||
|
|
"net/http"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestGetRequestedLimit(t *testing.T) {
|
||
|
|
r, _ := http.NewRequest("GET", "", nil)
|
||
|
|
limit := getRequestedLimit(r)
|
||
|
|
|
||
|
|
if limit != defaultLimit {
|
||
|
|
t.Errorf("Expected limit of %d does not match %d", defaultLimit, limit)
|
||
|
|
}
|
||
|
|
|
||
|
|
r, _ = http.NewRequest("GET", "?limit=50", nil)
|
||
|
|
limit = getRequestedLimit(r)
|
||
|
|
if limit != 50 {
|
||
|
|
t.Errorf("Expected limit of %d does not match %d", defaultLimit, limit)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetRequestedPeriods(t *testing.T) {
|
||
|
|
r, _ := http.NewRequest("GET", "?before=500&after=100", nil)
|
||
|
|
before, after := getRequestedPeriods(r)
|
||
|
|
|
||
|
|
if before != 500 || after != 100 {
|
||
|
|
t.Error("Expected URl argument for `before` or `after` does not match")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetRequestIp(t *testing.T) {
|
||
|
|
// test X-Forwarded-For header
|
||
|
|
ipAddress := "192.168.1.2"
|
||
|
|
r, _ := http.NewRequest("GET", "", nil)
|
||
|
|
r.Header.Set("X-Forwarded-For", ipAddress)
|
||
|
|
result := getRequestIp(r)
|
||
|
|
|
||
|
|
if result != ipAddress {
|
||
|
|
t.Errorf("Expected IP address of %s does not match %s", ipAddress, result)
|
||
|
|
}
|
||
|
|
|
||
|
|
// test RemoteAddr prop
|
||
|
|
r, _ = http.NewRequest("GET", "", nil)
|
||
|
|
r.RemoteAddr = ipAddress
|
||
|
|
result = getRequestIp(r)
|
||
|
|
if result != ipAddress {
|
||
|
|
t.Errorf("Expected IP address of %s does not match %s", ipAddress, result)
|
||
|
|
}
|
||
|
|
}
|