db/schema/dialect/sqlserver/sqlserver_test.go

package sqlserver_test

import (
	"os"
	"testing"

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

	_ "github.com/denisenkom/go-mssqldb"
)

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

var (
	tableName = "entry"

	dropTable = "IF OBJECT_ID('entry', 'U') IS NOT NULL DROP TABLE entry;"

	createTable = "CREATE TABLE entry (" +
		"id INT NOT NULL IDENTITY(1,1) PRIMARY KEY," +
		"int INT NOT NULL UNIQUE," +
		"float REAL NOT NULL," +
		"string NVARCHAR(255) NOT NULL," +
		"time DATETIME2 NOT NULL," +
		"bool BIT NOT NULL," +
		"bytes VARBINARY(MAX) NOT NULL" +
		");"
)

func TestSQLServerSuite(t *testing.T) {
	suite.Run(t, new(sqlServerTest))
}

func (t *sqlServerTest) SetupSuite() {
	db := engine.Open(map[string]engine.DBConfig{
		"test-sqlserver": engine.NewDBConfig("sqlserver",
			engine.WithHost(os.Getenv("SQLSERVER_HOST")),
			engine.WithPort(os.Getenv("SQLSERVER_PORT")),
			engine.WithDatabase(os.Getenv("SQLSERVER_DB")),
			engine.WithUsername(os.Getenv("SQLSERVER_USER")),
			engine.WithPassword(os.Getenv("SQLSERVER_PASSWD")),
		),
	}).Connection("test-sqlserver")

	t.schema = schema.GetSchemaDialect(db)

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

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

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

func (t *sqlServerTest) TestCompileCreate() {
	bp := schema.NewBlueprint("users")
	bp.Create()

	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().Comment("更新时间")
	bp.DateTime("deleted_at").Nullable().Comment("删除时间")

	sql := t.schema.CompileCreate(bp)

	t.Equal([]string{
		"CREATE TABLE [users] (\n" +
			"[id] BIGINT NOT NULL IDENTITY(1,1) PRIMARY KEY COMMENT 'ID',\n" +
			"[enabled] BIT 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] DATETIME2 NOT NULL DEFAULT GETDATE() COMMENT '创建时间',\n" +
			"[updated_at] DATETIME2 NOT NULL DEFAULT GETDATE() COMMENT '更新时间',\n" +
			"[deleted_at] DATETIME2 NULL COMMENT '删除时间'\n" +
			")",
	}, sql)

	t.T().Log(sql)
}

func (t *sqlServerTest) TestCompileAdd() {
	bp := schema.NewBlueprint("users")

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

	sql := t.schema.CompileAdd(bp)

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

	t.T().Log(sql)
}

func (t *sqlServerTest) TestCompileChange() {
	bp := schema.NewBlueprint("users")

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

	sql := t.schema.CompileChange(bp)

	t.Equal([]string{
		"ALTER TABLE [users]\n" +
			"ALTER COLUMN [name] NVARCHAR(50) NOT NULL,\n" +
			"ALTER COLUMN [age] SMALLINT NOT NULL DEFAULT 19 COMMENT '年龄'",
	}, sql)

	t.T().Log(sql)
}

func (t *sqlServerTest) 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 *sqlServerTest) 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 *sqlServerTest) 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 *sqlServerTest) TestCompileRename() {
	bp := schema.NewBlueprint("users")

	bp.Rename("user")

	sql := t.schema.CompileRename(bp)

	t.Equal([]string{"EXEC sp_rename [users], [user]"}, sql)

	t.T().Log(sql)
}