diff options
Diffstat (limited to 'middleware/proxy/policy_test.go')
-rw-r--r-- | middleware/proxy/policy_test.go | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/middleware/proxy/policy_test.go b/middleware/proxy/policy_test.go index 8f4f1f792..e55fceadc 100644 --- a/middleware/proxy/policy_test.go +++ b/middleware/proxy/policy_test.go @@ -28,13 +28,16 @@ func (r *customPolicy) Select(pool HostPool) *UpstreamHost { func testPool() HostPool { pool := []*UpstreamHost{ { - Name: workableServer.URL, // this should resolve (healthcheck test) + Name: workableServer.URL, // this should resolve (healthcheck test) + Unhealthy: newBool(), }, { - Name: "http://shouldnot.resolve", // this shouldn't + Name: "http://shouldnot.resolve", // this shouldn't + Unhealthy: newBool(), }, { - Name: "http://C", + Name: "http://C", + Unhealthy: newBool(), }, } return HostPool(pool) @@ -54,7 +57,7 @@ func TestRoundRobinPolicy(t *testing.T) { t.Error("Expected second round robin host to be third host in the pool.") } // mark host as down - pool[0].Unhealthy = true + *pool[0].Unhealthy = true h = rrPolicy.Select(pool) if h != pool[1] { t.Error("Expected third round robin host to be first host in the pool.") |