1
0
mirror of https://github.com/ipfs/kubo.git synced 2025-05-21 00:47:22 +08:00
Files
kubo/core/corehttp/option_test.go
Łukasz Magiera ffc74391d5 move version.go to top level package
License: MIT
Signed-off-by: Łukasz Magiera <magik6k@gmail.com>
2018-08-09 11:01:10 +02:00

76 lines
1.8 KiB
Go

package corehttp
import (
"fmt"
"io"
"net/http"
"net/http/httptest"
"testing"
version "github.com/ipfs/go-ipfs"
)
type testcasecheckversion struct {
userAgent string
uri string
shouldHandle bool
responseBody string
responseCode int
}
func (tc testcasecheckversion) body() string {
if !tc.shouldHandle && tc.responseBody == "" {
return fmt.Sprintf("%s (%s != %s)\n", errAPIVersionMismatch, version.ApiVersion, tc.userAgent)
}
return tc.responseBody
}
func TestCheckVersionOption(t *testing.T) {
tcs := []testcasecheckversion{
{"/go-ipfs/0.1/", APIPath + "/test/", false, "", http.StatusBadRequest},
{"/go-ipfs/0.1/", APIPath + "/version", true, "check!", http.StatusOK},
{version.ApiVersion, APIPath + "/test", true, "check!", http.StatusOK},
{"Mozilla Firefox/no go-ipfs node", APIPath + "/test", true, "check!", http.StatusOK},
{"/go-ipfs/0.1/", "/webui", true, "check!", http.StatusOK},
}
for _, tc := range tcs {
t.Logf("%#v", tc)
r := httptest.NewRequest("POST", tc.uri, nil)
r.Header.Add("User-Agent", tc.userAgent) // old version, should fail
called := false
root := http.NewServeMux()
mux, err := CheckVersionOption()(nil, nil, root)
if err != nil {
t.Fatal(err)
}
mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
called = true
if !tc.shouldHandle {
t.Error("handler was called even though version didn't match")
} else {
io.WriteString(w, "check!")
}
})
w := httptest.NewRecorder()
root.ServeHTTP(w, r)
if tc.shouldHandle && !called {
t.Error("handler wasn't called even though it should have")
}
if w.Code != tc.responseCode {
t.Errorf("expected code %d but got %d", tc.responseCode, w.Code)
}
if w.Body.String() != tc.body() {
t.Errorf("expected error message %q, got %q", tc.body(), w.Body.String())
}
}
}