aboutsummaryrefslogtreecommitdiff
path: root/server_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'server_test.go')
-rw-r--r--server_test.go28
1 files changed, 20 insertions, 8 deletions
diff --git a/server_test.go b/server_test.go
index 9dd10c9..9a8e651 100644
--- a/server_test.go
+++ b/server_test.go
@@ -5,6 +5,7 @@ import (
"io/ioutil"
"net/http"
"net/http/httptest"
+ "net/url"
"testing"
)
@@ -21,20 +22,31 @@ func TestNewServer(t *testing.T) {
t.Error(err)
}
- select {
- case err := <-server.Start():
+ rpcServer := httptest.NewServer(nil)
+ defer rpcServer.Close()
+
+ rpcURL, err := url.Parse(rpcServer.URL)
+ if err != nil {
+ t.Error(err)
+ }
+
+ err = Add(Entry{ServerName: "whatever", Upstream: backendServer.URL}, rpcURL.Host)
+ if err != nil {
t.Error(err)
- default:
}
- if err = Add(Entry{
- ServerName: "localhost",
- Upstream: backendServer.URL,
- }, ""); err != nil {
+ frontendServer := httptest.NewServer(server.Handler)
+ if err != nil {
t.Error(err)
}
- resp, err := http.Get("http://localhost")
+ client := &http.Client{}
+ req, err := http.NewRequest("GET", frontendServer.URL, nil)
+ if err != nil {
+ t.Error(err)
+ }
+ req.Header.Set("Host", "whatever")
+ resp, err := client.Do(req)
if err != nil {
t.Error(err)
}