func testServices(s store.Store, t *testing.T) {
require.Nil(t, s.AddService("svc", testService))
svc2, err := s.GetService("svc", store.QueryServiceOptions{})
require.Nil(t, err)
require.Equal(t, "svc", svc2.Name)
require.Equal(t, testService, svc2.Service)
require.Nil(t, s.CheckRegisteredService("svc"))
services := func() map[string]data.Service {
svcs := make(map[string]data.Service)
ss, err := s.GetAllServices(store.QueryServiceOptions{})
require.Nil(t, err)
for _, svc := range ss {
svcs[svc.Name] = svc.Service
}
return svcs
}
require.Equal(t, map[string]data.Service{"svc": testService}, services())
require.Nil(t, s.RemoveService("svc"))
require.Equal(t, map[string]data.Service{}, services())
require.Nil(t, s.AddService("svc", testService))
require.Nil(t, s.RemoveAllServices())
require.Equal(t, map[string]data.Service{}, services())
}
func testRules(s store.Store, t *testing.T) {
require.Nil(t, s.AddService("svc", testService))
require.Nil(t, s.SetContainerRule("svc", "group", testRule))
svc, err := s.GetService("svc", store.QueryServiceOptions{WithContainerRules: true})
require.Nil(t, err)
require.Equal(t, []store.ContainerRuleInfo{
store.ContainerRuleInfo{
Name: "group",
ContainerRule: testRule,
},
}, svc.ContainerRules)
require.Nil(t, s.RemoveContainerRule("svc", "group"))
svc, err = s.GetService("svc", store.QueryServiceOptions{WithContainerRules: true})
require.Nil(t, err)
require.Empty(t, svc.ContainerRules)
}