openbilibili/app/service/main/vip/dao/pay_test.go

package dao

import (
	"bytes"
	"context"
	"fmt"
	"net/http"
	"net/url"
	"testing"

	"go-common/app/service/main/vip/model"
	"go-common/library/ecode"

	"github.com/smartystreets/goconvey/convey"
	"gopkg.in/h2non/gock.v1"
)

func TestDaoproductID(t *testing.T) {
	var (
		months = int16(1)
	)
	convey.Convey("productID", t, func(ctx convey.C) {
		id := d.productID(months)
		ctx.Convey("id should not be nil", func(ctx convey.C) {
			ctx.So(id, convey.ShouldEqual, _productMonthID)
		})
	})
	convey.Convey("productID == _productQuarterID", t, func(ctx convey.C) {
		months = 3
		id := d.productID(months)
		ctx.Convey("productID == 61", func(ctx convey.C) {
			ctx.So(id, convey.ShouldEqual, 61)
		})
	})
	convey.Convey("productID == _productYearID", t, func(ctx convey.C) {
		months = _yearMonths
		id := d.productID(months)
		ctx.Convey("productID == 61", func(ctx convey.C) {
			ctx.So(id, convey.ShouldEqual, _productYearID)
		})
	})
}

func TestDaoPayWallet(t *testing.T) {
	var (
		c    = context.TODO()
		mid  = int64(20606508)
		ip   = ""
		data = &model.PayAccountResp{}
	)
	convey.Convey("PayWallet", t, func(ctx convey.C) {
		defer gock.OffAll()
		httpMock("POST", _payWallet).Reply(200).JSON(`{"code":0}`)
		err := d.PayWallet(c, mid, ip, data)
		ctx.Convey("Error should be nil", func(ctx convey.C) {
			ctx.So(err, convey.ShouldBeNil)
		})
	})
}

func TestDaoPayBanks(t *testing.T) {
	var (
		c    = context.TODO()
		ip   = ""
		data []*model.PayBankResp
	)
	convey.Convey("PayBanks", t, func(ctx convey.C) {
		defer gock.OffAll()
		httpMock("GET", _payBanks).Reply(200).JSON(`{"code":0}`)
		err := d.PayBanks(c, ip, data)
		ctx.Convey("Error should be nil", func(ctx convey.C) {
			ctx.So(err, convey.ShouldBeNil)
		})
	})
}

func TestDaoAddPayOrder(t *testing.T) {
	var (
		c  = context.TODO()
		ip = ""
		o  = &model.PayOrder{
			OrderNo: "201808141642435698853709",
			Money:   148,
		}
		data = &model.AddPayOrderResp{}
	)
	convey.Convey("AddPayOrder", t, func(ctx convey.C) {
		defer gock.OffAll()
		httpMock("POST", _addPayOrder).Reply(200).JSON(`{"code":0}`)
		err := d.AddPayOrder(c, ip, o, data)
		ctx.Convey("Error should be nil", func(ctx convey.C) {
			ctx.So(err, convey.ShouldBeNil)
		})
	})
}

func TestDaoPaySDK(t *testing.T) {
	var (
		c       = context.TODO()
		ip      = ""
		o       = &model.PayOrder{}
		data    = &model.APIPayOrderResp{}
		payCode = ""
	)
	convey.Convey("PaySDK", t, func(ctx convey.C) {
		defer gock.OffAll()
		httpMock("POST", _paySDK).Reply(200).JSON(`{"code":0}`)
		err := d.PaySDK(c, ip, o, data, payCode)
		ctx.Convey("Error should be nil", func(ctx convey.C) {
			ctx.So(err, convey.ShouldBeNil)
		})
	})
}

func TestDaoPayQrcode(t *testing.T) {
	var (
		c  = context.TODO()
		ip = "120.11.188.230"
		o  = &model.PayOrder{
			OrderNo: "2014100120411262876516",
			Mid:     4908640,
			Money:   148.00,
		}
		data    = &model.APIPayOrderResp{}
		payCode = "alipay"
	)
	convey.Convey("PayQrcode", t, func(ctx convey.C) {
		defer gock.OffAll()
		httpMock("POST", _payQrcode).Reply(200).JSON(`{"code":0}`)
		err := d.PayQrcode(c, ip, o, data, payCode)
		ctx.Convey("Error should be nil", func(ctx convey.C) {
			ctx.So(err, convey.ShouldBeNil)
		})
	})
}

