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

entc/gen: allow setting required fields as nillable

Browse Source
This commit is contained in:
Ariel Mashraki
2022-08-13 18:12:53 +03:00
committed by Ariel Mashraki
parent c8fc927f0d
commit b207429f15
17 changed files with 441 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
entc/integration/integration_test.go
View File

@@ -38,8 +38,8 @@ import (
"entgo.io/ent/entc/integration/ent/node"
"entgo.io/ent/entc/integration/ent/pet"
"entgo.io/ent/entc/integration/ent/schema"
enttask "entgo.io/ent/entc/integration/ent/task"
"entgo.io/ent/entc/integration/ent/user"
"entgo.io/ent/entc/integration/privacy/ent/task"
"github.com/go-sql-driver/mysql"
"github.com/lib/pq"
@@ -156,6 +156,7 @@ var (
Mutation,
CreateBulk,
ConstraintChecks,
NillableRequired,
}
)
@@ -753,7 +754,7 @@ func ExecQuery(t *testing.T, client *ent.Client) {
require.NoError(err)
tx.Task.Create().ExecX(ctx)
require.Equal(1, tx.Task.Query().CountX(ctx))
rows, err = tx.QueryContext(ctx, "SELECT COUNT(*) FROM "+task.Table)
rows, err = tx.QueryContext(ctx, "SELECT COUNT(*) FROM "+enttask.Table)
require.NoError(err)
count, err := sql.ScanInt(rows)
require.NoError(err)
@@ -762,6 +763,17 @@ func ExecQuery(t *testing.T, client *ent.Client) {
require.NoError(tx.Commit())
}
func NillableRequired(t *testing.T, client *ent.Client) {
require := require.New(t)
ctx := context.Background()
client.Task.Create().ExecX(ctx)
tk := client.Task.Query().OnlyX(ctx)
require.NotNil(tk.CreatedAt, "field value should be populated by default by the database")
require.False(reflect.ValueOf(tk.Update()).MethodByName("SetNillableCreatedAt").IsValid(), "immutable-nillable should not have SetNillable setter on update")
tk = client.Task.Query().Select(enttask.FieldID, enttask.FieldPriority).OnlyX(ctx)
require.Nil(tk.CreatedAt, "field should not be populated when it is not selected")
}
func Predicate(t *testing.T, client *ent.Client) {
require := require.New(t)
ctx := context.Background()