package test

import (
	//"go-admin/models/tools"
	//"os"
	"fmt"
	"github.com/jinzhu/gorm"
	_ "github.com/jinzhu/gorm/dialects/mysql"
	"go-admin/app/admin/models"
	"testing"
	"time"
	//"text/template"
)

func TestGoModelTemplate(t *testing.T) {
	//t1, err := template.ParseFiles("model.go.template")
	//if err != nil {
	//	t.Error(err)
	//}
	//table := tools.SysTables{}
	//table.TBName = "sys_tables"
	//tab, err := table.Get()
	//if err != nil {
	//	t.Error(err)
	//}
	//file, err := os.Create("models/" + table.PackageName + ".go")
	//if err != nil {
	//	t.Error(err)
	//}
	//defer file.Close()
	//
	//_ = t1.Execute(file, tab)
	t.Log("")
}

func TestGoApiTemplate(t *testing.T) {
	//t1, err := template.ParseFiles("api.go.template")
	//if err != nil {
	//	t.Error(err)
	//}
	//table := tools.SysTables{}
	//table.TBName = "sys_tables"
	//tab, _ := table.Get()
	//file, err := os.Create("apis/" + table.PackageName + ".go")
	//if err != nil {
	//	t.Error(err)
	//}
	//defer file.Close()
	//
	//_ = t1.Execute(file, tab)
	t.Log("")
}

var DBTest *gorm.DB

func InitDBTest() {
	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")
	//site_admin:ZxcZdDjDC3k72xsN@tcp(139.224.216.42:3306)/site_admin?charset=utf8&parseTime=True&loc=Local&timeout=1000ms
	dialect := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
		"site_admin", "ZxcZdDjDC3k72xsN", "139.224.216.42", "3306",
		"site_admin")

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

		models.Strategy{},
	)

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

func TestInitDBTest(t *testing.T) {
	InitDBTest()
}

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(
		models.SysUser{},
	)

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

func TestInitDBProd(t *testing.T) {
	InitDBProd()
}