func setup(hostIP string) (*Listener, store.Store, *mockInspector) { st := inmem.NewInMemStore() dc := newMockInspector() return NewListener(Config{ Store: st, HostIP: hostIP, Inspector: dc, }), st, dc }
func TestList(t *testing.T) { st := inmem.NewInMemStore() err := st.AddService("foo", data.Service{}) require.NoError(t, err) opts := &listOpts{} bufout, buferr := opts.tapOutput() err = runOptsWithStore(opts, st, []string{}) require.NoError(t, err) require.Equal(t, "foo\n", bufout.String()) require.Equal(t, "", buferr.String()) }
func TestListServices(t *testing.T) { st := inmem.NewInMemStore() st.AddService("svc", testService) st.AddInstance("svc", "inst", testInstance) resp := doRequest(t, st, "/api/services") require.Equal(t, 200, resp.Code) var deets []*store.ServiceInfo require.Nil(t, json.Unmarshal(resp.Body.Bytes(), &deets)) services := allServices(t, st) require.Equal(t, services, deets) }
func newBalancerAgent(t *testing.T) *BalancerAgent { dir, err := ioutil.TempDir("", "balagent_test") require.Nil(t, err) return &BalancerAgent{ errorSink: daemon.NewErrorSink(), store: inmem.NewInMemStore(), filename: path.Join(dir, "output"), generated: make(chan struct{}), updaterStopped: make(chan struct{}), generatorStopped: make(chan struct{}), } }
func runOpts(opts commandOpts, args []string) (store.Store, error) { st := inmem.NewInMemStore() return st, runOptsWithStore(opts, st, args) }