Function TestConsistentHashSearch

loadbalancer/algorithm_test.go:298–338

(t *testing.T)

}

func TestConsistentHashSearch(t *testing.T) {
	apply := func(key string, endpoints []string) string {
		p := NewAlgorithmProvider()
		endpointRegistry := routing.NewEndpointRegistry(routing.RegistryOptions{})
		defer endpointRegistry.Close()
		r := &routing.Route{
			Route: eskip.Route{
				BackendType: eskip.LBBackend,
				LBAlgorithm: ConsistentHash.String(),
				LBEndpoints: eskip.NewLBEndpoints(endpoints),
			},
		}
		p.Do([]*routing.Route{r})
		endpointRegistry.Do([]*routing.Route{r})

		ch := newConsistentHash(endpoints).(*consistentHash)
		ctx := &routing.LBContext{Route: r, LBEndpoints: r.LBEndpoints, Params: map[string]interface{}{ConsistentHashKey: key}}
		return endpoints[ch.search(key, ctx)]
	}

	endpoints := []string{"http://127.0.0.1:8080", "http://127.0.0.2:8080", "http://127.0.0.3:8080"}
	const key = "192.168.0.1"

	ep := apply(key, endpoints)

	// remove endpoint
	endpoints = endpoints[1:]

	ep1 := apply(key, endpoints)
	if ep != ep1 {
		t.Errorf("expected to select %s, got %s", ep, ep1)
	}

	// add endpoint
	endpoints = append(endpoints, "http://127.0.0.4:8080")

	ep2 := apply(key, endpoints)
	if ep != ep2 {
		t.Errorf("expected to select %s, got %s", ep, ep2)
	}
}

func TestConsistentHashBoundedLoadSearch(t *testing.T) {
	endpoints := []string{"http://127.0.0.1:8080", "http://127.0.0.2:8080", "http://127.0.0.3:8080"}