mh_server/model/model_test.go

package model

import (
	"fmt"
	"github.com/codinl/go-logger"
	"github.com/jinzhu/gorm"
	"math/rand"
	"testing"
	"time"
)

func TestGameList(t *testing.T) {
	InitTestDB()
	var game GameCard
	err := NewGameCardQuerySet(GormDb).IDEq(1).One(&game)
	if err != nil {
		logger.Error("err:", err)
		return
	}
	logger.Infof("game %#v", game)
	fmt.Printf("game %#v", game)
	gameName := game.Name
	for i := 50; i < 200; i++ {
		randNum := getRandNum()
		numUnix := getRandNumUnix()

		num := fmt.Sprintf("0%d", i+1)
		game.Name = num + gameName
		game.NewProducts = 2
		game.Price = uint32(randNum)
		game.OrderCount = uint32(randNum + numUnix)
		game.ID = 0
		GormDb.Create(&game)
	}
}

func getRandNum() int {
	rand.Seed(time.Now().UnixNano())
	return rand.Intn(1000)
}

func getRandNumUnix() int {
	rand.Seed(time.Now().Unix())
	return rand.Intn(1000)
}

var GormDb *gorm.DB

// 测试服务器
func InitTestDB() {
	var err error

	dialect := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
		"mh_dev", "m5h4parZsXjbeiFh", "39.108.188.218", "3306", "mh_dev")

	db, err := gorm.Open("mysql", dialect)
	if err != nil {
		logger.Infof("Open DB error [%s] with dialect[%s]", err.Error(), dialect)
		panic(err)
	}

	if db.DB().Ping() != nil {
		fmt.Printf("Ping DB fail: %#v", db.Error)
		panic(err)
	}
	db.SingularTable(true)

	db.AutoMigrate(
		&User{},
		//&LotteryInviteTargetUser{},
		&GameCard{},
		&Collection{},
		&Config{},
		&HistoryBrowsing{},
		&ShoppingCart{},
		&UserAddress{},
		&SearchHistory{},
		&Order{},
		&HotSearch{},
		&HomeCarousel{},
		&GameCardType{},
		&Store{},
		&GameType{},
		&UserOpenMemberRecord{},
		&CommonProblem{},
		&ExpressCompany{},
		&GameCardGoodsStock{},
		&GameCardGoods{},
		&UserInvite{},
		&ArticleTitlePanel{},
	)

	fmt.Println("DB init success")
	DB = db
	logger.Info("ok")
}

var DBProd *gorm.DB

func InitDBProd() {
	var err error
	dialect := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
		"mh_pro", "fLeytnBFCRB553ab", "39.108.188.218", "3306",
		"mh_pro")

	DBProd, err = gorm.Open("mysql", dialect)
	if err != nil {
		fmt.Println("Connect Database Error, error = ", err)
		panic(err)
	}
	DBProd.DB().SetMaxOpenConns(1000)
	DBProd.DB().SetMaxIdleConns(500)
	DBProd.DB().SetConnMaxLifetime(time.Second * 10)
	DBProd.LogMode(true)
	DBProd.SingularTable(true)
	DBProd.AutoMigrate()

	if err := DBProd.DB().Ping(); err != nil {
		fmt.Println("Ping Database Fail, error = ", err)
		panic(err)
	}
	fmt.Println("DB init success")
}

func TestHotSearch_TableName(t *testing.T) {
	InitTestDB()

	list := []string{"马里奥赛车8豪华版", "任天堂明星大乱斗 特别版", "超级马力欧 奥德赛", "塞尔达传说：旷野之息", "宝可梦：剑 / 盾", "精灵宝可梦 Let's Go! 皮卡丘/Let's Go! 伊布", "喷射战士2", "超级马力欧派对", "新超级马里奥兄弟U", "路易基鬼屋3", "超级马力欧创作家2", "塞尔达传说:织梦岛", "集合了！动物森友会", "1-2 Switch", "马里奥网球Aces", "马里奥+疯兔：王国之战", "火焰之纹章：风花雪月", "星之卡比：新星同盟", "大金刚国度：热带寒流"}
	for i, s := range list {
		hotSearch := &HotSearch{
			Keyword: s,
			Sort:    uint32(len(list) - i),
		}
		err := hotSearch.Create(GormDb)
		if err != nil {
			logger.Error("err:", err)
		}
	}
}

func TestGetHomeCarouselList(t *testing.T) {
	InitTestDB()
	//imgs := []string{"1-2011060021334X.jpg", "1-2011060021425N.jpg", "1-201106002150294.jpg", "1-201106002159522.jpg"}
	imgs := []string{"http://39.108.188.218:8000/img/1-2011060021334X.jpg", "http://39.108.188.218:8000/img/1-2011060021425N.jpg", "http://39.108.188.218:8000/img/1-201106002150294.jpg", "http://39.108.188.218:8000/img/1-201106002159522.jpg"}
	for i, img := range imgs {
		homeCarousel := &HomeCarousel{
			Name:   "banner" + fmt.Sprintf("%d", len(imgs)-i),
			Img:    img,
			Sort:   uint32(len(imgs) - i),
			IsShow: 1,
			Detail: "",
			Link:   "",
		}
		err := homeCarousel.Create(GormDb)
		if err != nil {
			logger.Error("err:", err)
		}
	}
}

