package client import ( "context" "fmt" "testing" "github.com/micro/go-micro/client/selector" "github.com/micro/go-micro/errors" "github.com/micro/go-micro/registry" "github.com/micro/go-micro/registry/memory" ) func newTestRegistry() registry.Registry { r := memory.NewRegistry() reg := r.(*memory.Registry) reg.Services = testData return reg } func TestCallAddress(t *testing.T) { var called bool service := "test.service" endpoint := "Test.Endpoint" address := "10.1.10.1" port := 8080 wrap := func(cf CallFunc) CallFunc { return func(ctx context.Context, node *registry.Node, req Request, rsp interface{}, opts CallOptions) error { called = true if req.Service() != service { return fmt.Errorf("expected service: %s got %s", service, req.Service()) } if req.Endpoint() != endpoint { return fmt.Errorf("expected service: %s got %s", endpoint, req.Endpoint()) } if node.Address != address { return fmt.Errorf("expected address: %s got %s", address, node.Address) } if node.Port != port { return fmt.Errorf("expected address: %d got %d", port, node.Port) } // don't do the call return nil } } r := newTestRegistry() c := NewClient( Registry(r), WrapCall(wrap), ) c.Options().Selector.Init(selector.Registry(r)) req := c.NewRequest(service, endpoint, nil) // test calling remote address if err := c.Call(context.Background(), req, nil, WithAddress(fmt.Sprintf("%s:%d", address, port))); err != nil { t.Fatal("call with address error", err) } if !called { t.Fatal("wrapper not called") } } func TestCallRetry(t *testing.T) { service := "test.service" endpoint := "Test.Endpoint" address := "10.1.10.1" var called int wrap := func(cf CallFunc) CallFunc { return func(ctx context.Context, node *registry.Node, req Request, rsp interface{}, opts CallOptions) error { called++ if called == 1 { return errors.InternalServerError("test.error", "retry request") } // don't do the call return nil } } r := newTestRegistry() c := NewClient( Registry(r), WrapCall(wrap), ) c.Options().Selector.Init(selector.Registry(r)) req := c.NewRequest(service, endpoint, nil) // test calling remote address if err := c.Call(context.Background(), req, nil, WithAddress(address)); err != nil { t.Fatal("call with address error", err) } // num calls if called < c.Options().CallOptions.Retries+1 { t.Fatal("request not retried") } } func TestCallWrapper(t *testing.T) { var called bool id := "test.1" service := "test.service" endpoint := "Test.Endpoint" address := "10.1.10.1" port := 8080 wrap := func(cf CallFunc) CallFunc { return func(ctx context.Context, node *registry.Node, req Request, rsp interface{}, opts CallOptions) error { called = true if req.Service() != service { return fmt.Errorf("expected service: %s got %s", service, req.Service()) } if req.Endpoint() != endpoint { return fmt.Errorf("expected service: %s got %s", endpoint, req.Endpoint()) } if node.Address != address { return fmt.Errorf("expected address: %s got %s", address, node.Address) } // don't do the call return nil } } r := newTestRegistry() c := NewClient( Registry(r), WrapCall(wrap), ) c.Options().Selector.Init(selector.Registry(r)) r.Register(®istry.Service{ Name: service, Version: "latest", Nodes: []*registry.Node{ ®istry.Node{ Id: id, Address: address, Port: port, }, }, }) req := c.NewRequest(service, endpoint, nil) if err := c.Call(context.Background(), req, nil); err != nil { t.Fatal("call wrapper error", err) } if !called { t.Fatal("wrapper not called") } }