func TestDaoQuickPayToken(t *testing.T) {
	var (
		c         = context.TODO()
		ip        = ""
		accessKey = ""
		cookie    []*http.Cookie
		data      = &model.QucikPayResp{}
	)
	convey.Convey("QuickPayToken", t, func(ctx convey.C) {
		defer gock.OffAll()
		httpMock("POST", d.c.Property.PayURL+_quickPayToken).Reply(200).JSON(`{"code":0}`)
		err := d.QuickPayToken(c, ip, accessKey, cookie, data)
		ctx.Convey("Error should be nil", func(ctx convey.C) {
			ctx.So(err, convey.ShouldBeNil)
		})
	})
}

func TestDaoDoQuickPay(t *testing.T) {
	var (
		c            = context.TODO()
		ip           = ""
		token        = ""
		thirdTradeNo = ""
		data         = &model.PayRetResp{}
	)
	convey.Convey("DoQuickPay", t, func(ctx convey.C) {
		defer gock.OffAll()
		httpMock("POST", _quickPayDo).Reply(200).JSON(`{"code":0}`)
		err := d.DoQuickPay(c, ip, token, thirdTradeNo, data)
		ctx.Convey("Error should be nil", func(ctx convey.C) {
			ctx.So(err, convey.ShouldBeNil)
		})
	})
}

func TestDaoPayCashier(t *testing.T) {
	var (
		c        = context.TODO()
		ip       = ""
		o        = &model.PayOrder{}
		data     = &model.APIPayOrderResp{}
		payCode  = ""
		bankCode = ""
	)
	convey.Convey("PayCashier", t, func(ctx convey.C) {
		defer gock.OffAll()
		httpMock("POST", _payCashier).Reply(200).JSON(`{"code":0}`)
		err := d.PayCashier(c, ip, o, data, payCode, bankCode)
		ctx.Convey("Error should be nil", func(ctx convey.C) {
			ctx.So(err, convey.ShouldBeNil)
		})
	})
}

func TestDaoPayIapAccess(t *testing.T) {
	var (
		c  = context.TODO()
		ip = ""
		o  = &model.PayOrder{
			OrderNo:      "123",
			AppID:        1,
			Platform:     1,
			OrderType:    1,
			AppSubID:     "456",
			Mid:          20606508,
			ToMid:        20606509,
			BuyMonths:    1,
			Money:        3.0,
			Status:       1,
			PayType:      1,
			RechargeBp:   1.0,
			ThirdTradeNo: "123",
			Ver:          1,
		}
		data      = &model.APIPayOrderResp{}
		productID = "123"
	)
	convey.Convey("PayIapAccess", t, func(ctx convey.C) {
		defer gock.OffAll()
		httpMock("POST", _payIapAccess).Reply(200).JSON(`{"code":0}`)
		err := d.PayIapAccess(c, ip, o, data, productID)
		ctx.Convey("Error should be nil", func(ctx convey.C) {
			ctx.So(err, convey.ShouldBeNil)
		})
	})
}

func TestDaoPayClose(t *testing.T) {
	var (
		c       = context.TODO()
		orderNO = "123456"
		ip      = "127.0.0.1"
	)
	convey.Convey("PayClose", t, func(ctx convey.C) {
		defer gock.OffAll()
		httpMock("POST", d.payCloseURL).Reply(200).JSON(`{"code":0}`)
		_, err := d.PayClose(c, orderNO, ip)
		ctx.Convey("Error should be nil", func(ctx convey.C) {
			ctx.So(err, convey.ShouldBeNil)
		})
	})
}

