This commit is contained in:
xinfei.wu 2018-12-20 13:52:07 +08:00 committed by Vasiliy Tolstov
parent 0374a2de2c
commit 60d75699e4

View File

@ -11,6 +11,7 @@ import (
api "github.com/micro/go-api/proto" api "github.com/micro/go-api/proto"
micro "github.com/micro/go-micro" micro "github.com/micro/go-micro"
"github.com/micro/go-micro/broker"
"github.com/micro/go-micro/client" "github.com/micro/go-micro/client"
"github.com/micro/go-micro/registry" "github.com/micro/go-micro/registry"
"github.com/micro/go-micro/registry/mock" "github.com/micro/go-micro/registry/mock"
@ -98,7 +99,10 @@ func TestSubscriber(t *testing.T) {
reg := mock.NewRegistry() reg := mock.NewRegistry()
// create server // create server
srv := NewServer(server.Registry(reg)) srv := NewServer(
server.Registry(reg),
server.Broker(broker.NewBroker(broker.Registry(reg))),
)
// register handler // register handler
if err := srv.Handle(srv.NewHandler(http.NewServeMux())); err != nil { if err := srv.Handle(srv.NewHandler(http.NewServeMux())); err != nil {
@ -133,6 +137,7 @@ func TestSubscriber(t *testing.T) {
defer wg.Done() defer wg.Done()
mClient := client.NewClient( mClient := client.NewClient(
client.Registry(reg), client.Registry(reg),
client.Transport(srv.Options().Transport),
client.Broker(srv.Options().Broker), client.Broker(srv.Options().Broker),
) )
pub := micro.NewPublisher(topic, mClient) pub := micro.NewPublisher(topic, mClient)
@ -148,6 +153,11 @@ func TestSubscriber(t *testing.T) {
t.Fatalf("subscriber is not working, err: %s", ctx.Err()) t.Fatalf("subscriber is not working, err: %s", ctx.Err())
} }
// deregister server
if err := srv.Deregister(); err != nil {
t.Fatal(err)
}
// stop server // stop server
if err := srv.Stop(); err != nil { if err := srv.Stop(); err != nil {
t.Fatal(err) t.Fatal(err)