micro/service_test.go

64 lines
1.0 KiB
Go
Raw Normal View History

package micro
import (
2018-03-03 14:53:52 +03:00
"context"
"sync"
"testing"
"github.com/micro/go-micro/registry/mock"
proto "github.com/micro/go-micro/server/debug/proto"
)
func TestService(t *testing.T) {
var wg sync.WaitGroup
wg.Add(1)
// cancellation context
ctx, cancel := context.WithCancel(context.Background())
// create service
service := NewService(
Name("test.service"),
Context(ctx),
Registry(mock.NewRegistry()),
AfterStart(func() error {
wg.Done()
return nil
}),
)
// we can't test service.Init as it parses the command line
// service.Init()
2018-03-13 21:50:58 +03:00
// run service
2018-05-25 16:39:50 +03:00
go func() {
// wait for start
wg.Wait()
2018-05-25 16:39:50 +03:00
// test call debug
req := service.Client().NewRequest(
"test.service",
"Debug.Health",
new(proto.HealthRequest),
)
2018-05-25 16:39:50 +03:00
rsp := new(proto.HealthResponse)
2018-05-25 16:39:50 +03:00
err := service.Client().Call(context.TODO(), req, rsp)
if err != nil {
t.Fatal(err)
}
2018-05-25 16:39:50 +03:00
if rsp.Status != "ok" {
t.Fatalf("service response: %s", rsp.Status)
}
2018-05-25 16:39:50 +03:00
// shutdown the service
cancel()
}()
2018-05-25 16:39:50 +03:00
if err := service.Run(); err != nil {
t.Fatal(err)
}
}