mirror of
https://github.com/caddyserver/caddy.git
synced 2025-01-09 20:38:49 +03:00
Merge pull request #347 from tw4452852/my_health
proxy: make tests workable when offline
This commit is contained in:
commit
7f98a6cccf
1 changed files with 16 additions and 1 deletions
|
@ -1,9 +1,24 @@
|
||||||
package proxy
|
package proxy
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"os"
|
||||||
"testing"
|
"testing"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
var workableServer *httptest.Server
|
||||||
|
|
||||||
|
func TestMain(m *testing.M) {
|
||||||
|
workableServer = httptest.NewServer(http.HandlerFunc(
|
||||||
|
func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
// do nothing
|
||||||
|
}))
|
||||||
|
r := m.Run()
|
||||||
|
workableServer.Close()
|
||||||
|
os.Exit(r)
|
||||||
|
}
|
||||||
|
|
||||||
type customPolicy struct{}
|
type customPolicy struct{}
|
||||||
|
|
||||||
func (r *customPolicy) Select(pool HostPool) *UpstreamHost {
|
func (r *customPolicy) Select(pool HostPool) *UpstreamHost {
|
||||||
|
@ -13,7 +28,7 @@ func (r *customPolicy) Select(pool HostPool) *UpstreamHost {
|
||||||
func testPool() HostPool {
|
func testPool() HostPool {
|
||||||
pool := []*UpstreamHost{
|
pool := []*UpstreamHost{
|
||||||
{
|
{
|
||||||
Name: "http://google.com", // this should resolve (healthcheck test)
|
Name: workableServer.URL, // this should resolve (healthcheck test)
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
Name: "http://shouldnot.resolve", // this shouldn't
|
Name: "http://shouldnot.resolve", // this shouldn't
|
||||||
|
|
Loading…
Reference in a new issue