package registry import ( "os" "testing" "time" ) func TestMDNS(t *testing.T) { // skip test in travis because of sendto: operation not permitted error if travis := os.Getenv("TRAVIS"); travis == "true" { t.Skip() } testData := []*Service{ { Name: "test1", Version: "1.0.1", Nodes: []*Node{ { Id: "test1-1", Address: "10.0.0.1:10001", Metadata: map[string]string{ "foo": "bar", }, }, }, }, { Name: "test2", Version: "1.0.2", Nodes: []*Node{ { Id: "test2-1", Address: "10.0.0.2:10002", Metadata: map[string]string{ "foo2": "bar2", }, }, }, }, { Name: "test3", Version: "1.0.3", Nodes: []*Node{ { Id: "test3-1", Address: "10.0.0.3:10003", Metadata: map[string]string{ "foo3": "bar3", }, }, }, }, } travis := os.Getenv("TRAVIS") var opts []Option if travis == "true" { opts = append(opts, Timeout(time.Millisecond*100)) } // new registry r := NewRegistry(opts...) for _, service := range testData { // register service if err := r.Register(service); err != nil { t.Fatal(err) } // get registered service s, err := r.GetService(service.Name) if err != nil { t.Fatal(err) } 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) } } services, err := r.ListServices() if err != nil { t.Fatal(err) } for _, service := range testData { var seen bool for _, s := range services { if s.Name == service.Name { seen = true break } } if !seen { t.Fatalf("Expected service %s got nothing", service.Name) } // deregister if err := r.Deregister(service); err != nil { t.Fatal(err) } time.Sleep(time.Millisecond * 5) // check its gone s, _ := r.GetService(service.Name) if len(s) > 0 { t.Fatalf("Expected nothing got %+v", s[0]) } } } func TestEncoding(t *testing.T) { testData := []*mdnsTxt{ { Version: "1.0.0", Metadata: map[string]string{ "foo": "bar", }, Endpoints: []*Endpoint{ { Name: "endpoint1", Request: &Value{ Name: "request", Type: "request", }, Response: &Value{ Name: "response", Type: "response", }, Metadata: map[string]string{ "foo1": "bar1", }, }, }, }, } for _, d := range testData { encoded, err := encode(d) if err != nil { t.Fatal(err) } for _, txt := range encoded { if len(txt) > 255 { t.Fatalf("One of parts for txt is %d characters", len(txt)) } } decoded, err := decode(encoded) if err != nil { t.Fatal(err) } if decoded.Version != d.Version { t.Fatalf("Expected version %s got %s", d.Version, decoded.Version) } if len(decoded.Endpoints) != len(d.Endpoints) { t.Fatalf("Expected %d endpoints, got %d", len(d.Endpoints), len(decoded.Endpoints)) } for k, v := range d.Metadata { if val := decoded.Metadata[k]; val != v { t.Fatalf("Expected %s=%s got %s=%s", k, v, k, val) } } } }