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

schema/mixin/mixin_test.go

@@ -0,0 +1,100 @@
+// 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 mixin_test
+
+import (
+	"testing"
+
+	"entgo.io/ent"
+	"entgo.io/ent/schema"
+	"entgo.io/ent/schema/edge"
+	"entgo.io/ent/schema/mixin"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestTimeMixin(t *testing.T) {
+	t.Run("Create", func(t *testing.T) {
+		t.Parallel()
+		fields := mixin.CreateTime{}.Fields()
+		require.Len(t, fields, 1)
+		desc := fields[0].Descriptor()
+		assert.Equal(t, "create_time", desc.Name)
+		assert.True(t, desc.Immutable)
+		assert.NotNil(t, desc.Default)
+		assert.Nil(t, desc.UpdateDefault)
+	})
+	t.Run("Update", func(t *testing.T) {
+		t.Parallel()
+		fields := mixin.UpdateTime{}.Fields()
+		require.Len(t, fields, 1)
+		desc := fields[0].Descriptor()
+		assert.Equal(t, "update_time", desc.Name)
+		assert.NotNil(t, desc.Default)
+		assert.NotNil(t, desc.UpdateDefault)
+	})
+	t.Run("Compose", func(t *testing.T) {
+		t.Parallel()
+		fields := mixin.Time{}.Fields()
+		require.Len(t, fields, 2)
+		assert.Equal(t, "create_time", fields[0].Descriptor().Name)
+		assert.Equal(t, "update_time", fields[1].Descriptor().Name)
+	})
+}
+
+type annotation string
+
+func (annotation) Name() string { return "" }
+
+func TestAnnotateFields(t *testing.T) {
+	annotations := []schema.Annotation{
+		annotation("foo"),
+		annotation("bar"),
+		annotation("baz"),
+	}
+	fields := mixin.AnnotateFields(
+		mixin.Time{}, annotations...,
+	).Fields()
+	require.Len(t, fields, 2)
+	for _, f := range fields {
+		desc := f.Descriptor()
+		require.Len(t, desc.Annotations, len(annotations))
+		for i := range desc.Annotations {
+			assert.Equal(t, annotations[i], desc.Annotations[i])
+		}
+	}
+}
+
+type TestSchema struct {
+	ent.Schema
+}
+
+func (TestSchema) Edges() []ent.Edge {
+	return []ent.Edge{
+		edge.To("one", TestSchema.Type),
+		edge.From("two", TestSchema.Type).
+			Ref("one"),
+	}
+}
+
+func TestAnnotateEdges(t *testing.T) {
+	annotations := []schema.Annotation{
+		annotation("foo"),
+		annotation("bar"),
+		annotation("baz"),
+	}
+	edges := mixin.AnnotateEdges(
+		TestSchema{}, annotations...,
+	).Edges()
+	require.Len(t, edges, 2)
+	for _, e := range edges {
+		desc := e.Descriptor()
+		require.Len(t, desc.Annotations, len(annotations))
+		for i := range desc.Annotations {
+			assert.Equal(t, annotations[i], desc.Annotations[i])
+		}
+	}
+}