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

entc/gen: make generated client to implement the ent.Mutator interface (#3161)

Browse Source
This commit is contained in:
Ariel Mashraki
2022-12-09 21:18:25 +02:00
committed by GitHub
parent 3f4916ff8b
commit d0c5afa705
85 changed files with 3522 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
entc/integration/hooks/hooks_test.go
View File

@@ -9,6 +9,7 @@ import (
"fmt"
"sort"
"testing"
"time"
"entgo.io/ent/entc/integration/hooks/ent"
"entgo.io/ent/entc/integration/hooks/ent/card"
@@ -105,6 +106,48 @@ func TestMutationClient(t *testing.T) {
require.Equal(t, a8m.Name, crd.Name)
}
func TestMutatorClient(t *testing.T) {
ctx := context.Background()
client := enttest.Open(t, "sqlite3", "file:ent?mode=memory&cache=shared&_fk=1")
defer client.Close()
client.Use(
hook.On(
func(next ent.Mutator) ent.Mutator {
return hook.CardFunc(func(ctx context.Context, m *ent.CardMutation) (ent.Value, error) {
op := ent.OpUpdate
if _, exists := m.ID(); exists && m.Op().Is(ent.OpDeleteOne) {
op = ent.OpUpdateOne
}
// Change the operation to update.
m.SetOp(op)
// Record when card was expired.
m.SetExpiredAt(time.Now())
// Ensure card was not expired before.
m.Where(card.ExpiredAtIsNil())
return m.Client().Mutate(ctx, m)
})
},
ent.OpDelete|ent.OpDeleteOne,
),
)
c1 := client.Card.Create().SetNumber("1234").SaveX(ctx)
client.Card.DeleteOne(c1).ExecX(ctx)
expired := client.Card.Query().OnlyX(ctx)
require.False(t, expired.ExpiredAt.IsZero())
client.Card.Create().SetNumber("4567").ExecX(ctx)
client.Card.Create().SetNumber("7890").ExecX(ctx)
client.Card.Delete().Where(card.Number("4567")).ExecX(ctx)
cards := client.Card.Query().Order(ent.Asc(card.FieldNumber)).AllX(ctx)
require.Len(t, cards, 3)
require.True(t, cards[0].ExpiredAt.Equal(expired.ExpiredAt), "expired field should not be updated")
require.False(t, cards[1].ExpiredAt.IsZero())
require.True(t, cards[2].ExpiredAt.IsZero())
client.Card.Delete().ExecX(ctx)
require.False(t, client.Card.Query().Where(card.ExpiredAtIsNil()).ExistX(ctx))
}
func TestMutationTx(t *testing.T) {
ctx := context.Background()
client := enttest.Open(t, "sqlite3", "file:ent?mode=memory&cache=shared&_fk=1", enttest.WithMigrateOptions(migrate.WithGlobalUniqueID(true)))