Go测试框架-Mock http请求

// 这是配置文件里面的，为了演示这里写死
// RemoteGamePointsApi 点券api地址
var RemoteGamePointsApi = "/gift_pack"

// RemoteGamePointsServiceAddress 点券服务地址
var RemoteGamePointsServiceAddress = ""

// GiftPackResp 外部点券发放接口响应
type GiftPackResp struct {
	QQ         string
	GamePoints int64
}

// GiftPackGet 礼包领取功能
func GiftPackGet(qq string) error {
	// step1. 参数校验

	// step2. 检查用户是否在本次活动黑名单 ...

	// step3.检查用户是否已经领取过了 ...

	// step4. 锁定当前账号, 防止同一个时刻同一个账号同时发放  ...

	// step5. 调用点券发放接口
	return sendGamePoints(qq)
}

// SendGamePoints 调用外部接口发送游戏点券
func sendGamePoints(qq string) error {
	if !qqValidator(qq) {
		return errors.New("qq is invalid")
	}

	// 发送点券
	resp, err := http.Get(fmt.Sprintf("%s/%s?qq=%s", RemoteGamePointsServiceAddress, RemoteGamePointsApi, qq))
	defer resp.Body.Close()

	// 解析响应结果
	bs, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return err
	}

	ret := GiftPackResp{}
	err = json.Unmarshal(bs, &ret)
	if err != nil {
		return err
	}
	fmt.Println(ret)
	return err

}

// qq 合法性校验
func qqValidator(qq string) bool {
	reg := regexp.MustCompile(`^[1-9]\d{3,10}$`)
	return reg.MatchString(qq)
}

// Test_GiftPackGet 测试礼包发放接口
func Test_GiftPackGet(t *testing.T) {
	Convey("test utils GiftPackGet", t, func() {
		Convey("When request use httptest mock remote api", func() {
			ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				if r.Method != http.MethodGet {
					w.WriteHeader(http.StatusNotFound)
				}
				if r.URL.EscapedPath() != RemoteGamePointsApi {
					w.WriteHeader(http.StatusNotFound)
				}
				qq := r.URL.Query().Get("qq")
				w.WriteHeader(http.StatusOK)
				g := GiftPackResp{qq, 100}
				bs, _ := json.Marshal(g)
				w.Write(bs)
			}))

			defer ts.Close()
			RemoteGamePointsServiceAddress = ts.URL
			err := GiftPackGet("3095764312")
			So(err, ShouldBeNil)
		})
	})
}