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.
58 lines
1.0 KiB
58 lines
1.0 KiB
package polaris
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/polarismesh/polaris-go"
|
|
|
|
"github.com/go-kratos/kratos/v2/registry"
|
|
)
|
|
|
|
// TestRegistry
|
|
func TestRegistry(t *testing.T) {
|
|
sdk, err := polaris.NewSDKContextByAddress("127.0.0.1:8091")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
p := New(sdk)
|
|
|
|
r := p.Registry(
|
|
WithRegistryTimeout(time.Second),
|
|
WithRegistryHealthy(true),
|
|
WithRegistryIsolate(false),
|
|
WithRegistryRetryCount(3),
|
|
WithRegistryWeight(100),
|
|
WithRegistryTTL(1000),
|
|
)
|
|
|
|
ins := ®istry.ServiceInstance{
|
|
ID: "test-ut",
|
|
Name: "test-ut",
|
|
Version: "v1.0.0",
|
|
Endpoints: []string{
|
|
"grpc://127.0.0.1:8080",
|
|
"http://127.0.0.1:9090",
|
|
},
|
|
}
|
|
|
|
err = r.Register(context.Background(), ins)
|
|
|
|
t.Cleanup(func() {
|
|
if err = r.Deregister(context.Background(), ins); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
})
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
time.Sleep(time.Second * 3)
|
|
service, err := r.GetService(context.Background(), "test-ut")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Log(service)
|
|
}
|
|
|