HiPoweredDev
/
xorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix vet

This commit is contained in:
Lunny Xiao 2023-12-31 22:00:44 +08:00
parent 97bf025984
commit 43f38242f8
No known key found for this signature in database
GPG Key ID: C3B7C91B632F738A
1 changed files with 22 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
migrate/migrate_test.go
View File

@ -26,8 +26,7 @@ const (
dbName = "testdb.sqlite3" dbName = "testdb.sqlite3"
) )
var ( var migrations = []*Migration{
migrations = []*Migration{
{ {
ID: "201608301400", ID: "201608301400",
Migrate: func(tx *xorm.Engine) error { Migrate: func(tx *xorm.Engine) error {
@ -47,7 +46,6 @@ var (
}, },
}, },
} }
)
func TestMigration(t *testing.T) { func TestMigration(t *testing.T) {
_ = os.Remove(dbName) _ = os.Remove(dbName)
@ -58,7 +56,7 @@ func TestMigration(t *testing.T) {
} }
defer db.Close() defer db.Close()
if err = db.DB().Ping(); err != nil { if err = db.Ping(); err != nil {
log.Fatal(err) log.Fatal(err)
} }
@ -97,7 +95,7 @@ func TestInitSchema(t *testing.T) {
log.Fatal(err) log.Fatal(err)
} }
defer db.Close() defer db.Close()
if err = db.DB().Ping(); err != nil { if err = db.Ping(); err != nil {
log.Fatal(err) log.Fatal(err)
} }
@ -126,7 +124,7 @@ func TestMissingID(t *testing.T) {
if db != nil { if db != nil {
defer db.Close() defer db.Close()
} }
assert.NoError(t, db.DB().Ping()) assert.NoError(t, db.Ping())
migrationsMissingID := []*Migration{ migrationsMissingID := []*Migration{
{ {
@ -141,7 +139,6 @@ func TestMissingID(t *testing.T) {
} }
func tableCount(db *xorm.Engine, tableName string) (count int) { func tableCount(db *xorm.Engine, tableName string) (count int) {
row := db.DB().QueryRow(fmt.Sprintf("SELECT COUNT(*) FROM %s", tableName)) _, _ = db.SQL(fmt.Sprintf("SELECT COUNT(*) FROM %s", tableName)).Get(&count)
_ = row.Scan(&count)
return return
} }