Function TestGetHost

scope_test.go:111–202

(t *testing.T)

}

func TestGetHost(t *testing.T) {
	c := &cluster{
		name:     "default",
		replicas: []*replica{{}},
	}
	r := c.replicas[0]
	r.cluster = c
	r.hosts = []*topology.Node{
		topology.NewNode(&url.URL{Host: "127.0.0.1"}, nil, "", r.name, topology.WithDefaultActiveState(true)),
		topology.NewNode(&url.URL{Host: "127.0.0.2"}, nil, "", r.name, topology.WithDefaultActiveState(true)),
		topology.NewNode(&url.URL{Host: "127.0.0.3"}, nil, "", r.name, topology.WithDefaultActiveState(true)),
	}

	// step | expected  | hosts running queries
	// 1    | 127.0.0.2 | 0, 1, 0
	// 2    | 127.0.0.3 | 0, 1, 1
	// 3    | 127.0.0.1 | 1, 1, 1
	// 4    | 127.0.0.2 | 1, 2, 2
	// 5    | 127.0.0.1 | 2, 2, 2
	// 6    | 127.0.0.1 | 3, 7, 2	// 2nd is penalized for `penaltySize`
	// 7    | 127.0.0.1 | 3, 7, 3

	// step: 1
	h := c.getHost()
	expected := "127.0.0.2"
	if h.Host() != expected {
		t.Fatalf("got host %q; expected %q", h.Host(), expected)
	}
	h.IncrementConnections()

	// step: 2
	h = c.getHost()
	expected = "127.0.0.3"
	if h.Host() != expected {
		t.Fatalf("got host %q; expected %q", h.Host(), expected)
	}
	h.IncrementConnections()

	// step: 3
	h = c.getHost()
	expected = "127.0.0.1"
	if h.Host() != expected {
		t.Fatalf("got host %q; expected %q", h.Host(), expected)
	}
	h.IncrementConnections()

	// step: 4
	h = c.getHost()
	expected = "127.0.0.2"
	if h.Host() != expected {
		t.Fatalf("got host %q; expected %q", h.Host(), expected)
	}
	h.IncrementConnections()

	// inc last host to get least-loaded 1st host
	r.hosts[2].IncrementConnections()

	// step: 5