mirror of https://github.com/mpolden/echoip
60 lines
1.5 KiB
Go
60 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"net/url"
|
|
"testing"
|
|
)
|
|
|
|
func TestIsCLI(t *testing.T) {
|
|
userAgents := []string{"curl/7.26.0", "Wget/1.13.4 (linux-gnu)",
|
|
"fetch libfetch/2.0"}
|
|
|
|
for _, userAgent := range userAgents {
|
|
if !isCLI(userAgent) {
|
|
t.Errorf("Expected true for %s", userAgent)
|
|
}
|
|
}
|
|
|
|
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"
|
|
if isCLI(browserUserAgent) {
|
|
t.Errorf("Expected false for %s", browserUserAgent)
|
|
}
|
|
}
|
|
|
|
func TestPathToKey(t *testing.T) {
|
|
if key := pathToKey("/ip"); key != "ip" {
|
|
t.Fatalf("Expected 'ip', got '%s'", key)
|
|
}
|
|
if key := pathToKey("/User-Agent"); key != "user-agent" {
|
|
t.Fatalf("Expected 'user-agent', got '%s'", key)
|
|
}
|
|
if key := pathToKey("/all.json"); key != "all" {
|
|
t.Fatalf("Expected 'all', got '%s'", key)
|
|
}
|
|
}
|
|
|
|
func TestLookupCmd(t *testing.T) {
|
|
values := url.Values{"cmd": []string{"curl"}}
|
|
if v := lookupCmd(values); v.Name != "curl" {
|
|
t.Fatalf("Expected 'curl', got '%s'", v)
|
|
}
|
|
values = url.Values{"cmd": []string{"foo"}}
|
|
if v := lookupCmd(values); v.Name != "curl" {
|
|
t.Fatalf("Expected 'curl', got '%s'", v)
|
|
}
|
|
values = url.Values{}
|
|
if v := lookupCmd(values); v.Name != "curl" {
|
|
t.Fatalf("Expected 'curl', got '%s'", v)
|
|
}
|
|
values = url.Values{"cmd": []string{"wget"}}
|
|
if v := lookupCmd(values); v.Name != "wget" {
|
|
t.Fatalf("Expected 'wget', got '%s'", v)
|
|
}
|
|
values = url.Values{"cmd": []string{"fetch"}}
|
|
if v := lookupCmd(values); v.Name != "fetch" {
|
|
t.Fatalf("Expected 'fetch', got '%s'", v)
|
|
}
|
|
}
|