// Copyright 2023 PingCAP, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// See the License for the specific language governing permissions and
// limitations under the License.

package simple

import (
	"testing"

	commonEvent "github.com/pingcap/ticdc/pkg/common/event"
	"github.com/stretchr/testify/require"
)

func TestNewTableSchema(t *testing.T) {
	helper := commonEvent.NewEventTestHelper(t)
	defer helper.Close()

	// case 1: test for primary key is not explicitly constraint
	sql := `create table test.t1(
		id int primary key,
		name varchar(64) not null,
		age int,
		email varchar(255) not null,
		unique index idx_name(name),
		index idx_age_email(age,email)
	);`
	job := helper.DDL2Job(sql)
	tableInfo := helper.GetTableInfo(job)
	want := &TableSchema{
		Schema:  tableInfo.TableName.Schema,
		Table:   tableInfo.TableName.Table,
		TableID: tableInfo.TableName.TableID,
		Version: tableInfo.GetUpdateTS(),
		Columns: []*columnSchema{
			{
				Name: "id",
				DataType: dataType{
					MySQLType: "int",
					Charset:   "binary",
					Collate:   "binary",
					Length:    11,
				},
				Nullable: false,
			},
			{
				Name: "name",
				DataType: dataType{
					MySQLType: "varchar",
					Charset:   "utf8mb4",
					Collate:   "utf8mb4_bin",
					Length:    64,
				},
				Nullable: false,
			},
			{
				Name: "age",
				DataType: dataType{
					MySQLType: "int",
					Charset:   "binary",
					Collate:   "binary",
					Length:    11,
				},
				Nullable: true,
			},
			{
				Name: "email",
				DataType: dataType{
					MySQLType: "varchar",
					Charset:   "utf8mb4",
					Collate:   "utf8mb4_bin",
					Length:    255,
				},
				Nullable: false,
			},
		},
		Indexes: []*IndexSchema{
			{
				Name:     "idx_name",
				Unique:   true,
				Primary:  false,
				Nullable: false,
				Columns:  []string{"name"},
			},
			{
				Name:     "idx_age_email",
				Unique:   false,
				Primary:  false,
				Nullable: true,
				Columns:  []string{"age", "email"},
			},
			{
				Name:     "primary",
				Unique:   true,
				Primary:  true,
				Nullable: false,
				Columns:  []string{"id"},
			},
		},
	}
	got := newTableSchema(tableInfo)
	require.Equal(t, want, got)

	// case 2: test for primary key is explicitly constraint
	sql = `create table test.t2(
		id int,
		name varchar(64) not null,
		age int,
		email varchar(255) not null,
		primary key(id),
		unique index idx_name(name),
		index idx_age_email(age,email)
	);`
	job = helper.DDL2Job(sql)
	tableInfo = helper.GetTableInfo(job)
	want = &TableSchema{
		Schema:  tableInfo.TableName.Schema,
		Table:   tableInfo.TableName.Table,
		TableID: tableInfo.TableName.TableID,
		Version: tableInfo.GetUpdateTS(),
		Columns: []*columnSchema{
			{
				Name: "id",
				DataType: dataType{
					MySQLType: "int",
					Charset:   "binary",
					Collate:   "binary",
					Length:    11,
				},
				Nullable: false,
			},
			{
				Name: "name",
				DataType: dataType{
					MySQLType: "varchar",
					Charset:   "utf8mb4",
					Collate:   "utf8mb4_bin",
					Length:    64,
				},
				Nullable: false,
			},
			{
				Name: "age",
				DataType: dataType{
					MySQLType: "int",
					Charset:   "binary",
					Collate:   "binary",
					Length:    11,
				},
				Nullable: true,
			},
			{
				Name: "email",
				DataType: dataType{
					MySQLType: "varchar",
					Charset:   "utf8mb4",
					Collate:   "utf8mb4_bin",
					Length:    255,
				},
				Nullable: false,
			},
		},
		Indexes: []*IndexSchema{
			{
				Name:     "idx_name",
				Unique:   true,
				Primary:  false,
				Nullable: false,
				Columns:  []string{"name"},
			},
			{
				Name:     "idx_age_email",
				Unique:   false,
				Primary:  false,
				Nullable: true,
				Columns:  []string{"age", "email"},
			},
			{
				Name:     "primary",
				Unique:   true,
				Primary:  true,
				Nullable: false,
				Columns:  []string{"id"},
			},
		},
	}
	got = newTableSchema(tableInfo)
	require.Equal(t, want, got)

	// case 3: test for all data types in TiDB
	sql = `create table test.t3(
		t tinyint primary key,
		tu1 tinyint unsigned,
		s smallint,
		su1 smallint unsigned,
		m mediumint,
		mu1 mediumint unsigned,
		i int default 100,
		iu1 int unsigned,
		bi bigint,
		biu1 bigint unsigned,
		floatT float,
		doubleT double,
		decimalT decimal,
		floatTu float unsigned,
		doubleTu double unsigned,
		decimalTu decimal unsigned,
		varcharT varchar(255),
		charT char(255),
		binaryT binary(255),
		 varbinaryT varbinary(255),
		 tinytextT tinytext,
		 textT text,
		 mediumtextT mediumtext,
		 longtextT longtext,
		 tinyblobT tinyblob,
		 blobT blob,
		 mediumblobT mediumblob,
		 longblobT longblob,
		 dateT date,
		 datetimeT datetime,
		 timestampT timestamp,
		 timeT time,
		 yearT year,
		 enumT enum('a', 'b', 'c') default 'b',
		 setT set('a', 'b', 'c'),
		 bitT bit(10),
		 jsonT json,
		 tgen tinyint AS (t+1))` // 38
	job = helper.DDL2Job(sql)
	tableInfo = helper.GetTableInfo(job)
	want = &TableSchema{
		Schema:  tableInfo.TableName.Schema,
		Table:   tableInfo.TableName.Table,
		TableID: tableInfo.TableName.TableID,
		Version: tableInfo.GetUpdateTS(),
		Columns: []*columnSchema{
			{
				Name: "t",
				DataType: dataType{
					MySQLType: "tinyint",
					Charset:   "binary",
					Collate:   "binary",
					Length:    4,
				},
				Nullable: false,
			},
			{
				Name: "tu1",
				DataType: dataType{
					MySQLType: "tinyint",
					Charset:   "binary",
					Collate:   "binary",
					Length:    3,
					Unsigned:  true,
				},
				Nullable: true,
			},
			{
				Name: "s",
				DataType: dataType{
					MySQLType: "smallint",
					Charset:   "binary",
					Collate:   "binary",
					Length:    6,
				},
				Nullable: true,
			},
			{
				Name: "su1",
				DataType: dataType{
					MySQLType: "smallint",
					Charset:   "binary",
					Collate:   "binary",
					Length:    5,
					Unsigned:  true,
				},
				Nullable: true,
			},
			{
				Name: "m",
				DataType: dataType{
					MySQLType: "mediumint",
					Charset:   "binary",
					Collate:   "binary",
					Length:    9,
				},
				Nullable: true,
			},
			{
				Name: "mu1",
				DataType: dataType{
					MySQLType: "mediumint",
					Charset:   "binary",
					Collate:   "binary",
					Length:    8,
					Unsigned:  true,
				},
				Nullable: true,
			},
			{
				Name: "i",
				DataType: dataType{
					MySQLType: "int",
					Charset:   "binary",
					Collate:   "binary",
					Length:    11,
				},
				Nullable: true,
				Default:  "100",
			},
			{
				Name: "iu1",
				DataType: dataType{
					MySQLType: "int",
					Charset:   "binary",
					Collate:   "binary",
					Length:    10,
					Unsigned:  true,
				},
				Nullable: true,
			},
			{
				Name: "bi",
				DataType: dataType{
					MySQLType: "bigint",
					Charset:   "binary",
					Collate:   "binary",
					Length:    20,
				},
				Nullable: true,
			},
			{
				Name: "biu1",
				DataType: dataType{
					MySQLType: "bigint",
					Charset:   "binary",
					Collate:   "binary",
					Length:    20,
					Unsigned:  true,
				},
				Nullable: true,
			},
			{
				Name: "floatT",
				DataType: dataType{
					MySQLType: "float",
					Charset:   "binary",
					Collate:   "binary",
					Length:    12,
				},
				Nullable: true,
			},
			{
				Name: "doubleT",
				DataType: dataType{
					MySQLType: "double",
					Charset:   "binary",
					Collate:   "binary",
					Length:    22,
				},
				Nullable: true,
			},
			{
				Name: "decimalT",
				DataType: dataType{
					MySQLType: "decimal",
					Charset:   "binary",
					Collate:   "binary",
					Length:    10,
				},
				Nullable: true,
			},
			{
				Name: "floatTu",
				DataType: dataType{
					MySQLType: "float",
					Charset:   "binary",
					Collate:   "binary",
					Length:    12,
					Unsigned:  true,
				},
				Nullable: true,
			},
			{
				Name: "doubleTu",
				DataType: dataType{
					MySQLType: "double",
					Charset:   "binary",
					Collate:   "binary",
					Length:    22,
					Unsigned:  true,
				},
				Nullable: true,
			},
			{
				Name: "decimalTu",
				DataType: dataType{
					MySQLType: "decimal",
					Charset:   "binary",
					Collate:   "binary",
					Length:    10,
					Unsigned:  true,
				},
				Nullable: true,
			},
			{
				Name: "varcharT",
				DataType: dataType{
					MySQLType: "varchar",
					Charset:   "utf8mb4",
					Collate:   "utf8mb4_bin",
					Length:    255,
				},
				Nullable: true,
			},
			{
				Name: "charT",
				DataType: dataType{
					MySQLType: "char",
					Charset:   "utf8mb4",
					Collate:   "utf8mb4_bin",
					Length:    255,
				},
				Nullable: true,
			},
			{
				Name: "binaryT",
				DataType: dataType{
					MySQLType: "binary",
					Charset:   "binary",
					Collate:   "binary",
					Length:    255,
				},
				Nullable: true,
			},
			{
				Name: "varbinaryT",
				DataType: dataType{
					MySQLType: "varbinary",
					Charset:   "binary",
					Collate:   "binary",
					Length:    255,
				},
				Nullable: true,
			},
			{
				Name: "tinytextT",
				DataType: dataType{
					MySQLType: "tinytext",
					Charset:   "utf8mb4",
					Collate:   "utf8mb4_bin",
					Length:    255,
				},
				Nullable: true,
			},
			{
				Name: "textT",
				DataType: dataType{
					MySQLType: "text",
					Charset:   "utf8mb4",
					Collate:   "utf8mb4_bin",
					Length:    65535,
				},
				Nullable: true,
			},
			{
				Name: "mediumtextT",
				DataType: dataType{
					MySQLType: "mediumtext",
					Charset:   "utf8mb4",
					Collate:   "utf8mb4_bin",
					Length:    16777215,
				},
				Nullable: true,
			},
			{
				Name: "longtextT",
				DataType: dataType{
					MySQLType: "longtext",
					Charset:   "utf8mb4",
					Collate:   "utf8mb4_bin",
					Length:    4294967295,
				},
				Nullable: true,
			},
			{
				Name: "tinyblobT",
				DataType: dataType{
					MySQLType: "tinyblob",
					Charset:   "binary",
					Collate:   "binary",
					Length:    255,
				},
				Nullable: true,
			},
			{
				Name: "blobT",
				DataType: dataType{
					MySQLType: "blob",
					Charset:   "binary",
					Collate:   "binary",
					Length:    65535,
				},
				Nullable: true,
			},
			{
				Name: "mediumblobT",
				DataType: dataType{
					MySQLType: "mediumblob",
					Charset:   "binary",
					Collate:   "binary",
					Length:    16777215,
				},
				Nullable: true,
			},
			{
				Name: "longblobT",
				DataType: dataType{
					MySQLType: "longblob",
					Charset:   "binary",
					Collate:   "binary",
					Length:    4294967295,
				},
				Nullable: true,
			},
			{
				Name: "dateT",
				DataType: dataType{
					MySQLType: "date",
					Charset:   "binary",
					Collate:   "binary",
					Length:    10,
				},
				Nullable: true,
			},
			{
				Name: "datetimeT",
				DataType: dataType{
					MySQLType: "datetime",
					Charset:   "binary",
					Collate:   "binary",
					Length:    19,
				},
				Nullable: true,
			},
			{
				Name: "timestampT",
				DataType: dataType{
					MySQLType: "timestamp",
					Charset:   "binary",
					Collate:   "binary",
					Length:    19,
				},
				Nullable: true,
			},
			{
				Name: "timeT",
				DataType: dataType{
					MySQLType: "time",
					Charset:   "binary",
					Collate:   "binary",
					Length:    10,
				},
				Nullable: true,
			},
			{
				Name: "yearT",
				DataType: dataType{
					MySQLType: "year",
					Charset:   "binary",
					Collate:   "binary",
					Length:    4,
					Unsigned:  true,
					Zerofill:  true,
				},
				Nullable: true,
			},
			{
				Name: "enumT",
				DataType: dataType{
					MySQLType: "enum",
					Charset:   "utf8mb4",
					Collate:   "utf8mb4_bin",
					Length:    1,
					Elements:  []string{"a", "b", "c"},
				},
				Nullable: true,
				Default:  "b",
			},
			{
				Name: "setT",
				DataType: dataType{
					MySQLType: "set",
					Charset:   "utf8mb4",
					Collate:   "utf8mb4_bin",
					Length:    5,
					Elements:  []string{"a", "b", "c"},
				},
				Nullable: true,
			},
			{
				Name: "bitT",
				DataType: dataType{
					MySQLType: "bit",
					Charset:   "binary",
					Collate:   "binary",
					Length:    10,
					Unsigned:  true,
				},
				Nullable: true,
			},
			{
				Name: "jsonT",
				DataType: dataType{
					MySQLType: "json",
					Charset:   "binary",
					Collate:   "binary",
					Length:    4294967295,
				},
				Nullable: true,
			},
			{
				Name: "tgen",
				DataType: dataType{
					MySQLType: "tinyint",
					Charset:   "binary",
					Collate:   "binary",
					Length:    4,
				},
				Nullable: true,
			},
		},
		Indexes: []*IndexSchema{
			{
				Name:     "primary",
				Unique:   true,
				Primary:  true,
				Nullable: false,
				Columns:  []string{"t"},
			},
		},
	}
	got = newTableSchema(tableInfo)
	require.Equal(t, want, got)
}