113 lines
1.8 KiB
Go
113 lines
1.8 KiB
Go
package http
|
|
|
|
import (
|
|
"net"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/micro/go-micro/v2/api/handler"
|
|
"github.com/micro/go-micro/v2/api/router"
|
|
regRouter "github.com/micro/go-micro/v2/api/router/registry"
|
|
"github.com/micro/go-micro/v2/registry"
|
|
"github.com/micro/go-micro/v2/registry/memory"
|
|
)
|
|
|
|
func testHttp(t *testing.T, path, service string) {
|
|
r := memory.NewRegistry()
|
|
|
|
l, err := net.Listen("tcp", "127.0.0.1:0")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer l.Close()
|
|
|
|
s := ®istry.Service{
|
|
Name: service,
|
|
Nodes: []*registry.Node{
|
|
{
|
|
Id: service + "-1",
|
|
Address: l.Addr().String(),
|
|
},
|
|
},
|
|
}
|
|
|
|
r.Register(s)
|
|
defer r.Deregister(s)
|
|
|
|
// setup the test handler
|
|
m := http.NewServeMux()
|
|
m.HandleFunc(path, func(w http.ResponseWriter, r *http.Request) {
|
|
w.Write([]byte(`you got served`))
|
|
})
|
|
|
|
// start http test serve
|
|
go http.Serve(l, m)
|
|
|
|
// create new request and writer
|
|
w := httptest.NewRecorder()
|
|
req, err := http.NewRequest("POST", path, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// initialise the handler
|
|
rt := regRouter.NewRouter(
|
|
router.WithHandler("http"),
|
|
router.WithRegistry(r),
|
|
)
|
|
|
|
p := NewHandler(handler.WithRouter(rt))
|
|
|
|
// execute the handler
|
|
p.ServeHTTP(w, req)
|
|
|
|
if w.Code != 200 {
|
|
t.Fatalf("Expected 200 response got %d %s", w.Code, w.Body.String())
|
|
}
|
|
|
|
if w.Body.String() != "you got served" {
|
|
t.Fatalf("Expected body: you got served. Got: %s", w.Body.String())
|
|
}
|
|
}
|
|
|
|
func TestHttpHandler(t *testing.T) {
|
|
testData := []struct {
|
|
path string
|
|
service string
|
|
}{
|
|
{
|
|
"/test/foo",
|
|
"test",
|
|
},
|
|
{
|
|
"/test/foo/baz",
|
|
"test",
|
|
},
|
|
{
|
|
"/v1/foo",
|
|
"v1.foo",
|
|
},
|
|
{
|
|
"/v1/foo/bar",
|
|
"v1.foo",
|
|
},
|
|
{
|
|
"/v2/baz",
|
|
"v2.baz",
|
|
},
|
|
{
|
|
"/v2/baz/bar",
|
|
"v2.baz",
|
|
},
|
|
{
|
|
"/v2/baz/bar",
|
|
"v2.baz",
|
|
},
|
|
}
|
|
|
|
for _, d := range testData {
|
|
testHttp(t, d.path, d.service)
|
|
}
|
|
}
|