mirror of https://github.com/mpolden/echoip
121 lines
2.9 KiB
Go
121 lines
2.9 KiB
Go
package api
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func httpGet(url string, json bool, userAgent string) (string, error) {
|
|
r, err := http.NewRequest("GET", url, nil)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if json {
|
|
r.Header.Set("Accept", "application/json")
|
|
}
|
|
r.Header.Set("User-Agent", userAgent)
|
|
res, err := http.DefaultClient.Do(r)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer res.Body.Close()
|
|
data, err := ioutil.ReadAll(res.Body)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return string(data), nil
|
|
}
|
|
|
|
func TestGetIP(t *testing.T) {
|
|
toJSON := func(k string, v string) string {
|
|
return fmt.Sprintf("{\n \"%s\": \"%s\"\n}", k, v)
|
|
}
|
|
s := httptest.NewServer(New().Handlers())
|
|
var tests = []struct {
|
|
url string
|
|
json bool
|
|
out string
|
|
userAgent string
|
|
}{
|
|
{s.URL, false, "127.0.0.1\n", "curl/7.26.0"},
|
|
{s.URL, false, "127.0.0.1\n", "Wget/1.13.4 (linux-gnu)"},
|
|
{s.URL, false, "127.0.0.1\n", "fetch libfetch/2.0"},
|
|
{s.URL + "/x-ifconfig-ip.json", false, toJSON("x-ifconfig-ip", "127.0.0.1"), ""},
|
|
{s.URL, true, toJSON("x-ifconfig-ip", "127.0.0.1"), ""},
|
|
}
|
|
for _, tt := range tests {
|
|
out, err := httpGet(tt.url, tt.json, tt.userAgent)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if out != tt.out {
|
|
t.Errorf("Expected %q, got %q", tt.out, out)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIPFromRequest(t *testing.T) {
|
|
var tests = []struct {
|
|
in *http.Request
|
|
out net.IP
|
|
}{
|
|
{&http.Request{RemoteAddr: "1.3.3.7:9999"}, net.ParseIP("1.3.3.7")},
|
|
{&http.Request{Header: http.Header{"X-Real-Ip": []string{"1.3.3.7"}}}, net.ParseIP("1.3.3.7")},
|
|
}
|
|
for _, tt := range tests {
|
|
ip, err := ipFromRequest(tt.in)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !ip.Equal(tt.out) {
|
|
t.Errorf("Expected %s, got %s", tt.out, ip)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestCmdFromParameters(t *testing.T) {
|
|
var tests = []struct {
|
|
in url.Values
|
|
out Cmd
|
|
}{
|
|
{url.Values{}, Cmd{Name: "curl"}},
|
|
{url.Values{"cmd": []string{"foo"}}, Cmd{Name: "curl"}},
|
|
{url.Values{"cmd": []string{"curl"}}, Cmd{Name: "curl"}},
|
|
{url.Values{"cmd": []string{"fetch"}}, Cmd{Name: "fetch", Args: "-qo -"}},
|
|
{url.Values{"cmd": []string{"wget"}}, Cmd{Name: "wget", Args: "-qO -"}},
|
|
}
|
|
for _, tt := range tests {
|
|
cmd := cmdFromQueryParams(tt.in)
|
|
if !reflect.DeepEqual(cmd, tt.out) {
|
|
t.Errorf("Expected %+v, got %+v", tt.out, cmd)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestCLIMatcher(t *testing.T) {
|
|
browserUserAgent := "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_8_4) " +
|
|
"AppleWebKit/537.36 (KHTML, like Gecko) Chrome/30.0.1599.28 " +
|
|
"Safari/537.36"
|
|
var tests = []struct {
|
|
in string
|
|
out bool
|
|
}{
|
|
{"curl/7.26.0", true},
|
|
{"Wget/1.13.4 (linux-gnu)", true},
|
|
{"fetch libfetch/2.0", true},
|
|
{browserUserAgent, false},
|
|
}
|
|
for _, tt := range tests {
|
|
r := &http.Request{Header: http.Header{"User-Agent": []string{tt.in}}}
|
|
if got := cliMatcher(r, nil); got != tt.out {
|
|
t.Errorf("Expected %t, got %t for %q", tt.out, got, tt.in)
|
|
}
|
|
}
|
|
}
|