You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							273 lines
						
					
					
						
							7.4 KiB
						
					
					
				
			
		
		
		
			
			
			
				
					
				
				
					
				
			
		
		
	
	
							273 lines
						
					
					
						
							7.4 KiB
						
					
					
				| // Copyright 2015 Matthew Holt and The Caddy Authors | |
| // | |
| // Licensed under the Apache License, Version 2.0 (the "License"); | |
| // you may not use this file except in compliance with the License. | |
| // You may obtain a copy of the License at | |
| // | |
| //     http://www.apache.org/licenses/LICENSE-2.0 | |
| // | |
| // Unless required by applicable law or agreed to in writing, software | |
| // distributed under the License is distributed on an "AS IS" BASIS, | |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
| // See the License for the specific language governing permissions and | |
| // limitations under the License. | |
|  | |
| package reverseproxy | |
| 
 | |
| import ( | |
| 	"net/http" | |
| 	"net/http/httptest" | |
| 	"testing" | |
| ) | |
| 
 | |
| func testPool() UpstreamPool { | |
| 	return UpstreamPool{ | |
| 		{Host: new(upstreamHost)}, | |
| 		{Host: new(upstreamHost)}, | |
| 		{Host: new(upstreamHost)}, | |
| 	} | |
| } | |
| 
 | |
| func TestRoundRobinPolicy(t *testing.T) { | |
| 	pool := testPool() | |
| 	rrPolicy := new(RoundRobinSelection) | |
| 	req, _ := http.NewRequest("GET", "/", nil) | |
| 
 | |
| 	h := rrPolicy.Select(pool, req) | |
| 	// First selected host is 1, because counter starts at 0 | |
| 	// and increments before host is selected | |
| 	if h != pool[1] { | |
| 		t.Error("Expected first round robin host to be second host in the pool.") | |
| 	} | |
| 	h = rrPolicy.Select(pool, req) | |
| 	if h != pool[2] { | |
| 		t.Error("Expected second round robin host to be third host in the pool.") | |
| 	} | |
| 	h = rrPolicy.Select(pool, req) | |
| 	if h != pool[0] { | |
| 		t.Error("Expected third round robin host to be first host in the pool.") | |
| 	} | |
| 	// mark host as down | |
| 	pool[1].SetHealthy(false) | |
| 	h = rrPolicy.Select(pool, req) | |
| 	if h != pool[2] { | |
| 		t.Error("Expected to skip down host.") | |
| 	} | |
| 	// mark host as up | |
| 	pool[1].SetHealthy(true) | |
| 
 | |
| 	h = rrPolicy.Select(pool, req) | |
| 	if h == pool[2] { | |
| 		t.Error("Expected to balance evenly among healthy hosts") | |
| 	} | |
| 	// mark host as full | |
| 	pool[1].CountRequest(1) | |
| 	pool[1].MaxRequests = 1 | |
| 	h = rrPolicy.Select(pool, req) | |
| 	if h != pool[2] { | |
| 		t.Error("Expected to skip full host.") | |
| 	} | |
| } | |
| 
 | |
| func TestLeastConnPolicy(t *testing.T) { | |
| 	pool := testPool() | |
| 	lcPolicy := new(LeastConnSelection) | |
| 	req, _ := http.NewRequest("GET", "/", nil) | |
| 
 | |
| 	pool[0].CountRequest(10) | |
| 	pool[1].CountRequest(10) | |
| 	h := lcPolicy.Select(pool, req) | |
| 	if h != pool[2] { | |
| 		t.Error("Expected least connection host to be third host.") | |
| 	} | |
| 	pool[2].CountRequest(100) | |
| 	h = lcPolicy.Select(pool, req) | |
| 	if h != pool[0] && h != pool[1] { | |
| 		t.Error("Expected least connection host to be first or second host.") | |
| 	} | |
| } | |
| 
 | |
