fathom/api/api_test.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)
}
}