Initial commit

2024-03-27 23:25:08 +08:00
commit 0884384e91
127 changed files with 9353 additions and 0 deletions
@@ -0,0 +1,40 @@
+package operation
+
+import (
+	_ "embed"
+	"os"
+	"testing"
+
+	"github.com/bwmarrin/snowflake"
+	"pet-house.com/core/g"
+	"pet-house.com/core/helper/str"
+	"pet-house.com/core/server/database"
+	"pet-house.com/core/server/zap_server"
+)
+
+func TestMain(m *testing.M) {
+
+	node, _ := snowflake.NewNode(1)
+	uuid := str.Join("operation", "_", node.Generate().String())
+
+	database.CONFIG.DbName = uuid
+	database.CONFIG.Path = g.TestMysqlAddr
+	database.CONFIG.Password = g.TestMysqlPwd
+	database.Instance()
+
+	code := m.Run()
+
+	err := database.DorpDB(database.CONFIG.BaseDsn(), "mysql", uuid)
+	if err != nil {
+		zap_server.ZAPLOG.Error(err.Error())
+	}
+
+	db, _ := database.Instance().DB()
+	if db != nil {
+		db.Close()
+	}
+	database.Remove()
+	Remove()
+	zap_server.Remove()
+	os.Exit(code)
+}