package postgres_test

import (
	"testing"

	"git.fsdpf.net/go/db/engine"
	"git.fsdpf.net/go/db/schema"
	sqlmock "github.com/DATA-DOG/go-sqlmock"
	"github.com/stretchr/testify/suite"
)

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

var tableName = "entry"

func TestPostgresSuite(t *testing.T) {
	suite.Run(t, new(postgresTest))
}

func (t *postgresTest) SetupSuite() {
	mockDB, mock, _ := sqlmock.New()

	mock.ExpectQuery("SELECT.*column_name.*columns").
		WillReturnRows(sqlmock.NewRows([]string{"column_name"}).
			AddRow("id").AddRow("int").AddRow("float").
			AddRow("string").AddRow("time").AddRow("bool").AddRow("bytes"))

	mock.ExpectQuery("SELECT.*COUNT.*tables").
		WillReturnRows(sqlmock.NewRows([]string{"count"}).AddRow(1))

	conn := engine.Mock(map[string]engine.MockDBConfig{
		"test-postgres": {Driver: "postgres", Mock: mockDB},
	}).Connection("test-postgres")
	t.schema = schema.GetSchemaDialect(conn)
}

func (t *postgresTest) 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 *postgresTest) TestTableExists() {
	result, err := t.schema.TableExists(tableName)
	t.Require().NoError(err)
	t.Require().Equal(result, true)
}

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

	bp.BigIncrements("id").AutoIncrement().Comment("ID")
	bp.Boolean("enabled").Default("true").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 IF NOT EXISTS \"users\" (\n" +
			"\"id\" bigint NOT NULL BIGSERIAL PRIMARY KEY,\n" +
			"\"enabled\" boolean NOT NULL DEFAULT 'true',\n" +
			"\"created_user\" char(36) NOT NULL DEFAULT '00000000-0000-0000-0000-000000000000',\n" +
			"\"owned_user\" char(36) NOT NULL DEFAULT '00000000-0000-0000-0000-000000000000',\n" +
			"\"created_at\" timestamp with time zone NOT NULL DEFAULT CURRENT_TIMESTAMP,\n" +
			"\"updated_at\" timestamp with time zone NOT NULL DEFAULT CURRENT_TIMESTAMP,\n" +
			"\"deleted_at\" timestamp without time zone NULL\n" +
			")",
	}, sql)

	t.T().Log(sql)
}

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

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

	sql := t.schema.CompileAdd(bp)

	t.Equal([]string{
		"ALTER TABLE \"users\"\n" +
			"ADD COLUMN \"name\" varchar(50) NOT NULL,\n" +
			"ADD COLUMN \"age\" smallint NOT NULL DEFAULT '18',\n" +
			"ADD COLUMN \"sex\" varchar(1) NOT NULL DEFAULT '0'",
	}, sql)

	t.T().Log(sql)
}

func (t *postgresTest) 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\" TYPE varchar(50) NOT NULL,\n" +
			"ALTER COLUMN \"age\" TYPE smallint NOT NULL DEFAULT '19'",
	}, sql)

	t.T().Log(sql)
}

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

	bp.Rename("user")

	sql := t.schema.CompileRename(bp)

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

	t.T().Log(sql)
}