snapshot of https://github.com/ent/ent.git (master, 2026-09-15; blobs > 131072B omitted)

entc/integration/template/template_test.go

@@ -0,0 +1,96 @@
+// 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())
+}