func TestDaodopay(t *testing.T) {
	var (
		c       = context.TODO()
		urlPath = _payWallet
		ip      = "172.18.35.12"
		params  = url.Values{}
		data    = &model.PayAccountResp{}
		mid     = int64(20606508)
	)
	convey.Convey("dopay", t, func(ctx convey.C) {
		params.Add("mid", fmt.Sprintf("%d", mid))
		defer gock.OffAll()
		httpMock("POST", _payWallet).Reply(200).JSON(`{"code":0}`)
		err := d.dopay(c, urlPath, ip, params, data, d.client.Post)
		ctx.Convey("Error should be nil", func(ctx convey.C) {
			ctx.So(err, convey.ShouldBeNil)
		})
	})
}

func TestDaoPayRecission(t *testing.T) {
	var (
		c       = context.TODO()
		params  map[string]interface{}
		clietIP = ""
	)
	convey.Convey("PayRecission", t, func(ctx convey.C) {
		defer gock.OffAll()
		httpMock("POST", _payWallet).Reply(200).JSON(`{"code":0}`)
		err := d.PayRecission(c, params, clietIP)
		ctx.Convey("Error should be nil", func(ctx convey.C) {
			ctx.So(err, convey.ShouldEqual, ecode.VipRescissionErr)
		})
	})
}

func TestDaoPaySign(t *testing.T) {
	var (
		params map[string]string
		token  = ""
	)
	convey.Convey("PaySign", t, func(ctx convey.C) {
		sign := d.PaySign(params, token)
		ctx.Convey("sign should not be nil", func(ctx convey.C) {
			ctx.So(sign, convey.ShouldNotBeNil)
		})
	})
}

func TestDaoPaySignNotDel(t *testing.T) {
	var (
		params map[string]string
		token  = ""
	)
	convey.Convey("PaySignNotDel", t, func(ctx convey.C) {
		sign := d.PaySignNotDel(params, token)
		ctx.Convey("sign should not be nil", func(ctx convey.C) {
			ctx.So(sign, convey.ShouldNotBeNil)
		})
	})
}

func TestDaosortParamsKey(t *testing.T) {
	var (
		v map[string]string
	)
	convey.Convey("sortParamsKey", t, func(ctx convey.C) {
		p1 := d.sortParamsKey(v)
		ctx.Convey("p1 should not be nil", func(ctx convey.C) {
			ctx.So(p1, convey.ShouldNotBeNil)
		})
	})
}

func TestDaodoPaySend(t *testing.T) {
	var (
		c        = context.TODO()
		basePath = d.c.Property.PayURL
		path     = _quickPayToken
		IP       = ""
		cookie   []*http.Cookie
		header   map[string]string
		method   = http.MethodPost
		params   = map[string]string{"name": "vip"}
		data     = new(struct {
			Code int64 `json:"errno"`
			Data int64 `json:"msg"`
		})
	)
	convey.Convey("doPaySend", t, func(ctx convey.C) {
		defer gock.OffAll()
		httpMock("POST", basePath+path).Reply(200).JSON(`{"code":0}`)
		err := d.doPaySend(c, basePath, path, IP, cookie, header, method, params, data, d.PaySign)
		ctx.Convey("Error should be nil", func(ctx convey.C) {
			ctx.So(err, convey.ShouldBeNil)
		})
	})
}

func TestDaoreadAll(t *testing.T) {
	var (
		r        = bytes.NewReader([]byte{})
		capacity = int64(0)
	)
	convey.Convey("readAll", t, func(ctx convey.C) {

		b, err := readAll(r, capacity)
		ctx.Convey("Error should be nil", func(ctx convey.C) {
			ctx.So(err, convey.ShouldBeNil)
		})
		ctx.Convey("b should not be nil", func(ctx convey.C) {
			ctx.So(b, convey.ShouldNotBeNil)
		})
	})
}

// go test  -test.v -test.run TestDaoPayClose
func TestDaoPayQrCode(t *testing.T) {
	convey.Convey("TestDaoPayQrCode", t, func() {
		res, err := d.PayQrCode(context.TODO(), 1, "1", make(map[string]interface{}))
		convey.So(err, convey.ShouldBeNil)
		convey.So(res, convey.ShouldNotBeNil)
	})
}