| func TestIPHashPolicy(t *testing.T) { | |
| 	pool := testPool() | |
| 	ipHash := new(IPHashSelection) | |
| 	req, _ := http.NewRequest("GET", "/", nil) | |
| 
 | |
| 	// We should be able to predict where every request is routed. | |
| 	req.RemoteAddr = "172.0.0.1:80" | |
| 	h := ipHash.Select(pool, req) | |
| 	if h != pool[1] { | |
| 		t.Error("Expected ip hash policy host to be the second host.") | |
| 	} | |
| 	req.RemoteAddr = "172.0.0.2:80" | |
| 	h = ipHash.Select(pool, req) | |
| 	if h != pool[1] { | |
| 		t.Error("Expected ip hash policy host to be the second host.") | |
| 	} | |
| 	req.RemoteAddr = "172.0.0.3:80" | |
| 	h = ipHash.Select(pool, req) | |
| 	if h != pool[2] { | |
| 		t.Error("Expected ip hash policy host to be the third host.") | |
| 	} | |
| 	req.RemoteAddr = "172.0.0.4:80" | |
| 	h = ipHash.Select(pool, req) | |
| 	if h != pool[1] { | |
| 		t.Error("Expected ip hash policy host to be the second host.") | |
| 	} | |
| 
 | |
| 	// we should get the same results without a port | |
| 	req.RemoteAddr = "172.0.0.1" | |
| 	h = ipHash.Select(pool, req) | |
| 	if h != pool[1] { | |
| 		t.Error("Expected ip hash policy host to be the second host.") | |
| 	} | |
| 	req.RemoteAddr = "172.0.0.2" | |
| 	h = ipHash.Select(pool, req) | |
| 	if h != pool[1] { | |
| 		t.Error("Expected ip hash policy host to be the second host.") | |
| 	} | |
| 	req.RemoteAddr = "172.0.0.3" | |
| 	h = ipHash.Select(pool, req) | |
| 	if h != pool[2] { | |
| 		t.Error("Expected ip hash policy host to be the third host.") | |
| 	} | |
| 	req.RemoteAddr = "172.0.0.4" | |
| 	h = ipHash.Select(pool, req) | |
| 	if h != pool[1] { | |
| 		t.Error("Expected ip hash policy host to be the second host.") | |
| 	} | |
| 
 | |
| 	// we should get a healthy host if the original host is unhealthy and a | |
| 	// healthy host is available | |
| 	req.RemoteAddr = "172.0.0.1" | |
| 	pool[1].SetHealthy(false) | |
| 	h = ipHash.Select(pool, req) | |
| 	if h != pool[2] { | |
| 		t.Error("Expected ip hash policy host to be the third host.") | |
| 	} | |
| 
 | |
| 	req.RemoteAddr = "172.0.0.2" | |
| 	h = ipHash.Select(pool, req) | |
| 	if h != pool[2] { | |
| 		t.Error("Expected ip hash policy host to be the third host.") | |
| 	} | |
| 	pool[1].SetHealthy(true) | |
| 
 | |
| 	req.RemoteAddr = "172.0.0.3" | |
| 	pool[2].SetHealthy(false) | |
| 	h = ipHash.Select(pool, req) | |
| 	if h != pool[0] { | |
| 		t.Error("Expected ip hash policy host to be the first host.") | |
| 	} | |
| 	req.RemoteAddr = "172.0.0.4" | |
| 	h = ipHash.Select(pool, req) | |
| 	if h != pool[1] { | |
| 		t.Error("Expected ip hash policy host to be the second host.") | |
| 	} | |
| 
 | |
| 	// We should be able to resize the host pool and still be able to predict | |
| 	// where a req will be routed with the same IP's used above | |
| 	pool = UpstreamPool{ | |
| 		{Host: new(upstreamHost)}, | |
| 		{Host: new(upstreamHost)}, | |
| 	} | |
| 	req.RemoteAddr = "172.0.0.1:80" | |
| 	h = ipHash.Select(pool, req) | |
| 	if h != pool[0] { | |
| 		t.Error("Expected ip hash policy host to be the first host.") | |
| 	} | |
| 	req.RemoteAddr = "172.0.0.2:80" | |
| 	h = ipHash.Select(pool, req) | |
| 	if h != pool[1] { | |
| 		t.Error("Expected ip hash policy host to be the second host.") | |
| 	} | |
| 	req.RemoteAddr = "172.0.0.3:80" | |
| 	h = ipHash.Select(pool, req) | |
| 	if h != pool[0] { | |
| 		t.Error("Expected ip hash policy host to be the first host.") | |
| 	} | |
| 	req.RemoteAddr = "172.0.0.4:80" | |
| 	h = ipHash.Select(pool, req) | |
| 	if h != pool[1] { | |
| 		t.Error("Expected ip hash policy host to be the second host.") | |
| 	} | |
| 
 | |
| 	// We should get nil when there are no healthy hosts | |
| 	pool[0].SetHealthy(false) | |
| 	pool[1].SetHealthy(false) | |
| 	h = ipHash.Select(pool, req) | |
| 	if h != nil { | |
| 		t.Error("Expected ip hash policy host to be nil.") | |
| 	} | |
| } | |
| 
 | |
