Go测试框架-Mock http请求

原创

Johns

修改于 2022-06-30 02:26:06

3K0

文章被收录于专栏：代码工具代码工具

一. 使用httptest来mock一些http的服务

1.1 背景说明

最近做了一个营销活动，开发礼包领取功能时依赖了外部的系统发放点券，已知点券发放接口是非常稳定ok的了，目前这个接口仅可以在生产被调用。为了测试礼包领取功能，需要mock掉依赖的外部点券发放接口。

1.2 操作实践

// 这是配置文件里面的，为了演示这里写死
// 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)
		})
	})
}