Skip to content

Commit

Permalink
test: add guarentee for server run
Browse files Browse the repository at this point in the history
  • Loading branch information
@welkeyever
welkeyever committed Aug 21, 2024
1 parent 76f13da commit d82d201
Showing 1 changed file with 108 additions and 18 deletions.
126 changes: 108 additions & 18 deletions pkg/app/client/client_test.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,12 @@ func TestCloseIdleConnections(t *testing.T) {
defer func() {
engine.Close()
}()
time.Sleep(time.Millisecond * 500)
for {
time.Sleep(1 * time.Second)
if engine.IsRunning() {
break
}
}

c, _ := NewClient(WithDialer(newMockDialerWithCustomFunc(opt.Network, opt.Addr, 1*time.Second, nil)))

Expand Down Expand Up @@ -136,7 +141,12 @@ func TestClientInvalidURI(t *testing.T) {
defer func() {
engine.Close()
}()
time.Sleep(time.Millisecond * 500)
for {
time.Sleep(1 * time.Second)
if engine.IsRunning() {
break
}
}

c, _ := NewClient(WithDialer(newMockDialerWithCustomFunc(opt.Network, opt.Addr, 1*time.Second, nil)))
req, res := protocol.AcquireRequest(), protocol.AcquireResponse()
Expand Down Expand Up @@ -170,7 +180,12 @@ func TestClientGetWithBody(t *testing.T) {
defer func() {
engine.Close()
}()
time.Sleep(time.Millisecond * 500)
for {
time.Sleep(1 * time.Second)
if engine.IsRunning() {
break
}
}

c, _ := NewClient(WithDialer(newMockDialerWithCustomFunc(opt.Network, opt.Addr, 1*time.Second, nil)))
req, res := protocol.AcquireRequest(), protocol.AcquireResponse()
Expand Down Expand Up @@ -205,7 +220,12 @@ func TestClientPostBodyStream(t *testing.T) {
defer func() {
engine.Close()
}()
time.Sleep(time.Millisecond * 500)
for {
time.Sleep(1 * time.Second)
if engine.IsRunning() {
break
}
}

cStream, _ := NewClient(WithDialer(newMockDialerWithCustomFunc(opt.Network, opt.Addr, 1*time.Second, nil)), WithResponseBodyStream(true))
args := &protocol.Args{}
Expand Down Expand Up @@ -246,7 +266,12 @@ func TestClientURLAuth(t *testing.T) {
defer func() {
engine.Close()
}()
time.Sleep(time.Millisecond * 500)
for {
time.Sleep(1 * time.Second)
if engine.IsRunning() {
break
}
}

c, _ := NewClient(WithDialer(newMockDialerWithCustomFunc(opt.Network, opt.Addr, 1*time.Second, nil)))
for up, expected := range cases {
Expand Down Expand Up @@ -278,7 +303,12 @@ func TestClientNilResp(t *testing.T) {
defer func() {
engine.Close()
}()
time.Sleep(time.Millisecond * 500)
for {
time.Sleep(1 * time.Second)
if engine.IsRunning() {
break
}
}

c, _ := NewClient(WithDialer(newMockDialerWithCustomFunc(opt.Network, opt.Addr, 1*time.Second, nil)))

Expand All @@ -304,7 +334,12 @@ func TestClientParseConn(t *testing.T) {
defer func() {
engine.Close()
}()
time.Sleep(time.Millisecond * 500)
for {
time.Sleep(1 * time.Second)
if engine.IsRunning() {
break
}
}

c, _ := NewClient(WithDialer(newMockDialerWithCustomFunc(opt.Network, opt.Addr, 1*time.Second, nil)))
req, res := protocol.AcquireRequest(), protocol.AcquireResponse()
Expand Down Expand Up @@ -346,7 +381,12 @@ func TestClientPostArgs(t *testing.T) {
defer func() {
engine.Close()
}()
time.Sleep(time.Millisecond * 500)
for {
time.Sleep(1 * time.Second)
if engine.IsRunning() {
break
}
}
c, _ := NewClient(WithDialer(newMockDialerWithCustomFunc(opt.Network, opt.Addr, 1*time.Second, nil)))
req, res := protocol.AcquireRequest(), protocol.AcquireResponse()
defer func() {
Expand Down Expand Up @@ -497,7 +537,12 @@ func TestClientDefaultUserAgent(t *testing.T) {
defer func() {
engine.Close()
}()
time.Sleep(time.Millisecond * 500)
for {
time.Sleep(1 * time.Second)
if engine.IsRunning() {
break
}
}

c, _ := NewClient(WithDialer(newMockDialerWithCustomFunc(opt.Network, opt.Addr, 1*time.Second, nil)))
req := protocol.AcquireRequest()
Expand Down Expand Up @@ -531,7 +576,12 @@ func TestClientSetUserAgent(t *testing.T) {
defer func() {
engine.Close()
}()
time.Sleep(time.Millisecond * 500)
for {
time.Sleep(1 * time.Second)
if engine.IsRunning() {
break
}
}

userAgent := "I'm not hertz"
c, _ := NewClient(WithDialer(newMockDialerWithCustomFunc(opt.Network, opt.Addr, time.Second, nil)), WithName(userAgent))
Expand Down Expand Up @@ -562,7 +612,12 @@ func TestClientNoUserAgent(t *testing.T) {
defer func() {
engine.Close()
}()
time.Sleep(time.Millisecond * 500)
for {
time.Sleep(1 * time.Second)
if engine.IsRunning() {
break
}
}
c, _ := NewClient(WithDialer(newMockDialerWithCustomFunc(opt.Network, opt.Addr, time.Second, nil)), WithDialTimeout(1*time.Second), WithNoDefaultUserAgentHeader(true))

req := protocol.AcquireRequest()
Expand Down Expand Up @@ -643,7 +698,12 @@ func TestClientDoWithCustomHeaders(t *testing.T) {
defer func() {
engine.Close()
}()
time.Sleep(time.Millisecond * 500)
for {
time.Sleep(1 * time.Second)
if engine.IsRunning() {
break
}
}

// make sure that the client sends all the request headers and body.
c, _ := NewClient(WithDialer(newMockDialerWithCustomFunc(opt.Network, opt.Addr, 1*time.Second, nil)))
Expand Down Expand Up @@ -688,7 +748,12 @@ func TestClientDoTimeoutDisablePathNormalizing(t *testing.T) {
defer func() {
engine.Close()
}()
time.Sleep(time.Millisecond * 500)
for {
time.Sleep(1 * time.Second)
if engine.IsRunning() {
break
}
}
c, _ := NewClient(WithDialer(newMockDialerWithCustomFunc(opt.Network, opt.Addr, time.Second, nil)), WithDisablePathNormalizing(true))

urlWithEncodedPath := "http://example.com/encoded/Y%2BY%2FY%3D/stuff"
Expand Down Expand Up @@ -818,7 +883,12 @@ func TestHostClientMaxConnsWithDeadline(t *testing.T) {
defer func() {
engine.Close()
}()
time.Sleep(time.Millisecond * 500)
for {
time.Sleep(1 * time.Second)
if engine.IsRunning() {
break
}
}

c := &http1.HostClient{
ClientOptions: &http1.ClientOptions{
Expand Down Expand Up @@ -887,7 +957,12 @@ func TestHostClientMaxConnDuration(t *testing.T) {
defer func() {
engine.Close()
}()
time.Sleep(time.Millisecond * 500)
for {
time.Sleep(1 * time.Second)
if engine.IsRunning() {
break
}
}

c := &http1.HostClient{
ClientOptions: &http1.ClientOptions{
Expand Down Expand Up @@ -932,7 +1007,12 @@ func TestHostClientMultipleAddrs(t *testing.T) {
defer func() {
engine.Close()
}()
time.Sleep(time.Millisecond * 500)
for {
time.Sleep(1 * time.Second)
if engine.IsRunning() {
break
}
}

dialsCount := make(map[string]int)
c := &http1.HostClient{
Expand Down Expand Up @@ -1093,7 +1173,12 @@ func TestHostClientMaxConnWaitTimeoutSuccess(t *testing.T) {
defer func() {
engine.Close()
}()
time.Sleep(time.Millisecond * 500)
for {
time.Sleep(1 * time.Second)
if engine.IsRunning() {
break
}
}

c := &http1.HostClient{
ClientOptions: &http1.ClientOptions{
Expand Down Expand Up @@ -1162,7 +1247,12 @@ func TestHostClientMaxConnWaitTimeoutError(t *testing.T) {
defer func() {
engine.Close()
}()
time.Sleep(time.Millisecond * 500)
for {
time.Sleep(1 * time.Second)
if engine.IsRunning() {
break
}
}

c := &http1.HostClient{
ClientOptions: &http1.ClientOptions{
Expand Down

0 comments on commit d82d201

Please sign in to comment.