micro-register-mdns/watcher_test.go

package mdns

import (
	"testing"

	"github.com/micro/go-micro/registry"
)

func TestWatcher(t *testing.T) {
	testData := []*registry.Service{
		&registry.Service{
			Name:    "test1",
			Version: "1.0.1",
			Nodes: []*registry.Node{
				&registry.Node{
					Id:      "test1-1",
					Address: "10.0.0.1",
					Port:    10001,
					Metadata: map[string]string{
						"foo": "bar",
					},
				},
			},
		},
		&registry.Service{
			Name:    "test2",
			Version: "1.0.2",
			Nodes: []*registry.Node{
				&registry.Node{
					Id:      "test2-1",
					Address: "10.0.0.2",
					Port:    10002,
					Metadata: map[string]string{
						"foo2": "bar2",
					},
				},
			},
		},
		&registry.Service{
			Name:    "test3",
			Version: "1.0.3",
			Nodes: []*registry.Node{
				&registry.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)
		}
	}

	for i := 0; i < 3; i++ {
		// get registered service
		res, err := w.Next()
		if err != nil {
			t.Fatal(err)
		}

		if res.Action != "create" {
			t.Fatal("Expected create action for %s", res.Service.Name)
		}

		var service *registry.Service

		for _, srv := range testData {
			if res.Service.Name == srv.Name {
				service = res.Service
				break
			}
		}

		if service == nil {
			t.Fatal("Got watch result for unknown service %s", res.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", res.Service.Name)
		}

		var service *registry.Service

		for _, srv := range testData {
			if res.Service.Name == srv.Name {
				service = res.Service
				break
			}
		}

		if service == nil {
			t.Fatal("Got watch result for unknown service %s", res.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)
		}
	}

}