Add watcher test
This commit is contained in:
parent
96104aa209
commit
50eb97129e
@ -63,26 +63,22 @@ func TestMDNS(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// get registered service
|
// get registered service
|
||||||
|
|
||||||
s, err := r.GetService(service.Name)
|
s, err := r.GetService(service.Name)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(s) != 1 {
|
if len(s) != 1 {
|
||||||
for _, srv := range s {
|
|
||||||
t.Logf("%+v", srv)
|
|
||||||
}
|
|
||||||
t.Fatalf("Expected one result for %s got %d", service.Name, len(s))
|
t.Fatalf("Expected one result for %s got %d", service.Name, len(s))
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if s[0].Name != service.Name {
|
if s[0].Name != service.Name {
|
||||||
t.Fatal("Expected name %s got %s", service.Name, s[0].Name)
|
t.Fatalf("Expected name %s got %s", service.Name, s[0].Name)
|
||||||
}
|
}
|
||||||
|
|
||||||
if s[0].Version != service.Version {
|
if s[0].Version != service.Version {
|
||||||
t.Fatal("Expected version %s got %s", service.Version, s[0].Version)
|
t.Fatalf("Expected version %s got %s", service.Version, s[0].Version)
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(s[0].Nodes) != 1 {
|
if len(s[0].Nodes) != 1 {
|
||||||
|
164
registry/mdns/watcher_test.go
Normal file
164
registry/mdns/watcher_test.go
Normal file
@ -0,0 +1,164 @@
|
|||||||
|
package mdns
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/micro/go-micro/registry"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestWatcher(t *testing.T) {
|
||||||
|
testData := []*registry.Service{
|
||||||
|
®istry.Service{
|
||||||
|
Name: "test1",
|
||||||
|
Version: "1.0.1",
|
||||||
|
Nodes: []*registry.Node{
|
||||||
|
®istry.Node{
|
||||||
|
Id: "test1-1",
|
||||||
|
Address: "10.0.0.1",
|
||||||
|
Port: 10001,
|
||||||
|
Metadata: map[string]string{
|
||||||
|
"foo": "bar",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
®istry.Service{
|
||||||
|
Name: "test2",
|
||||||
|
Version: "1.0.2",
|
||||||
|
Nodes: []*registry.Node{
|
||||||
|
®istry.Node{
|
||||||
|
Id: "test2-1",
|
||||||
|
Address: "10.0.0.2",
|
||||||
|
Port: 10002,
|
||||||
|
Metadata: map[string]string{
|
||||||
|
"foo2": "bar2",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
®istry.Service{
|
||||||
|
Name: "test3",
|
||||||
|
Version: "1.0.3",
|
||||||
|
Nodes: []*registry.Node{
|
||||||
|
®istry.Node{
|
||||||
|
Id: "test3-1",
|
||||||
|
Address: "10.0.0.3",
|
||||||
|
Port: 10003,
|
||||||
|
Metadata: map[string]string{
|
||||||
|
"foo3": "bar3",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
// new registry
|
||||||
|
r := NewRegistry()
|
||||||
|
|
||||||
|
w, err := r.Watch()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
defer w.Stop()
|
||||||
|
|
||||||
|
for _, service := range testData {
|
||||||
|
// register service
|
||||||
|
if err := r.Register(service); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// get registered service
|
||||||
|
res, err := w.Next()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if res.Action != "create" {
|
||||||
|
t.Fatal("Expected create action for %s", service.Name)
|
||||||
|
}
|
||||||
|
|
||||||
|
s := []*registry.Service{res.Service}
|
||||||
|
|
||||||
|
if len(s) != 1 {
|
||||||
|
t.Fatalf("Expected one result for %s got %d", service.Name, len(s))
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
if s[0].Name != service.Name {
|
||||||
|
t.Fatalf("Expected name %s got %s", service.Name, s[0].Name)
|
||||||
|
}
|
||||||
|
|
||||||
|
if s[0].Version != service.Version {
|
||||||
|
t.Fatalf("Expected version %s got %s", service.Version, s[0].Version)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(s[0].Nodes) != 1 {
|
||||||
|
t.Fatalf("Expected 1 node, got %d", len(s[0].Nodes))
|
||||||
|
}
|
||||||
|
|
||||||
|
node := s[0].Nodes[0]
|
||||||
|
|
||||||
|
if node.Id != service.Nodes[0].Id {
|
||||||
|
t.Fatalf("Expected node id %s got %s", service.Nodes[0].Id, node.Id)
|
||||||
|
}
|
||||||
|
|
||||||
|
if node.Address != service.Nodes[0].Address {
|
||||||
|
t.Fatalf("Expected node address %s got %s", service.Nodes[0].Address, node.Address)
|
||||||
|
}
|
||||||
|
|
||||||
|
if node.Port != service.Nodes[0].Port {
|
||||||
|
t.Fatalf("Expected node port %s got %s", service.Nodes[0].Port, node.Port)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, service := range testData {
|
||||||
|
// deregister
|
||||||
|
if err := r.Deregister(service); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// get registered service
|
||||||
|
res, err := w.Next()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if res.Action != "delete" {
|
||||||
|
t.Fatalf("Expected create action for %s", service.Name)
|
||||||
|
}
|
||||||
|
|
||||||
|
s := []*registry.Service{res.Service}
|
||||||
|
|
||||||
|
if len(s) != 1 {
|
||||||
|
t.Fatalf("Expected one result for %s got %d", service.Name, len(s))
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
if s[0].Name != service.Name {
|
||||||
|
t.Fatalf("Expected name %s got %s", service.Name, s[0].Name)
|
||||||
|
}
|
||||||
|
|
||||||
|
if s[0].Version != service.Version {
|
||||||
|
t.Fatalf("Expected version %s got %s", service.Version, s[0].Version)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(s[0].Nodes) != 1 {
|
||||||
|
t.Fatalf("Expected 1 node, got %d", len(s[0].Nodes))
|
||||||
|
}
|
||||||
|
|
||||||
|
node := s[0].Nodes[0]
|
||||||
|
|
||||||
|
if node.Id != service.Nodes[0].Id {
|
||||||
|
t.Fatalf("Expected node id %s got %s", service.Nodes[0].Id, node.Id)
|
||||||
|
}
|
||||||
|
|
||||||
|
if node.Address != service.Nodes[0].Address {
|
||||||
|
t.Fatalf("Expected node address %s got %s", service.Nodes[0].Address, node.Address)
|
||||||
|
}
|
||||||
|
|
||||||
|
if node.Port != service.Nodes[0].Port {
|
||||||
|
t.Fatalf("Expected node port %s got %s", service.Nodes[0].Port, node.Port)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user