mirror of https://github.com/status-im/fathom.git
85 lines
2.0 KiB
Go
85 lines
2.0 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"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)
|
|
}
|
|
}
|
|
|
|
func TestRespond(t *testing.T) {
|
|
w := httptest.NewRecorder()
|
|
respond(w, 15)
|
|
|
|
// assert json header
|
|
if w.Header().Get("Content-Type") != "application/json" {
|
|
t.Errorf("Invalid Content-Type header")
|
|
}
|
|
|
|
// assert json response
|
|
var d int
|
|
err := json.NewDecoder(w.Body).Decode(&d)
|
|
if err != nil {
|
|
t.Errorf("Invalid response body: %s", err)
|
|
}
|
|
|
|
}
|