golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-05-04 08:30:57 +03:00
Code Issues Packages Projects Releases Wiki Activity

dialect/sql/schema: make Field.Unique and Index.Unique equal (#2004)

Browse Source 
This will allow moving from:

	field.String(c).Unique()

To:

	index.Field(c).Unique()

And vice versa
This commit is contained in:
Ariel Mashraki
2021-10-03 16:20:28 +03:00
committed by GitHub
parent 842fc9df1c
commit 14cba0abb6
17 changed files with 687 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
dialect/sql/schema/postgres_test.go
View File

@@ -617,6 +617,7 @@ func TestPostgres_Create(t *testing.T) {
c2 := []*Column{
{Name: "id", Type: field.TypeInt, Increment: true},
{Name: "score", Type: field.TypeInt},
{Name: "email", Type: field.TypeString},
}
return []*Table{
{
@@ -633,7 +634,9 @@ func TestPostgres_Create(t *testing.T) {
Columns: c2,
PrimaryKey: c2[0:1],
Indexes: Indexes{
{Name: "equipment_score", Columns: c2[1:]},
{Name: "equipment_score", Columns: c2[1:2]},
// Index should not be changed.
{Name: "equipment_email", Unique: true, Columns: c2[2:]},
},
},
}
@@ -667,11 +670,13 @@ func TestPostgres_Create(t *testing.T) {
WithArgs("equipment").
WillReturnRows(sqlmock.NewRows([]string{"column_name", "data_type", "is_nullable", "column_default", "udt_name", "numeric_precision", "numeric_scale"}).
AddRow("id", "bigint", "NO", "NULL", "int8", nil, nil).
AddRow("score", "bigint", "NO", "NULL", "int8", nil, nil))
AddRow("score", "bigint", "NO", "NULL", "int8", nil, nil).
AddRow("email", "character varying", "YES", "NULL", "varchar", nil, nil))
mock.ExpectQuery(escape(fmt.Sprintf(indexesQuery, "CURRENT_SCHEMA()", "equipment"))).
WillReturnRows(sqlmock.NewRows([]string{"index_name", "column_name", "primary", "unique", "seq_in_index"}).
AddRow("users_pkey", "id", "t", "t", 0).
AddRow("equipment_score", "score", "f", "f", 0))
AddRow("equipment_score", "score", "f", "f", 0).
AddRow("equipment_email", "email", "f", "t", 0))
mock.ExpectCommit()
},
},