bilibili-backup/app/service/main/resource/dao/cpm/cpm_test.go
2019-04-22 02:59:20 +00:00

37 lines
1.4 KiB
Go

package cpm
import (
"context"
"testing"
"go-common/app/service/main/location/model"
"github.com/smartystreets/goconvey/convey"
)
func TestCpmCpmsAPP(t *testing.T) {
convey.Convey("When cpm returns code = 0", t, func(ctx convey.C) {
data := `{"code":0,"message":"successed","data":{}}`
httpMock("GET", d.cpmAppURL).Reply(200).JSON(data)
_, err := d.CpmsAPP(context.Background(), 0, 182504479, 6190, "457", "iphone", "phone", "222", "wifi", "", "", &model.Info{Addr: "218.4.147.222"})
ctx.Convey("Then Error should be nil", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
})
})
convey.Convey("When cpm returns code != 0", t, func(ctx convey.C) {
data := `{"code":-3,"message":"faild","data":{}}`
httpMock("GET", d.cpmAppURL).Reply(200).JSON(data)
_, err := d.CpmsAPP(context.Background(), 0, 182504479, 6190, "457", "iphone", "phone", "222", "wifi", "", "", &model.Info{Addr: "218.4.147.222"})
ctx.Convey("Then Error should not be nil", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
})
})
convey.Convey("When cpm http request gets 404", t, func(ctx convey.C) {
httpMock("GET", d.cpmAppURL).Reply(404)
_, err := d.CpmsAPP(context.Background(), 0, 182504479, 6190, "457", "iphone", "phone", "222", "wifi", "", "", &model.Info{Addr: "218.4.147.222"})
ctx.Convey("Then Error should not be nil", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
})
})
}