| func TestFirstPolicy(t *testing.T) { | |
| 	pool := testPool() | |
| 	firstPolicy := new(FirstSelection) | |
| 	req := httptest.NewRequest(http.MethodGet, "/", nil) | |
| 
 | |
| 	h := firstPolicy.Select(pool, req) | |
| 	if h != pool[0] { | |
| 		t.Error("Expected first policy host to be the first host.") | |
| 	} | |
| 
 | |
| 	pool[0].SetHealthy(false) | |
| 	h = firstPolicy.Select(pool, req) | |
| 	if h != pool[1] { | |
| 		t.Error("Expected first policy host to be the second host.") | |
| 	} | |
| } | |
| 
 | |
| func TestURIHashPolicy(t *testing.T) { | |
| 	pool := testPool() | |
| 	uriPolicy := new(URIHashSelection) | |
| 
 | |
| 	request := httptest.NewRequest(http.MethodGet, "/test", nil) | |
| 	h := uriPolicy.Select(pool, request) | |
| 	if h != pool[0] { | |
| 		t.Error("Expected uri policy host to be the first host.") | |
| 	} | |
| 
 | |
| 	pool[0].SetHealthy(false) | |
| 	h = uriPolicy.Select(pool, request) | |
| 	if h != pool[1] { | |
| 		t.Error("Expected uri policy host to be the first host.") | |
| 	} | |
| 
 | |
| 	request = httptest.NewRequest(http.MethodGet, "/test_2", nil) | |
| 	h = uriPolicy.Select(pool, request) | |
| 	if h != pool[1] { | |
| 		t.Error("Expected uri policy host to be the second host.") | |
| 	} | |
| 
 | |
| 	// We should be able to resize the host pool and still be able to predict | |
| 	// where a request will be routed with the same URI's used above | |
| 	pool = UpstreamPool{ | |
| 		{Host: new(upstreamHost)}, | |
| 		{Host: new(upstreamHost)}, | |
| 	} | |
| 
 | |
| 	request = httptest.NewRequest(http.MethodGet, "/test", nil) | |
| 	h = uriPolicy.Select(pool, request) | |
| 	if h != pool[0] { | |
| 		t.Error("Expected uri policy host to be the first host.") | |
| 	} | |
| 
 | |
| 	pool[0].SetHealthy(false) | |
| 	h = uriPolicy.Select(pool, request) | |
| 	if h != pool[1] { | |
| 		t.Error("Expected uri policy host to be the first host.") | |
| 	} | |
| 
 | |
| 	request = httptest.NewRequest(http.MethodGet, "/test_2", nil) | |
| 	h = uriPolicy.Select(pool, request) | |
| 	if h != pool[1] { | |
| 		t.Error("Expected uri policy host to be the second host.") | |
| 	} | |
| 
 | |
| 	pool[0].SetHealthy(false) | |
| 	pool[1].SetHealthy(false) | |
| 	h = uriPolicy.Select(pool, request) | |
| 	if h != nil { | |
| 		t.Error("Expected uri policy policy host to be nil.") | |
| 	} | |
| }
 | |
| 
 |