Function TestProxyStats

pkg/topom/topom_stats_test.go:19–65

(x *testing.T)

)

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

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

	p1, c1 := openProxy()
	defer c1.Shutdown()

	p2, c2 := openProxy()
	defer c2.Shutdown()

	contextCreateProxy(t, p1)
	check([]string{p1.Token}, []string{})

	contextCreateProxy(t, p2)
	check([]string{p1.Token, p2.Token}, []string{})

	assert.MustNoError(c1.Shutdown())
	check([]string{p2.Token}, []string{p1.Token})

	assert.MustNoError(c2.Shutdown())
	check([]string{}, []string{p1.Token, p2.Token})

	p3, c3 := openProxy()
	defer c3.Shutdown()

	contextCreateProxy(t, p3)
	check([]string{p3.Token}, []string{p1.Token, p2.Token})

	contextRemoveProxy(t, p1)
	check([]string{p3.Token}, []string{p2.Token})
}

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