更新测试用例

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