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

entc/integration/index_test.go

@@ -0,0 +1,73 @@
+// 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 integration
+
+import (
+	"context"
+	"testing"
+
+	"entgo.io/ent/entc/integration/ent"
+
+	"github.com/stretchr/testify/require"
+)
+
+func Indexes(t *testing.T, client *ent.Client) {
+	ctx := context.Background()
+	require := require.New(t)
+	t.Log("prevent inserting 2 files with the same (name, user)")
+	f1, err := client.File.Create().SetName("foo").SetSize(10).SetUser("bar").Save(ctx)
+	require.NoError(err)
+	require.Equal("foo", f1.Name)
+	require.Equal("bar", *f1.User)
+	f2, err := client.File.Create().SetName("foo").SetSize(10).SetUser("bar").Save(ctx)
+	require.Nil(f2)
+	require.Error(err)
+	require.True(ent.IsConstraintError(err))
+
+	t.Log("deletion should allow recreation")
+	client.File.DeleteOne(f1).ExecX(ctx)
+	f3, err := client.File.Create().SetName("foo").SetSize(10).SetUser("bar").Save(ctx)
+	require.NoError(err)
+	require.Equal("foo", f3.Name)
+	require.Equal("bar", *f3.User)
+
+	t.Log("allow inserting 2 files the same name, type and NULL user (optional field)")
+	png := client.FileType.Create().SetName("png").SaveX(ctx)
+	f4 := client.File.Create().SetName("foo").SetSize(10).SetType(png).SaveX(ctx)
+	f5 := client.File.Create().SetName("foo").SetSize(10).SetType(png).SaveX(ctx)
+
+	t.Log("index on edge sub-graph")
+	a8m := client.User.Create().SetName("a8m").SetAge(18).SaveX(ctx)
+	err = a8m.Update().AddFiles(f4).Exec(ctx)
+	require.NoError(err)
+	err = a8m.Update().AddFiles(f5).Exec(ctx)
+	require.Error(err)
+	require.True(ent.IsConstraintError(err), "cannot have 2 files with the same (name, type, owner)")
+	png.Update().RemoveFiles(f5).ExecX(ctx)
+	err = a8m.Update().AddFiles(f5).Exec(ctx)
+	require.NoError(err)
+	err = png.Update().AddFiles(f5).Exec(ctx)
+	require.Error(err)
+	require.True(ent.IsConstraintError(err))
+	a8m.Update().RemoveFiles(f4, f5).ExecX(ctx)
+	png.Update().AddFiles(f5).ExecX(ctx)
+
+	t.Log("prevent inserting duplicates files in the same insert")
+	err = a8m.Update().AddFiles(f4, f5).Exec(ctx)
+	require.Error(err)
+	require.True(ent.IsConstraintError(err))
+	require.Zero(a8m.QueryFiles().CountX(ctx))
+
+	t.Log("edge indexes should applied on the edge sub-graph")
+	nati := client.User.Create().SetName("nati").SetAge(18).AddFiles(f5).SaveX(ctx)
+	err = nati.Update().AddFiles(f4).Exec(ctx)
+	require.Error(err)
+	require.True(ent.IsConstraintError(err))
+	err = a8m.Update().AddFiles(f4).Exec(ctx)
+	require.NoError(err)
+
+	require.Equal(1, a8m.QueryFiles().CountX(ctx))
+	require.Equal(1, nati.QueryFiles().CountX(ctx))
+}