whatismyip/router/geo_test.go
2021-11-10 20:06:12 +01:00

107 lines
2.4 KiB
Go

package router
import (
"net/http"
"net/http/httptest"
"testing"
"github.com/stretchr/testify/assert"
)
func TestGeo(t *testing.T) {
expected := `City: London
Country: United Kingdom
Country Code: GB
Latitude: 51.514200
Longitude: -0.093100
Postal Code:
Time Zone: Europe/London
`
req, _ := http.NewRequest("GET", "/geo", nil)
req.Header.Set("X-Real-IP", testIP.ipv4)
w := httptest.NewRecorder()
app.ServeHTTP(w, req)
assert.Equal(t, 200, w.Code)
assert.Equal(t, contentType.text, w.Header().Get("Content-Type"))
assert.Equal(t, expected, w.Body.String())
}
func TestGeoField(t *testing.T) {
req, _ := http.NewRequest("GET", "/geo/latitude", nil)
req.Header.Set("X-Real-IP", testIP.ipv4)
w := httptest.NewRecorder()
app.ServeHTTP(w, req)
assert.Equal(t, 200, w.Code)
assert.Equal(t, contentType.text, w.Header().Get("Content-Type"))
assert.Equal(t, "51.514200", w.Body.String())
}
func TestGeoField404(t *testing.T) {
req, _ := http.NewRequest("GET", "/geo/not-found", nil)
req.Header.Set("X-Real-IP", testIP.ipv4)
w := httptest.NewRecorder()
app.ServeHTTP(w, req)
assert.Equal(t, 404, w.Code)
}
func TestASN(t *testing.T) {
expected := `ASN Number: 12552
ASN Organization: IP-Only
`
req, _ := http.NewRequest("GET", "/asn", nil)
req.Header.Set("X-Real-IP", testIP.ipv4ASN)
w := httptest.NewRecorder()
app.ServeHTTP(w, req)
assert.Equal(t, 200, w.Code)
assert.Equal(t, contentType.text, w.Header().Get("Content-Type"))
assert.Equal(t, expected, w.Body.String())
}
func TestASNField(t *testing.T) {
req, _ := http.NewRequest("GET", "/asn/organization", nil)
req.Header.Set("X-Real-IP", testIP.ipv4ASN)
w := httptest.NewRecorder()
app.ServeHTTP(w, req)
assert.Equal(t, 200, w.Code)
assert.Equal(t, contentType.text, w.Header().Get("Content-Type"))
assert.Equal(t, "IP-Only", w.Body.String())
}
func TestASNField404(t *testing.T) {
req, _ := http.NewRequest("GET", "/asn/not-found", nil)
req.Header.Set("X-Real-IP", testIP.ipv4ASN)
w := httptest.NewRecorder()
app.ServeHTTP(w, req)
assert.Equal(t, 404, w.Code)
}
func TestASN_IPv6(t *testing.T) {
expected := `ASN Number: 6739
ASN Organization: Cableuropa - ONO
`
req, _ := http.NewRequest("GET", "/asn", nil)
req.Header.Set("X-Real-IP", testIP.ipv6ASN)
w := httptest.NewRecorder()
app.ServeHTTP(w, req)
assert.Equal(t, 200, w.Code)
assert.Equal(t, contentType.text, w.Header().Get("Content-Type"))
assert.Equal(t, expected, w.Body.String())
}