Function TestRedisStats

pkg/topom/topom_stats_test.go:67–122

(x *testing.T)

}

func TestRedisStats(x *testing.T) {
	t := openTopom()
	defer t.Close()

	check := func(succ, fail []string) {
		w, err := t.RefreshRedisStats(time.Second * 5)
		assert.MustNoError(err)
		m := w.Wait()
		assert.Must(len(m) == len(succ)+len(fail))
		for _, addr := range succ {
			s, ok := m[addr].(*RedisStats)
			assert.Must(ok && s != nil && s.Stats != nil)
		}
		for _, addr := range fail {
			s, ok := m[addr].(*RedisStats)
			assert.Must(ok && s != nil && s.Stats == nil)
			assert.Must(s.Error != nil)
		}
	}

	g := &models.Group{Id: 1}

	s1 := newFakeServer()
	defer s1.Close()

	s2 := newFakeServer()
	defer s2.Close()

	g.Servers = []*models.GroupServer{
		&models.GroupServer{Addr: s1.Addr},
		&models.GroupServer{Addr: s2.Addr},
	}

	check([]string{}, []string{})

	contextCreateGroup(t, g)
	check([]string{s1.Addr, s2.Addr}, []string{})

	s1.Close()
	check([]string{s2.Addr}, []string{s1.Addr})

	s2.Close()
	check([]string{}, []string{s1.Addr, s2.Addr})

	s3 := newFakeServer()
	defer s3.Close()

	g.Servers = []*models.GroupServer{
		&models.GroupServer{Addr: s3.Addr},
	}
	contextUpdateGroup(t, g)
	check([]string{s3.Addr}, []string{})

	contextRemoveGroup(t, g)
	check([]string{}, []string{})
}

type fakeServer struct {