db/schema/dialect/mysql/mysql_test.go

package mysql_test

import (
	"os"
	"testing"

	"git.fsdpf.net/go/db/v2"
	"git.fsdpf.net/go/db/v2/engine"
	"git.fsdpf.net/go/db/v2/schema"
	"github.com/stretchr/testify/suite"

	_ "github.com/go-sql-driver/mysql"
)

type mysqlTest struct {
	suite.Suite
	schema schema.Schema
}

var (
	tableName = "entry"

	dropTable = "DROP TABLE IF EXISTS `entry`;"

	createTable = "CREATE TABLE IF NOT EXISTS `entry` (" +
		"`id` INT NOT NULL AUTO_INCREMENT ," +
		"`int` INT NOT NULL UNIQUE," +
		"`float` FLOAT NOT NULL ," +
		"`string` VARCHAR(255) NOT NULL ," +
		"`time` DATETIME NOT NULL ," +
		"`bool` TINYINT NOT NULL ," +
		"`bytes` BLOB NOT NULL ," +
		"PRIMARY KEY (`id`) );"
)

func TestMysqlSuite(t *testing.T) {
	suite.Run(t, new(mysqlTest))
}

func (t *mysqlTest) SetupSuite() {
	db := engine.Open(map[string]engine.DBConfig{
		"test-mysql": engine.NewDBConfig("mysql",
			engine.WithHost(os.Getenv("MYSQL_HOST")),
			engine.WithPort(os.Getenv("MYSQL_PORT")),
			engine.WithDatabase(os.Getenv("MYSQL_DB")),
			engine.WithUsername(os.Getenv("MYSQL_USER")),
			engine.WithPassword(os.Getenv("MYSQL_PASSWD")),
			engine.WithParseTime(true),
		),
	}).Connection("test-mysql")

	t.schema = schema.GetSchemaDialect(db)

	// db.Exec(dropTable)
	db.Exec(createTable)
}

func (t *mysqlTest) TestGetColumnListing() {
	result, err := t.schema.GetColumnListing(tableName)
	t.Require().NoError(err)
	t.Require().ElementsMatch([]string{"bool", "bytes", "float", "id", "int", "string", "time"}, result)
}

func (t *mysqlTest) TestTableExists() {
	result, err := t.schema.TableExists(tableName)
	t.Require().NoError(err)
	t.Require().Equal(result, true)
}

// 创建列表
func (t *mysqlTest) TestCompileCreate() {
	bp := schema.NewBlueprint("users")
	bp.Create()
	bp.Charset("utf8mb4")

	bp.BigIncrements("id").AutoIncrement().Comment("ID")
	bp.Boolean("enabled").Default("1").Comment("是否有效")
	bp.Char("created_user", 36).Default("00000000-0000-0000-0000-000000000000").Comment("创建者")
	bp.Char("owned_user", 36).Default("00000000-0000-0000-0000-000000000000").Comment("拥有者")
	bp.Timestamp("created_at").UseCurrent().Comment("创建时间")
	bp.Timestamp("updated_at").UseCurrent().Default(db.L("ON UPDATE CURRENT_TIMESTAMP")).Comment("更新时间")
	bp.DateTime("deleted_at").Nullable().Comment("删除时间")

	sql := t.schema.CompileCreate(bp)

	t.Equal([]string{
		"CREATE TABLE `users` (\n" +
			"`id` bigint(20) UNSIGNED NOT NULL AUTO_INCREMENT PRIMARY KEY COMMENT 'ID',\n" +
			"`enabled` tinyint(1) NOT NULL DEFAULT '1' COMMENT '是否有效',\n" +
			"`created_user` char(36) NOT NULL DEFAULT '00000000-0000-0000-0000-000000000000' COMMENT '创建者',\n" +
			"`owned_user` char(36) NOT NULL DEFAULT '00000000-0000-0000-0000-000000000000' COMMENT '拥有者',\n" +
			"`created_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',\n" +
			"`updated_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间',\n" +
			"`deleted_at` datetime NULL COMMENT '删除时间'\n" +
			") default charset='utf8mb4' collate='utf8mb4_general_ci'",
	}, sql)

	t.T().Log(sql)
}

// 添加字段
func (t *mysqlTest) TestCompileAdd() {
	bp := schema.NewBlueprint("users")

	bp.String("name", 50).Comment("用户名")
	bp.SmallInteger("age").Default("18").Comment("年龄")
	bp.Enum("sex", []string{"0", "1"}).Default("0").Comment("性别")

	sql := t.schema.CompileAdd(bp)

	t.Equal([]string{
		"ALTER TABLE `users`\n" +
			"ADD COLUMN `name` varchar(50) NOT NULL COMMENT '用户名',\n" +
			"ADD COLUMN `age` smallint(4) NOT NULL DEFAULT '18' COMMENT '年龄',\n" +
			"ADD COLUMN `sex` enum('0', '1') NOT NULL DEFAULT '0' COMMENT '性别'",
	}, sql)

	t.T().Log(sql)
}

// 修改字段
func (t *mysqlTest) TestCompileChange() {
	bp := schema.NewBlueprint("users")

	bp.String("name", 50).Change("username")
	bp.SmallInteger("age").Default("19").Comment("年龄").Change()

	sql := "ALTER TABLE `users`\n" +
		"CHANGE COLUMN `name` `username` varchar(50) NOT NULL,\n" +
		"CHANGE COLUMN `age` `age` smallint(4) NOT NULL DEFAULT '19' COMMENT '年龄';"

	t.Equal(t.schema.CompileChange(bp), sql)

	t.T().Log(sql)
}

// 删除字段
func (t *mysqlTest) TestCompileDropColumn() {
	bp := schema.NewBlueprint("users")

	bp.DropColumn("age", "sex")

	sql := t.schema.CompileDropColumn(bp)

	t.Equal([]string{
		"ALTER TABLE `users`\n" +
			"DROP COLUMN `age`,\n" +
			"DROP COLUMN `sex`",
	}, sql)

	t.T().Log(sql)
}

// 删除表
func (t *mysqlTest) TestCompileDrop() {
	bp := schema.NewBlueprint("users")

	bp.Drop()

	sql := t.schema.CompileDrop(bp)

	t.Equal([]string{"DROP TABLE `users`"}, sql)

	t.T().Log(sql)
}

// 删除表
func (t *mysqlTest) TestCompileDropIfExists() {
	bp := schema.NewBlueprint("users")

	bp.Drop()

	sql := t.schema.CompileDropIfExists(bp)
	t.Equal([]string{"DROP TABLE IF EXISTS `users`"}, sql)

	t.T().Log(sql)
}

// 表重命名
func (t *mysqlTest) TestCompileRename() {
	bp := schema.NewBlueprint("users")

	bp.Rename("user")

	sql := t.schema.CompileRename(bp)

	t.Equal([]string{"RENAME TABLE `users` TO `user`"}, sql)

	t.T().Log(sql)
}