优化数据库相关代码

This commit is contained in:
刘祥超
2022-04-08 14:15:45 +08:00
parent fe448e6556
commit c03d9f1880
10 changed files with 99 additions and 16 deletions

View File

@@ -12,6 +12,12 @@ func TestSQLDump_Dump(t *testing.T) {
Dsn: "root:123456@tcp(127.0.0.1:3306)/db_edge?charset=utf8mb4&timeout=30s",
Prefix: "edge",
})
if err != nil {
t.Fatal(err)
}
defer func() {
_ = db.Close()
}()
dump := NewSQLDump()
result, err := dump.Dump(db)
@@ -49,6 +55,12 @@ func TestSQLDump_Apply(t *testing.T) {
Dsn: "root:123456@tcp(127.0.0.1:3306)/db_edge?charset=utf8mb4&timeout=30s",
Prefix: "edge",
})
if err != nil {
t.Fatal(err)
}
defer func() {
_ = db.Close()
}()
dump := NewSQLDump()
result, err := dump.Dump(db)
@@ -64,6 +76,9 @@ func TestSQLDump_Apply(t *testing.T) {
if err != nil {
t.Fatal(err)
}
defer func() {
_ = db2.Close()
}()
ops, err := dump.Apply(db2, result, false)
if err != nil {
t.Fatal(err)