mirror of
https://github.com/ent/ent.git
synced 2026-05-22 09:31:45 +03:00
97 lines
2.9 KiB
Go
97 lines
2.9 KiB
Go
// Copyright 2019-present Facebook Inc. All rights reserved.
|
|
// This source code is licensed under the Apache 2.0 license found
|
|
// in the LICENSE file in the root directory of this source tree.
|
|
|
|
package template
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"entgo.io/ent/dialect/sql"
|
|
"entgo.io/ent/entc/integration/template/ent"
|
|
"entgo.io/ent/entc/integration/template/ent/hook"
|
|
"entgo.io/ent/entc/integration/template/ent/migrate"
|
|
"entgo.io/ent/entc/integration/template/ent/pet"
|
|
"entgo.io/ent/entc/integration/template/ent/user"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCustomTemplate(t *testing.T) {
|
|
client, err := ent.Open(
|
|
"sqlite3",
|
|
"file:ent?mode=memory&cache=shared&_fk=1",
|
|
// Custom config option.
|
|
ent.HTTPClient(http.DefaultClient),
|
|
)
|
|
require.NoError(t, err)
|
|
defer client.Close()
|
|
ctx := context.Background()
|
|
require.NoError(t, client.Schema.Create(ctx, migrate.WithGlobalUniqueID(true)))
|
|
client.User.Use(func(next ent.Mutator) ent.Mutator {
|
|
return hook.UserFunc(func(ctx context.Context, m *ent.UserMutation) (ent.Value, error) {
|
|
// Access the injected HTTP client here.
|
|
_ = m.HTTPClient
|
|
return next.Mutate(ctx, m)
|
|
})
|
|
})
|
|
|
|
p := client.Pet.Create().SetAge(1).SaveX(ctx)
|
|
u := client.User.Create().SetName("a8m").AddPets(p).SaveX(ctx)
|
|
g := client.Group.Create().SetMaxUsers(10).SaveX(ctx)
|
|
|
|
node, err := client.Node(ctx, p.ID)
|
|
require.NoError(t, err)
|
|
require.Equal(t, p.ID, node.ID)
|
|
require.Equal(t, &ent.Field{Type: "int", Name: "Age", Value: "1"}, node.Fields[0])
|
|
require.Equal(t, &ent.Edge{Type: "User", Name: "Owner", IDs: []int{u.ID}}, node.Edges[0])
|
|
|
|
node, err = client.Node(ctx, u.ID)
|
|
require.NoError(t, err)
|
|
require.Equal(t, u.ID, node.ID)
|
|
require.Equal(t, &ent.Field{Type: "string", Name: "Name", Value: "\"a8m\""}, node.Fields[0])
|
|
require.Equal(t, &ent.Edge{Type: "Pet", Name: "Pets", IDs: []int{p.ID}}, node.Edges[0])
|
|
|
|
node, err = client.Node(ctx, g.ID)
|
|
require.NoError(t, err)
|
|
require.Equal(t, g.ID, node.ID)
|
|
require.Equal(t, &ent.Field{Type: "int", Name: "MaxUsers", Value: "10"}, node.Fields[0])
|
|
|
|
// check for client additional fields.
|
|
require.True(t, reflect.ValueOf(client).Elem().FieldByName("tables").IsValid())
|
|
|
|
result := client.User.Query().Where(user.NameGlob("a8*")).
|
|
AllX(ctx)
|
|
require.Equal(t, 1, len(result))
|
|
|
|
var v []struct{ ID, Owner int }
|
|
client.Pet.Query().
|
|
Modify(func(s *sql.Selector) {
|
|
t := sql.Table(user.Table)
|
|
s.Join(t).On(s.C(pet.OwnerColumn), t.C(user.FieldID))
|
|
s.Select(s.C(pet.FieldID), sql.As(t.C(user.FieldID), "owner"))
|
|
}).
|
|
Select().
|
|
ScanX(ctx, &v)
|
|
require.Equal(t, p.ID, v[0].ID)
|
|
require.Equal(t, u.ID, v[0].Owner)
|
|
|
|
var sum int
|
|
for _, age := range client.Pet.Query().Select(pet.FieldAge).IntsX(ctx) {
|
|
sum += age
|
|
}
|
|
got := client.Pet.Query().
|
|
Modify(func(s *sql.Selector) {
|
|
s.Select(sql.Sum(pet.FieldAge))
|
|
}).
|
|
Select().
|
|
IntX(ctx)
|
|
require.Equal(t, sum, got)
|
|
|
|
require.Equal(t, 20, client.HiddenData())
|
|
}
|