registry/memory: add support for domain options (#1713)

* registry/memory: add support for the domain options

* registry/memory: swap Fatal test cases with Error

* registry/memory: fix wildcard not found bug

* registry/memory: replace locks with rlocks

* registry/memory: fix deregistration bug
This commit is contained in:
ben-toogood
2020-06-18 12:39:19 +01:00
committed by GitHub
parent 5fd36d6cc0
commit 8c7c27c573
4 changed files with 251 additions and 72 deletions

View File

@@ -244,3 +244,39 @@ func TestMemoryRegistryTTLConcurrent(t *testing.T) {
}
}
}
func TestMemoryWildcard(t *testing.T) {
m := NewRegistry()
testSrv := &registry.Service{Name: "foo", Version: "1.0.0"}
if err := m.Register(testSrv, registry.RegisterDomain("one")); err != nil {
t.Fatalf("Register err: %v", err)
}
if err := m.Register(testSrv, registry.RegisterDomain("two")); err != nil {
t.Fatalf("Register err: %v", err)
}
if recs, err := m.ListServices(registry.ListDomain("one")); err != nil {
t.Errorf("List err: %v", err)
} else if len(recs) != 1 {
t.Errorf("Expected 1 record, got %v", len(recs))
}
if recs, err := m.ListServices(registry.ListDomain("*")); err != nil {
t.Errorf("List err: %v", err)
} else if len(recs) != 2 {
t.Errorf("Expected 2 records, got %v", len(recs))
}
if recs, err := m.GetService(testSrv.Name, registry.GetDomain("one")); err != nil {
t.Errorf("Get err: %v", err)
} else if len(recs) != 1 {
t.Errorf("Expected 1 record, got %v", len(recs))
}
if recs, err := m.GetService(testSrv.Name, registry.GetDomain("*")); err != nil {
t.Errorf("Get err: %v", err)
} else if len(recs) != 2 {
t.Errorf("Expected 2 records, got %v", len(recs))
}
}