mirror of https://github.com/status-im/fathom.git
65 lines
1.6 KiB
Go
65 lines
1.6 KiB
Go
package api
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
)
|
|
|
|
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)
|
|
}
|
|
}
|
|
|
|
func TestParseMajorMinor(t *testing.T) {
|
|
actual := parseMajorMinor("50.0.0")
|
|
expected := "50.0"
|
|
if actual != expected {
|
|
t.Errorf("Return value should be %s, is %s instead", expected, actual)
|
|
}
|
|
|
|
actual = parseMajorMinor("1.1")
|
|
expected = "1.1"
|
|
if actual != expected {
|
|
t.Errorf("Return value should be %s is %s instead", expected, actual)
|
|
}
|
|
}
|