代码拉取完成,页面将自动刷新
package gout
import (
"context"
"fmt"
"net/http"
"net/http/httptest"
"testing"
"time"
"github.com/gin-gonic/gin"
"github.com/stretchr/testify/assert"
)
func setupDataFlow(t *testing.T) *gin.Engine {
router := gin.New()
router.GET("/timeout", func(c *gin.Context) {
ctx := c.Request.Context()
select {
case <-ctx.Done():
fmt.Printf("setTimeout done\n")
case <-time.After(2 * time.Second):
assert.Fail(t, "test timeout fail")
}
})
router.GET("/setdebug", func(c *gin.Context) {
c.String(200, "setdebug")
})
return router
}
func Test_Global_SetTimeout(t *testing.T) {
router := setupDataFlow(t)
const (
longTimeout = 400
middleTimeout = 300
shortTimeout = 200
)
ts := httptest.NewServer(http.HandlerFunc(router.ServeHTTP))
defer ts.Close()
// 只设置timeout
SetTimeout(shortTimeout * time.Millisecond) //设置全局超时时间
err := GET(ts.URL + "/timeout").Do()
// 期望的结果是返回错误
assert.Error(t, err)
// 使用互斥api的原则,后面的覆盖前面的
// 这里是WithContext生效, 预期超时时间400ms
ctx, cancel := context.WithTimeout(context.Background(), longTimeout*time.Millisecond)
defer cancel()
s := time.Now()
SetTimeout(shortTimeout * time.Millisecond) // 设置全局超时时间
err = GET(ts.URL + "/timeout").
WithContext(ctx).
Do()
assert.Error(t, err)
assert.GreaterOrEqual(t, int(time.Since(s)), int(middleTimeout*time.Millisecond))
SetTimeout(time.Duration(0))
}
func Test_NewWithOpt_Timeout(t *testing.T) {
router := setupDataFlow(t)
const (
longTimeout = 400
middleTimeout = 300
shortTimeout = 200
)
ts := httptest.NewServer(http.HandlerFunc(router.ServeHTTP))
defer ts.Close()
// 只设置timeout
greq := NewWithOpt(WithTimeout(shortTimeout * time.Millisecond)) //设置全局超时时间
err := greq.GET(ts.URL + "/timeout").Do()
// 期望的结果是返回错误
assert.Error(t, err)
// 使用互斥api的原则,后面的覆盖前面的
// 这里是WithContext生效, 预期超时时间400ms
ctx, cancel := context.WithTimeout(context.Background(), longTimeout*time.Millisecond)
defer cancel()
s := time.Now()
greq = NewWithOpt(WithTimeout(shortTimeout * time.Millisecond)) // 设置全局超时时间
err = greq.GET(ts.URL + "/timeout").
WithContext(ctx).
Do()
assert.Error(t, err)
assert.GreaterOrEqual(t, int(time.Since(s)), int(middleTimeout*time.Millisecond))
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。