mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
9e809635b2
Also, for some reason, the TimeMixin.UpdateTime was an immutable field, but this was incorrent, because the codegen just skip generating update setters to it. Removing the Immutable modifier allows users to set this field explicitly.
101 lines
2.4 KiB
Go
101 lines
2.4 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 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])
|
|
}
|
|
}
|
|
}
|