func TestPrimaryKeyUnspecified(t *testing.T) {
defer leaktest.AfterTest(t)()
stmt, err := parser.ParseOneTraditional(
"CREATE TABLE foo.test (a INT, b INT, CONSTRAINT c UNIQUE (b))")
if err != nil {
t.Fatal(err)
}
create := stmt.(*parser.CreateTable)
if err := create.Table.NormalizeTableName(""); err != nil {
t.Fatal(err)
}
desc, err := sqlbase.MakeTableDesc(create, 1)
if err != nil {
t.Fatal(err)
}
err = desc.AllocateIDs()
if !testutils.IsError(err, sqlbase.ErrMissingPrimaryKey.Error()) {
t.Fatalf("unexpected error: %s", err)
}
}
func TestMakeTableDescColumns(t *testing.T) {
defer leaktest.AfterTest(t)()
testData := []struct {
sqlType string
colType sqlbase.ColumnType
nullable bool
}{
{
"BIT(1)",
sqlbase.ColumnType{Kind: sqlbase.ColumnType_INT, Width: 1},
true,
},
{
"BOOLEAN",
sqlbase.ColumnType{Kind: sqlbase.ColumnType_BOOL},
true,
},
{
"INT",
sqlbase.ColumnType{Kind: sqlbase.ColumnType_INT},
true,
},
{
"FLOAT(3)",
sqlbase.ColumnType{Kind: sqlbase.ColumnType_FLOAT, Precision: 3},
true,
},
{
"DECIMAL(6,5)",
sqlbase.ColumnType{Kind: sqlbase.ColumnType_DECIMAL, Precision: 6, Width: 5},
true,
},
{
"DATE",
sqlbase.ColumnType{Kind: sqlbase.ColumnType_DATE},
true,
},
{
"TIMESTAMP",
sqlbase.ColumnType{Kind: sqlbase.ColumnType_TIMESTAMP},
true,
},
{
"INTERVAL",
sqlbase.ColumnType{Kind: sqlbase.ColumnType_INTERVAL},
true,
},
{
"CHAR",
sqlbase.ColumnType{Kind: sqlbase.ColumnType_STRING},
true,
},
{
"TEXT",
sqlbase.ColumnType{Kind: sqlbase.ColumnType_STRING},
true,
},
{
"BLOB",
sqlbase.ColumnType{Kind: sqlbase.ColumnType_BYTES},
true,
},
{
"INT NOT NULL",
sqlbase.ColumnType{Kind: sqlbase.ColumnType_INT},
false,
},
{
"INT NULL",
sqlbase.ColumnType{Kind: sqlbase.ColumnType_INT},
true,
},
}
for i, d := range testData {
stmt, err := parser.ParseOneTraditional(
"CREATE TABLE foo.test (a " + d.sqlType + " PRIMARY KEY, b " + d.sqlType + ")")
if err != nil {
t.Fatalf("%d: %v", i, err)
}
create := stmt.(*parser.CreateTable)
if err := create.Table.NormalizeTableName(""); err != nil {
t.Fatalf("%d: %v", i, err)
}
schema, err := sqlbase.MakeTableDesc(create, 1)
if err != nil {
t.Fatalf("%d: %v", i, err)
}
if !reflect.DeepEqual(d.colType, schema.Columns[0].Type) {
t.Fatalf("%d: expected %+v, but got %+v", i, d.colType, schema.Columns[0])
}
if schema.Columns[0].Nullable {
t.Fatalf("%d: expected non-nullable primary key, but got %+v", i, schema.Columns[0].Nullable)
}
if !reflect.DeepEqual(d.colType, schema.Columns[1].Type) {
t.Fatalf("%d: expected %+v, but got %+v", i, d.colType, schema.Columns[1])
}
if d.nullable != schema.Columns[1].Nullable {
t.Fatalf("%d: expected %+v, but got %+v", i, d.nullable, schema.Columns[1].Nullable)
}
}
}
func TestMakeTableDescIndexes(t *testing.T) {
defer leaktest.AfterTest(t)()
testData := []struct {
sql string
primary sqlbase.IndexDescriptor
indexes []sqlbase.IndexDescriptor
}{
{
"a INT PRIMARY KEY",
sqlbase.IndexDescriptor{
Name: sqlbase.PrimaryKeyIndexName,
Unique: true,
ColumnNames: []string{"a"},
ColumnDirections: []sqlbase.IndexDescriptor_Direction{sqlbase.IndexDescriptor_ASC},
},
[]sqlbase.IndexDescriptor{},
},
{
"a INT UNIQUE, b INT PRIMARY KEY",
sqlbase.IndexDescriptor{
Name: "primary",
Unique: true,
ColumnNames: []string{"b"},
ColumnDirections: []sqlbase.IndexDescriptor_Direction{sqlbase.IndexDescriptor_ASC},
},
[]sqlbase.IndexDescriptor{
{
Name: "",
Unique: true,
ColumnNames: []string{"a"},
ColumnDirections: []sqlbase.IndexDescriptor_Direction{sqlbase.IndexDescriptor_ASC},
},
},
},
{
"a INT, b INT, CONSTRAINT c PRIMARY KEY (a, b)",
sqlbase.IndexDescriptor{
Name: "c",
Unique: true,
ColumnNames: []string{"a", "b"},
ColumnDirections: []sqlbase.IndexDescriptor_Direction{sqlbase.IndexDescriptor_ASC, sqlbase.IndexDescriptor_ASC},
},
[]sqlbase.IndexDescriptor{},
},
{
"a INT, b INT, CONSTRAINT c UNIQUE (b), PRIMARY KEY (a, b)",
sqlbase.IndexDescriptor{
Name: "primary",
Unique: true,
ColumnNames: []string{"a", "b"},
ColumnDirections: []sqlbase.IndexDescriptor_Direction{sqlbase.IndexDescriptor_ASC, sqlbase.IndexDescriptor_ASC},
},
[]sqlbase.IndexDescriptor{
{
Name: "c",
Unique: true,
ColumnNames: []string{"b"},
ColumnDirections: []sqlbase.IndexDescriptor_Direction{sqlbase.IndexDescriptor_ASC},
},
},
},
{
"a INT, b INT, PRIMARY KEY (a, b)",
sqlbase.IndexDescriptor{
Name: sqlbase.PrimaryKeyIndexName,
Unique: true,
ColumnNames: []string{"a", "b"},
ColumnDirections: []sqlbase.IndexDescriptor_Direction{sqlbase.IndexDescriptor_ASC, sqlbase.IndexDescriptor_ASC},
},
[]sqlbase.IndexDescriptor{},
},
}
for i, d := range testData {
stmt, err := parser.ParseOneTraditional("CREATE TABLE foo.test (" + d.sql + ")")
if err != nil {
t.Fatalf("%d: %v", i, err)
}
create := stmt.(*parser.CreateTable)
if err := create.Table.NormalizeTableName(""); err != nil {
t.Fatalf("%d: %v", i, err)
}
schema, err := sqlbase.MakeTableDesc(create, 1)
if err != nil {
t.Fatalf("%d: %v", i, err)
}
if !reflect.DeepEqual(d.primary, schema.PrimaryIndex) {
t.Fatalf("%d: expected %+v, but got %+v", i, d.primary, schema.PrimaryIndex)
}
if !reflect.DeepEqual(d.indexes, append([]sqlbase.IndexDescriptor{}, schema.Indexes...)) {
t.Fatalf("%d: expected %+v, but got %+v", i, d.indexes, schema.Indexes)
}
}
}