You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 

157 lines
3.8 KiB

package servicecomb
import (
"context"
"testing"
pb "github.com/go-chassis/cari/discovery"
"github.com/go-chassis/sc-client"
"github.com/go-kratos/kratos/v2/registry"
"github.com/gofrs/uuid"
)
var r *Registry
func init() {
r = NewRegistry(&mockClient{})
}
type mockClient struct{}
func (receiver *mockClient) WatchMicroService(microServiceID string, callback func(*sc.MicroServiceInstanceChangedEvent)) error {
return nil
}
//nolint
func (receiver *mockClient) FindMicroServiceInstances(consumerID,
appID, microServiceName, versionRule string, opts ...sc.CallOption,
) ([]*pb.MicroServiceInstance, error) {
if microServiceName == "KratosServicecomb" {
return []*pb.MicroServiceInstance{{}}, nil
}
return nil, nil
}
func (receiver *mockClient) RegisterService(microService *pb.MicroService) (string, error) {
return "", nil
}
func (receiver *mockClient) RegisterMicroServiceInstance(microServiceInstance *pb.MicroServiceInstance) (string, error) {
return "", nil
}
func (receiver *mockClient) Heartbeat(microServiceID, microServiceInstanceID string) (bool, error) {
return true, nil
}
func (receiver *mockClient) UnregisterMicroServiceInstance(microServiceID, microServiceInstanceID string) (bool, error) {
return true, nil
}
func (receiver *mockClient) GetMicroServiceID(appID, microServiceName, version, env string, opts ...sc.CallOption) (string, error) {
return "", nil
}
func TestRegistry(t *testing.T) {
instanceID, err := uuid.NewV4()
if err != nil {
t.Fatal(err)
}
svc := &registry.ServiceInstance{
Name: "KratosServicecomb",
Version: "0.0.1",
Metadata: map[string]string{"app": "kratos"},
Endpoints: []string{"tcp://127.0.0.1:9000?isSecure=false"},
ID: instanceID.String(),
}
ctx := context.TODO()
t.Run("Register test, expected: success.", func(t *testing.T) {
err = r.Register(ctx, svc)
if err != nil {
t.Error(err)
}
})
t.Run("GetService test, expected: success.", func(t *testing.T) {
var insts []*registry.ServiceInstance
insts, err = r.GetService(ctx, svc.Name)
if err != nil {
t.Fatal(err)
}
if len(insts) <= 0 {
t.Errorf("inst len less than 0")
}
})
t.Run("Deregister test, expected: success.", func(t *testing.T) {
svc.ID = instanceID.String()
err = r.Deregister(ctx, svc)
if err != nil {
t.Error(err)
}
})
}
func TestWatcher(t *testing.T) {
instanceID1, err := uuid.NewV4()
if err != nil {
t.Fatal(err)
}
svc1 := &registry.ServiceInstance{
Name: "WatcherTest",
Version: "0.0.1",
Metadata: map[string]string{"app": "kratos"},
Endpoints: []string{"tcp://127.0.0.1:9000?isSecure=false"},
ID: instanceID1.String(),
}
ctx := context.TODO()
err = r.Register(ctx, svc1)
if err != nil {
t.Fatal(err)
}
w, err := r.Watch(ctx, "WatcherTest")
if err != nil {
t.Fatal(err)
}
if w == nil {
t.Fatal("w is nil")
}
sbWatcher := w.(*Watcher)
t.Run("Watch register event, expected: success", func(t *testing.T) {
go sbWatcher.Put(svc1)
var instances []*registry.ServiceInstance
instances, err = w.Next()
if err != nil {
t.Fatal(err)
}
if len(instances) == 0 {
t.Errorf("instances is empty")
}
if instanceID1.String() != instances[0].ID {
t.Errorf("expected %v, got %v", instanceID1.String(), instances[0].ID)
}
})
t.Run("Watch deregister event, expected: success", func(t *testing.T) {
// Deregister instance1.
err = r.Deregister(ctx, svc1)
if err != nil {
t.Fatal(err)
}
go sbWatcher.Put(svc1)
var instances []*registry.ServiceInstance
instances, err = w.Next()
if err != nil {
t.Fatal(err)
}
if len(instances) == 0 {
t.Errorf("instances is empty")
}
if instanceID1.String() != instances[0].ID {
t.Errorf("expected %v, got %v", instanceID1.String(), instances[0].ID)
}
})
t.Run("Stop test, expected: success", func(t *testing.T) {
err = w.Stop()
if err != nil {
t.Error(err)
}
})
}