package http import ( "net" "net/http" "net/http/httptest" "testing" "github.com/micro/go-micro/api/handler" "github.com/micro/go-micro/api/router" regRouter "github.com/micro/go-micro/api/router/registry" "github.com/micro/go-micro/config/cmd" "github.com/micro/go-micro/registry" "github.com/micro/go-micro/registry/memory" ) func testHttp(t *testing.T, path, service, ns string) { r := memory.NewRegistry() cmd.DefaultCmd = cmd.NewCmd(cmd.Registry(&r)) 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{ ®istry.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.WithNamespace(ns), ) 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 namespace string }{ { "/test/foo", "go.micro.api.test", "go.micro.api", }, { "/test/foo/baz", "go.micro.api.test", "go.micro.api", }, { "/v1/foo", "go.micro.api.v1.foo", "go.micro.api", }, { "/v1/foo/bar", "go.micro.api.v1.foo", "go.micro.api", }, { "/v2/baz", "go.micro.api.v2.baz", "go.micro.api", }, { "/v2/baz/bar", "go.micro.api.v2.baz", "go.micro.api", }, { "/v2/baz/bar", "v2.baz", "", }, } for _, d := range testData { testHttp(t, d.path, d.service, d.namespace) } }