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: add postgres tests for array type

Browse Source
This commit is contained in:
Ariel Mashraki
2021-04-14 14:11:44 +03:00
committed by Ariel Mashraki
parent fd1f2df150
commit 5bcee6b51c
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -64,13 +64,14 @@ func TestPostgres_Create(t *testing.T) {
{Name: "enums", Type: field.TypeEnum, Enums: []string{"a", "b"}, Default: "a"},
{Name: "uuid", Type: field.TypeUUID, Default: "uuid_generate_v4()"},
{Name: "price", Type: field.TypeFloat64, SchemaType: map[string]string{dialect.Postgres: "numeric(5,2)"}},
{Name: "strings", Type: field.TypeOther, SchemaType: map[string]string{dialect.Postgres: "text[]"}, Nullable: true},
},
},
},
before: func(mock pgMock) {
mock.start("120000")
mock.tableExists("users", false)
mock.ExpectExec(escape(`CREATE TABLE IF NOT EXISTS "users"("id" bigint GENERATED BY DEFAULT AS IDENTITY NOT NULL, "name" varchar NULL, "age" bigint NOT NULL, "doc" jsonb NULL, "enums" varchar NOT NULL DEFAULT 'a', "uuid" uuid NOT NULL DEFAULT uuid_generate_v4(), "price" numeric(5,2) NOT NULL, PRIMARY KEY("id"))`)).
mock.ExpectExec(escape(`CREATE TABLE IF NOT EXISTS "users"("id" bigint GENERATED BY DEFAULT AS IDENTITY NOT NULL, "name" varchar NULL, "age" bigint NOT NULL, "doc" jsonb NULL, "enums" varchar NOT NULL DEFAULT 'a', "uuid" uuid NOT NULL DEFAULT uuid_generate_v4(), "price" numeric(5,2) NOT NULL, "strings" text[] NULL, PRIMARY KEY("id"))`)).
WillReturnResult(sqlmock.NewResult(0, 1))
mock.ExpectCommit()
},
@@ -248,6 +249,7 @@ func TestPostgres_Create(t *testing.T) {
{Name: "inet", Type: field.TypeString, Nullable: true, SchemaType: map[string]string{dialect.Postgres: "inet"}},
{Name: "macaddr", Type: field.TypeString, Nullable: true, SchemaType: map[string]string{dialect.Postgres: "macaddr"}},
{Name: "macaddr8", Type: field.TypeString, Nullable: true, SchemaType: map[string]string{dialect.Postgres: "macaddr8"}},
{Name: "strings", Type: field.TypeOther, SchemaType: map[string]string{dialect.Postgres: "text[]"}, Nullable: true},
},
PrimaryKey: []*Column{
{Name: "id", Type: field.TypeInt, Increment: true},
@@ -270,7 +272,8 @@ func TestPostgres_Create(t *testing.T) {
AddRow("cidr", "cidr", "NO", "NULL", "cidr").
AddRow("inet", "inet", "YES", "NULL", "inet").
AddRow("macaddr", "macaddr", "YES", "NULL", "macaddr").
AddRow("macaddr8", "macaddr8", "YES", "NULL", "macaddr8"))
AddRow("macaddr8", "macaddr8", "YES", "NULL", "macaddr8").
AddRow("strings", "ARRAY", "YES", "NULL", "_text"))
mock.ExpectQuery(escape(fmt.Sprintf(indexesQuery, "CURRENT_SCHEMA()", "users"))).
WillReturnRows(sqlmock.NewRows([]string{"index_name", "column_name", "primary", "unique", "seq_in_index"}).
AddRow("users_pkey", "id", "t", "t", 0))