unistack-org/micro
4
0
Fork 3
Code Issues 19 Pull Requests 1 Actions Packages Projects Releases 259 Wiki Activity

unexport mock registry

Browse Source
This commit is contained in:
Asim 2016-03-17 22:54:18 +00:00
parent 01a642410f
commit 0a81cb1190
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
registry/mock/mock.go
View File

@ -4,7 +4,7 @@ import (
"github.com/micro/go-micro/registry" "github.com/micro/go-micro/registry"
) )
type MockRegistry struct { type mockRegistry struct {
Services map[string][]*registry.Service Services map[string][]*registry.Service
} }
@ -53,12 +53,12 @@ var (
} }
) )
func (m *MockRegistry) init() { func (m *mockRegistry) init() {
// add some mock data // add some mock data
m.Services = mockData m.Services = mockData
} }
func (m *MockRegistry) GetService(service string) ([]*registry.Service, error) { func (m *mockRegistry) GetService(service string) ([]*registry.Service, error) {
s, ok := m.Services[service] s, ok := m.Services[service]
if !ok { if !ok {
return nil, registry.ErrNotFound return nil, registry.ErrNotFound
@ -67,7 +67,7 @@ func (m *MockRegistry) GetService(service string) ([]*registry.Service, error) {
} }
func (m *MockRegistry) ListServices() ([]*registry.Service, error) { func (m *mockRegistry) ListServices() ([]*registry.Service, error) {
var services []*registry.Service var services []*registry.Service
for _, service := range m.Services { for _, service := range m.Services {
services = append(services, service...) services = append(services, service...)
@ -75,28 +75,28 @@ func (m *MockRegistry) ListServices() ([]*registry.Service, error) {
return services, nil return services, nil
} }
func (m *MockRegistry) Register(s *registry.Service, opts ...registry.RegisterOption) error { func (m *mockRegistry) Register(s *registry.Service, opts ...registry.RegisterOption) error {
services := addServices(m.Services[s.Name], []*registry.Service{s}) services := addServices(m.Services[s.Name], []*registry.Service{s})
m.Services[s.Name] = services m.Services[s.Name] = services
return nil return nil
} }
func (m *MockRegistry) Deregister(s *registry.Service) error { func (m *mockRegistry) Deregister(s *registry.Service) error {
services := delServices(m.Services[s.Name], []*registry.Service{s}) services := delServices(m.Services[s.Name], []*registry.Service{s})
m.Services[s.Name] = services m.Services[s.Name] = services
return nil return nil
} }
func (m *MockRegistry) Watch() (registry.Watcher, error) { func (m *mockRegistry) Watch() (registry.Watcher, error) {
return nil, nil return nil, nil
} }
func (m *MockRegistry) String() string { func (m *mockRegistry) String() string {
return "mock" return "mock"
} }
func NewRegistry() *MockRegistry { func NewRegistry() registry.Registry {
m := &MockRegistry{Services: make(map[string][]*registry.Service)} m := &mockRegistry{Services: make(map[string][]*registry.Service)}
m.init() m.init()
return m return m
} }