85 lines
1.7 KiB
Go
85 lines
1.7 KiB
Go
package server
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/myodc/go-micro/registry"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
type testHandler struct{}
|
|
|
|
type testRequest struct{}
|
|
|
|
type testResponse struct{}
|
|
|
|
func (t *testHandler) Test(ctx context.Context, req *testRequest, rsp *testResponse) error {
|
|
return nil
|
|
}
|
|
|
|
func TestExtractAddress(t *testing.T) {
|
|
data := []struct {
|
|
Input string
|
|
Output string
|
|
}{
|
|
{"10.0.0.1", "10.0.0.1"},
|
|
}
|
|
|
|
for _, d := range data {
|
|
addr, err := extractAddress(d.Input)
|
|
if err != nil {
|
|
t.Errorf("Expected %s: %v", d.Output, err)
|
|
}
|
|
if addr != d.Output {
|
|
t.Errorf("Expected %s, got %s", d.Output, addr)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestExtractEndpoint(t *testing.T) {
|
|
handler := &testHandler{}
|
|
typ := reflect.TypeOf(handler)
|
|
|
|
var endpoints []*registry.Endpoint
|
|
|
|
for m := 0; m < typ.NumMethod(); m++ {
|
|
if e := extractEndpoint(typ.Method(m)); e != nil {
|
|
endpoints = append(endpoints, e)
|
|
}
|
|
}
|
|
|
|
if i := len(endpoints); i != 1 {
|
|
t.Errorf("Expected 1 endpoint, have %d", i)
|
|
}
|
|
|
|
if endpoints[0].Name != "Test" {
|
|
t.Errorf("Expected handler Test, got %s", endpoints[0].Name)
|
|
}
|
|
|
|
if endpoints[0].Request == nil {
|
|
t.Error("Expected non nil request")
|
|
}
|
|
|
|
if endpoints[0].Response == nil {
|
|
t.Error("Expected non nil request")
|
|
}
|
|
|
|
if endpoints[0].Request.Name != "testRequest" {
|
|
t.Errorf("Expected testRequest got %s", endpoints[0].Request.Name)
|
|
}
|
|
|
|
if endpoints[0].Response.Name != "testResponse" {
|
|
t.Errorf("Expected testResponse got %s", endpoints[0].Response.Name)
|
|
}
|
|
|
|
if endpoints[0].Request.Type != "testRequest" {
|
|
t.Errorf("Expected testRequest type got %s", endpoints[0].Request.Type)
|
|
}
|
|
|
|
if endpoints[0].Response.Type != "testResponse" {
|
|
t.Errorf("Expected testResponse type got %s", endpoints[0].Response.Type)
|
|
}
|
|
|
|
}
|