更新测试用例

This commit is contained in:
余 欣怀 2023-03-13 16:16:41 +08:00
parent fbfb1d80a8
commit 89c18a567f

View File

@ -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)
}