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

entc/integration/config/config_test.go

@@ -0,0 +1,94 @@
+// 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"
+	"fmt"
+	"go/ast"
+	"go/parser"
+	"go/token"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+
+	"entgo.io/ent/dialect/entsql"
+	"entgo.io/ent/dialect/sql"
+	"entgo.io/ent/entc/integration/config/ent"
+	"entgo.io/ent/entc/integration/config/ent/migrate"
+	"entgo.io/ent/entc/integration/config/ent/schema"
+
+	_ "github.com/go-sql-driver/mysql"
+	_ "github.com/mattn/go-sqlite3"
+)
+
+func TestSchemaConfig(t *testing.T) {
+	drv, err := sql.Open("sqlite3", "file:ent?mode=memory&cache=shared&_fk=1")
+	require.NoError(t, err)
+	defer drv.Close()
+	ctx := context.Background()
+	client := ent.NewClient(ent.Driver(drv))
+	require.NoError(t, client.Schema.Create(ctx, migrate.WithGlobalUniqueID(true)))
+	client.User.Create().SetID(1).SaveX(ctx)
+
+	// Check that the table was created with the given custom name.
+	table := schema.User{}.Annotations()[0].(entsql.Annotation).Table
+	query, args := sql.Select().Count().
+		From(sql.Table("sqlite_master")).
+		Where(sql.And(sql.EQ("type", "table"), sql.EQ("name", table))).
+		Query()
+	rows := &sql.Rows{}
+	require.NoError(t, drv.Query(ctx, query, args, rows))
+	defer rows.Close()
+	require.True(t, rows.Next(), "no rows returned")
+	var n int
+	require.NoError(t, rows.Scan(&n), "scanning count")
+	require.Equalf(t, 1, n, "expecting table %q to be exist", table)
+
+	// Check that the table was created with the expected values.
+	idIncremental := schema.User{}.Fields()[0].Descriptor().Annotations[0].(entsql.Annotation).Incremental
+	require.Equal(t, *idIncremental, migrate.Tables[0].Columns[0].Increment)
+	size := schema.User{}.Fields()[1].Descriptor().Annotations[0].(entsql.Annotation).Size
+	require.Equal(t, size, migrate.Tables[0].Columns[1].Size)
+
+	fd := schema.User{}.Fields()[1].Descriptor()
+	f, err := parser.ParseFile(token.NewFileSet(), "ent/user.go", nil, parser.ParseComments)
+	require.NoError(t, err)
+	ast.Inspect(f, func(n ast.Node) bool {
+		if f, ok := n.(*ast.Field); ok && len(f.Names) > 0 && f.Names[0].Name == fd.Name {
+			require.Contains(t, fd.Comment, f.Doc.Text())
+			return false
+		}
+		return true
+	})
+}
+
+func TestMySQL(t *testing.T) {
+	for version, port := range map[string]int{"56": 3306, "57": 3307, "8": 3308} {
+		t.Run(version, func(t *testing.T) {
+			root, err := sql.Open("mysql", fmt.Sprintf("root:pass@tcp(localhost:%d)/", port))
+			require.NoError(t, err)
+			defer root.Close()
+			ctx := context.Background()
+			err = root.Exec(ctx, "CREATE DATABASE IF NOT EXISTS config", []any{}, new(sql.Result))
+			require.NoError(t, err, "creating database")
+			defer root.Exec(ctx, "DROP DATABASE IF EXISTS config", []any{}, new(sql.Result))
+
+			drv, err := sql.Open("mysql", fmt.Sprintf("root:pass@tcp(localhost:%d)/config?parseTime=True", port))
+			require.NoError(t, err, "connecting to migrate database")
+
+			client := ent.NewClient(ent.Driver(drv))
+			// Run schema creation.
+			require.NoError(t, client.Schema.Create(ctx))
+
+			u, err := client.User.Create().SetID(200).Save(ctx)
+			require.NoError(t, err)
+			assert.Equal(t, 200, u.ID)
+			_, err = client.User.Create().Save(ctx)
+			assert.Error(t, err)
+		})
+	}
+}