From 5215daf6ebf0316a57b09029ffe7b71608baafa1 Mon Sep 17 00:00:00 2001 From: Dimitri Sokolyuk Date: Mon, 28 Mar 2016 04:43:07 +0200 Subject: Not ready jet --- server_test.go | 28 ++++++++++++++++++++-------- 1 file changed, 20 insertions(+), 8 deletions(-) (limited to 'server_test.go') 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) } -- cgit v1.2.3