mirror of https://github.com/linka-cloud/grpc.git
33 lines
651 B
Go
33 lines
651 B
Go
package mdns
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"gitlab.bertha.cloud/partitio/lab/grpc/registry"
|
|
)
|
|
|
|
func TestRegistry(t *testing.T) {
|
|
assert := assert.New(t)
|
|
reg := NewRegistry()
|
|
svc := ®istry.Service{Name: "test", Nodes: []*registry.Node{{Id: "test-1", Address: "127.0.0.1:8888"}}}
|
|
if err := reg.Register(svc); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
svcs, err := reg.GetService("test")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Len(svcs, 1)
|
|
assert.Contains(svcs, svc)
|
|
if err := reg.Deregister(svc); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
svcs, err = reg.ListServices()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Len(svcs, 0)
|
|
}
|