docs/api/primary_test.go

82 lines
1.5 KiB
Go

package api
import (
"net/http"
"net/http/httptest"
"testing"
"github.com/gorilla/mux"
)
func TestRequest(t *testing.T) {
t.Parallel()
context := &context{}
r := mux.NewRouter()
setupPrimaryRouter(r, context, false)
w := httptest.NewRecorder()
req, e := http.NewRequest("GET", "/version", nil)
if nil != e {
t.Fatalf("couldn't set up test request")
}
r.ServeHTTP(w, req)
t.Logf("%d - %s", w.Code, w.Body.String())
t.Log("Expecting no extra Headers")
for k, v := range w.Header() {
t.Log(k, " : ", v)
}
if w.Code == 404 {
t.Fatalf("failed not found")
}
}
func TestCorsRequest(t *testing.T) {
t.Parallel()
context := &context{}
primary := mux.NewRouter()
setupPrimaryRouter(primary, context, true)
w := httptest.NewRecorder()
// test an OPTIONS request when cors enabled
r, e := http.NewRequest("OPTIONS", "/version", nil)
if nil != e {
t.Fatalf("couldn't set up test request")
}
primary.ServeHTTP(w, r)
t.Logf("%d - %s", w.Code, w.Body.String())
t.Log("Expecting extra cors Headers")
for k, v := range w.Header() {
t.Log(k, " : ", v)
}
if w.Code == 404 {
t.Fatalf("failed not found")
}
// test a normal request ( GET /_ping ) when cors enabled
w2 := httptest.NewRecorder()
r2, e2 := http.NewRequest("GET", "/_ping", nil)
if nil != e2 {
t.Fatalf("couldn't set up test request")
}
primary.ServeHTTP(w2, r2)
if w2.Body.String() != "OK" {
t.Fatalf("couldn't get body content when cors enabled")
}
if w2.Code == 404 {
t.Fatalf("failed not found")
}
}