ProfessionalUwU/echoip
1
0
Fork 0
mirror of https://github.com/mpolden/echoip.git synced 2025-02-20 14:08:58 +01:00
Code Issues Packages Projects Releases Wiki Activity

Send plain response for Accept: text/plain

Browse Source 
Fixes #24
This commit is contained in:
Martin Polden 2016-11-16 20:00:03 +01:00
parent 4a9710dcea
commit faf2f62612
2 changed files with 21 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/api.go
View File

@ -19,7 +19,10 @@ import (
"github.com/gorilla/mux" "github.com/gorilla/mux"
) )
const jsonMediaType = "application/json" const (
jsonMediaType = "application/json"
textMediaType = "text/plain"
)
var userAgentPattern = regexp.MustCompile( var userAgentPattern = regexp.MustCompile(
`^(?:curl|Wget|fetch\slibfetch|ddclient|Go-http-client|HTTPie)\/.*|Go\s1\.1\spackage\shttp$`, `^(?:curl|Wget|fetch\slibfetch|ddclient|Go-http-client|HTTPie)\/.*|Go\s1\.1\spackage\shttp$`,
@ -243,6 +246,7 @@ func (a *API) Router() http.Handler {
// CLI // CLI
r.Handle("/", appHandler(a.CLIHandler)).Methods("GET").MatcherFunc(cliMatcher) r.Handle("/", appHandler(a.CLIHandler)).Methods("GET").MatcherFunc(cliMatcher)
r.Handle("/", appHandler(a.CLIHandler)).Methods("GET").Headers("Accept", textMediaType)
r.Handle("/ip", appHandler(a.CLIHandler)).Methods("GET") r.Handle("/ip", appHandler(a.CLIHandler)).Methods("GET")
r.Handle("/country", appHandler(a.CLICountryHandler)).Methods("GET") r.Handle("/country", appHandler(a.CLICountryHandler)).Methods("GET")
r.Handle("/city", appHandler(a.CLICityHandler)).Methods("GET") r.Handle("/city", appHandler(a.CLICityHandler)).Methods("GET")

30
api/api_test.go
View File

@ -25,13 +25,13 @@ func newTestAPI() *API {
return &API{oracle: &mockOracle{}} return &API{oracle: &mockOracle{}}
} }
func httpGet(url string, json bool, userAgent string) (string, int, error) { func httpGet(url string, acceptMediaType string, userAgent string) (string, int, error) {
r, err := http.NewRequest("GET", url, nil) r, err := http.NewRequest("GET", url, nil)
if err != nil { if err != nil {
return "", 0, err return "", 0, err
} }
if json { if acceptMediaType != "" {
r.Header.Set("Accept", "application/json") r.Header.Set("Accept", acceptMediaType)
} }
r.Header.Set("User-Agent", userAgent) r.Header.Set("User-Agent", userAgent)
res, err := http.DefaultClient.Do(r) res, err := http.DefaultClient.Do(r)
@ -51,20 +51,22 @@ func TestCLIHandlers(t *testing.T) {
s := httptest.NewServer(newTestAPI().Router()) s := httptest.NewServer(newTestAPI().Router())
var tests = []struct { var tests = []struct {
url string url string
out string out string
status int status int
userAgent string userAgent string
acceptMediaType string
}{ }{
{s.URL, "127.0.0.1\n", 200, "curl/7.43.0"}, {s.URL, "127.0.0.1\n", 200, "curl/7.43.0", ""},
{s.URL + "/ip", "127.0.0.1\n", 200, ""}, {s.URL, "127.0.0.1\n", 200, "foo/bar", textMediaType},
{s.URL + "/country", "Elbonia\n", 200, ""}, {s.URL + "/ip", "127.0.0.1\n", 200, "", ""},
{s.URL + "/city", "Bornyasherk\n", 200, ""}, {s.URL + "/country", "Elbonia\n", 200, "", ""},
{s.URL + "/foo", "404 page not found", 404, ""}, {s.URL + "/city", "Bornyasherk\n", 200, "", ""},
{s.URL + "/foo", "404 page not found", 404, "", ""},
} }
for _, tt := range tests { for _, tt := range tests {
out, status, err := httpGet(tt.url /* json = */, false, tt.userAgent) out, status, err := httpGet(tt.url, tt.acceptMediaType, tt.userAgent)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -95,7 +97,7 @@ func TestJSONHandlers(t *testing.T) {
} }
for _, tt := range tests { for _, tt := range tests {
out, status, err := httpGet(tt.url /* json = */, true, "curl/7.2.6.0") out, status, err := httpGet(tt.url, jsonMediaType, "curl/7.2.6.0")
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }