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:
@@ -244,3 +244,39 @@ func TestMemoryRegistryTTLConcurrent(t *testing.T) {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestMemoryWildcard(t *testing.T) {
|
||||
m := NewRegistry()
|
||||
testSrv := ®istry.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))
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user