mirror of
https://github.com/go-shiori/shiori.git
synced 2025-10-09 05:07:46 +08:00
31767f75e6
* feat: new migration system * use newFuncMigration * database version -> database schema version * column name * use path instead of filepath for goembed * simplified migrations, added backwards compatible migrations
61 lines
1.2 KiB
Go
61 lines
1.2 KiB
Go
//go:build !test_sqlite_only
|
|
// +build !test_sqlite_only
|
|
|
|
package database
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
)
|
|
|
|
func init() {
|
|
connString := os.Getenv("SHIORI_TEST_MYSQL_URL")
|
|
if connString == "" {
|
|
log.Fatal("mysql tests can't run without a MysQL database, set SHIORI_TEST_MYSQL_URL environment variable")
|
|
}
|
|
}
|
|
|
|
func mysqlTestDatabaseFactory(_ *testing.T, ctx context.Context) (DB, error) {
|
|
connString := os.Getenv("SHIORI_TEST_MYSQL_URL")
|
|
db, err := OpenMySQLDatabase(ctx, connString)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var dbname string
|
|
err = db.withTx(ctx, func(tx *sqlx.Tx) error {
|
|
err := tx.QueryRow("SELECT DATABASE()").Scan(&dbname)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = tx.ExecContext(ctx, "DROP DATABASE IF EXISTS "+dbname)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = tx.ExecContext(ctx, "CREATE DATABASE "+dbname)
|
|
return err
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if _, err := db.Exec("USE " + dbname); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err = db.Migrate(context.TODO()); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return db, err
|
|
}
|
|
|
|
func TestMysqlsDatabase(t *testing.T) {
|
|
testDatabase(t, mysqlTestDatabaseFactory)
|
|
}
|