|
|
@@ -5,14 +5,13 @@ |
|
|
|
package goes |
|
|
|
|
|
|
|
import ( |
|
|
|
. "gopkg.in/check.v1" |
|
|
|
"net/http" |
|
|
|
"net/url" |
|
|
|
"os" |
|
|
|
"strings" |
|
|
|
"testing" |
|
|
|
"time" |
|
|
|
|
|
|
|
. "github.com/go-check/check" |
|
|
|
) |
|
|
|
|
|
|
|
var ( |
|
|
@@ -41,7 +40,7 @@ func (s *GoesTestSuite) SetUpTest(c *C) { |
|
|
|
|
|
|
|
func (s *GoesTestSuite) TestNewClient(c *C) { |
|
|
|
conn := NewClient(ESHost, ESPort) |
|
|
|
c.Assert(conn, DeepEquals, &Client{ESHost, ESPort, http.DefaultClient, ""}) |
|
|
|
c.Assert(conn, DeepEquals, &Client{ESHost, ESPort, http.DefaultClient, "", "", ""}) |
|
|
|
} |
|
|
|
|
|
|
|
func (s *GoesTestSuite) TestWithHTTPClient(c *C) { |
|
|
@@ -54,7 +53,7 @@ func (s *GoesTestSuite) TestWithHTTPClient(c *C) { |
|
|
|
} |
|
|
|
conn := NewClient(ESHost, ESPort).WithHTTPClient(cl) |
|
|
|
|
|
|
|
c.Assert(conn, DeepEquals, &Client{ESHost, ESPort, cl, ""}) |
|
|
|
c.Assert(conn, DeepEquals, &Client{ESHost, ESPort, cl, "", "", ""}) |
|
|
|
c.Assert(conn.Client.Transport.(*http.Transport).DisableCompression, Equals, true) |
|
|
|
c.Assert(conn.Client.Transport.(*http.Transport).ResponseHeaderTimeout, Equals, 1*time.Second) |
|
|
|
} |
|
|
|