func TestGameCardType_Create(t *testing.T) {
	InitTestDB()
	types := []string{"角色扮演", "射击游戏", "动作游戏", "冒险游戏", "赛车游戏", "策略游戏", "即时战略", "养成游戏", "桌面棋牌", "休闲益智", "体育运动", "模拟经营", "音乐游戏", "格斗游戏", "动作角色", "第一人称射击", "第三人称射击"}
	for i, s := range types {
		cardType := GameCardType{
			Name: s,
			Sort: uint32(i + 100),
		}
		err := cardType.Create(GormDb)
		if err != nil {
			logger.Error("err:", err)
		}
	}

}

func TestMemberExprireTime(t *testing.T) {
	InitTestDB()
}

func TestStore_TableName(t *testing.T) {
	InitTestDB()

	for i := 0; i < 10; i++ {
		idx := fmt.Sprintf("%d", i+1)
		store := Store{
			Name:      "门店" + idx,
			Img:       "http://39.108.188.218:8000/img/1599445323_430355.jpg",
			Tel:       "021-54473036",
			Province:  "广东省",
			City:      "深圳市",
			District:  "南山区",
			Address:   "高新园",
			Longitude: 0,
			Latitude:  0,
		}
		err := store.Create(GormDb)
		if err != nil {
			logger.Error("err:", err)
		}
	}

}

func TestPess(t *testing.T) {
	InitTestDB()
	comp := []string{"圆通速递", "韵达快递", "中通快递", "顺丰速运", "申通快递", "百世快递", "邮政快递包裹", "京东物流", "EMS", "极兔速递", "EMS包裹"}
	compCode := []string{"yuantong", "yunda", "zhongtong", "shunfeng", "shentong", "huitongkuaidi", "youzhengguonei", "jd", "ems", "jtexpress", "emsbg"}
	for i, _ := range comp {
		company := &ExpressCompany{
			CompanyName: comp[i],
			CompanyCode: compCode[i],
		}
		err := company.Create(GormDb)
		if err != nil {
			logger.Error("err:", err)
		}
	}
}

func TestGoodsStock(t *testing.T) {
	InitTestDB()
	cards := make([]GameCard, 0)
	err := NewGameCardQuerySet(GormDb).All(&cards)
	if err != nil {
		logger.Error("err:", err)
	}
	stores := make([]Store, 0)
	err = NewStoreQuerySet(GormDb).All(&stores)
	if err != nil {
		logger.Error("err:", err)
	}

	for _, store := range stores {
		for _, card := range cards {
			stock := &GameCardGoodsStock{
				StoreId:    uint64(store.ID),
				GameCardId: uint64(card.ID),
				StoreStock: 1000,
				RentStock:  10,
			}
			err := stock.Create(GormDb)
			if err != nil {
				logger.Error("err:", err)
			}
		}
	}
}

func TestGoodsCard(t *testing.T) {
	InitTestDB()
	cards := make([]GameCard, 0)
	err := NewGameCardQuerySet(GormDb).All(&cards)
	if err != nil {
		logger.Error("err:", err)
	}
	stores := make([]Store, 0)
	err = NewStoreQuerySet(GormDb).All(&stores)
	if err != nil {
		logger.Error("err:", err)
	}

	for _, store := range stores {
		for _, card := range cards {
			stock := &GameCardGoodsStock{
				StoreId:    uint64(store.ID),
				GameCardId: uint64(card.ID),
				StoreStock: 1000,
				RentStock:  10,
			}
			err := stock.Create(GormDb)
			if err != nil {
				logger.Error("err:", err)
			}
		}
	}
}

func TestOrderList(t *testing.T) {
	InitTestDB()
	count, err := NewOrderQuerySet(GormDb.Debug()).UidEq(uint64(55191926)).PayStatusEq(PayStatusPaid).CardStatusIn([]uint8{1, 2, 3}...).Count()
	if err != nil {
		logger.Error("err:", err)

	}

	fmt.Println("count:", count)
}

func TestGenShareCode(t *testing.T) {

}

func TestUserBundleFirstOrderStore(t *testing.T) {
	UserBundleFirstOrderStore()
}

func UserBundleFirstOrderStore() {
	InitDBProd()
	var users []User
	//err := NewUserQuerySet(DBProd).StoreIdEq(0).MemberLevelGte(2).OrderAscByID().All(&users)
	err := NewUserQuerySet(DBProd).StoreTypeEq(1).OrderAscByID().All(&users)
	if err != nil {
		fmt.Println("users err:", err)
		return
	}

	for i, _ := range users {
		//var order Order
		//err := NewOrderQuerySet(DBProd).UidEq(uint64(users[i].Uid)).PayStatusEq(PayStatusPaid).
		//	CardStatusEq(OrderCardStatusCompleted).OrderAscByCreatedAt().Limit(1).One(&order)
		//if err != nil {
		//	fmt.Println("store_id err:", err)
		//	//return
		//}

		//if i > 3{
		//	return
		//}
		if users[i].Uid == 0 {
			continue
		}
		//sql := fmt.Sprintf("UPDATE `user` SET store_id =%d,store_type=1 WHERE uid = %d;", order.StoreId, users[i].Uid)
		//fmt.Println("sql:", sql)
		//err = DBProd.Exec(sql).Error
		//if err != nil {
		//	fmt.Println("store_id err:", err)
		//	return
		//}
		userInvite := &UserInvite{
			FromUid:        0,
			UserType:       0,
			StoreId:        users[i].StoreId,
			MemberOpenTime: users[i].MemberExpire.AddDate(-1, 0, 0),
			MemberType:     2,
			MemberStatus:   2,
			ToUid:          users[i].Uid,
			StoreType:      1,
		}

		err = DBProd.Create(userInvite).Error
		if err != nil {
			fmt.Println("create err:", err)
		}
	}
}