forked from mirror/gorm
1507 lines
30 KiB
Go
1507 lines
30 KiB
Go
package tests_test
|
|
|
|
import (
|
|
"database/sql"
|
|
"encoding/json"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func toJSONString(v interface{}) []byte {
|
|
r, _ := json.Marshal(v)
|
|
return r
|
|
}
|
|
|
|
func TestNestedPreload1(t *testing.T) {
|
|
type (
|
|
Level1 struct {
|
|
ID uint
|
|
Value string
|
|
Level2ID uint
|
|
}
|
|
Level2 struct {
|
|
ID uint
|
|
Level1 Level1
|
|
Level3ID uint
|
|
}
|
|
Level3 struct {
|
|
ID uint
|
|
Name string
|
|
Level2 Level2
|
|
}
|
|
)
|
|
DB.Migrator().DropTable(&Level3{}, &Level2{}, &Level1{})
|
|
if err := DB.AutoMigrate(&Level3{}, &Level2{}, &Level1{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want := Level3{Level2: Level2{Level1: Level1{Value: "value"}}}
|
|
if err := DB.Create(&want).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
var got Level3
|
|
if err := DB.Preload("Level2").Preload("Level2.Level1").Find(&got).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Errorf("got %s; want %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
|
|
if err := DB.Preload("Level2").Preload("Level2.Level1").First(&got, "name = ?", "not_found").Error; err != gorm.ErrRecordNotFound {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestNestedPreload2(t *testing.T) {
|
|
type (
|
|
Level1 struct {
|
|
ID uint
|
|
Value string
|
|
Level2ID uint
|
|
}
|
|
Level2 struct {
|
|
ID uint
|
|
Level1s []*Level1
|
|
Level3ID uint
|
|
}
|
|
Level3 struct {
|
|
ID uint
|
|
Name string
|
|
Level2s []Level2
|
|
}
|
|
)
|
|
DB.Migrator().DropTable(&Level3{}, &Level2{}, &Level1{})
|
|
if err := DB.AutoMigrate(&Level3{}, &Level2{}, &Level1{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want := Level3{
|
|
Level2s: []Level2{
|
|
{
|
|
Level1s: []*Level1{
|
|
{Value: "value1"},
|
|
{Value: "value2"},
|
|
},
|
|
},
|
|
{
|
|
Level1s: []*Level1{
|
|
{Value: "value3"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
if err := DB.Create(&want).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
var got Level3
|
|
if err := DB.Preload("Level2s.Level1s").Find(&got).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Errorf("got %s; want %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
}
|
|
|
|
func TestNestedPreload3(t *testing.T) {
|
|
type (
|
|
Level1 struct {
|
|
ID uint
|
|
Value string
|
|
Level2ID uint
|
|
}
|
|
Level2 struct {
|
|
ID uint
|
|
Level1 Level1
|
|
Level3ID uint
|
|
}
|
|
Level3 struct {
|
|
Name string
|
|
ID uint
|
|
Level2s []Level2
|
|
}
|
|
)
|
|
DB.Migrator().DropTable(&Level3{}, &Level2{}, &Level1{})
|
|
if err := DB.AutoMigrate(&Level3{}, &Level2{}, &Level1{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want := Level3{
|
|
Level2s: []Level2{
|
|
{Level1: Level1{Value: "value1"}},
|
|
{Level1: Level1{Value: "value2"}},
|
|
},
|
|
}
|
|
if err := DB.Create(&want).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
var got Level3
|
|
if err := DB.Preload("Level2s.Level1").Find(&got).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Errorf("got %s; want %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
}
|
|
|
|
func TestNestedPreload4(t *testing.T) {
|
|
type (
|
|
Level1 struct {
|
|
ID uint
|
|
Value string
|
|
Level2ID uint
|
|
}
|
|
Level2 struct {
|
|
ID uint
|
|
Level1s []Level1
|
|
Level3ID uint
|
|
}
|
|
Level3 struct {
|
|
ID uint
|
|
Name string
|
|
Level2 Level2
|
|
}
|
|
)
|
|
DB.Migrator().DropTable(&Level3{}, &Level2{}, &Level1{})
|
|
if err := DB.AutoMigrate(&Level3{}, &Level2{}, &Level1{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want := Level3{
|
|
Level2: Level2{
|
|
Level1s: []Level1{
|
|
{Value: "value1"},
|
|
{Value: "value2"},
|
|
},
|
|
},
|
|
}
|
|
if err := DB.Create(&want).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
var got Level3
|
|
if err := DB.Preload("Level2.Level1s").Find(&got).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Errorf("got %s; want %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
}
|
|
|
|
// Slice: []Level3
|
|
func TestNestedPreload5(t *testing.T) {
|
|
type (
|
|
Level1 struct {
|
|
ID uint
|
|
Value string
|
|
Level2ID uint
|
|
}
|
|
Level2 struct {
|
|
ID uint
|
|
Level1 Level1
|
|
Level3ID uint
|
|
}
|
|
Level3 struct {
|
|
ID uint
|
|
Name string
|
|
Level2 Level2
|
|
}
|
|
)
|
|
DB.Migrator().DropTable(&Level3{}, &Level2{}, &Level1{})
|
|
if err := DB.AutoMigrate(&Level3{}, &Level2{}, &Level1{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want := make([]Level3, 2)
|
|
want[0] = Level3{Level2: Level2{Level1: Level1{Value: "value"}}}
|
|
if err := DB.Create(&want[0]).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
want[1] = Level3{Level2: Level2{Level1: Level1{Value: "value2"}}}
|
|
if err := DB.Create(&want[1]).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
var got []Level3
|
|
if err := DB.Preload("Level2").Preload("Level2.Level1").Find(&got).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Errorf("got %s; want %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
}
|
|
|
|
func TestNestedPreload6(t *testing.T) {
|
|
type (
|
|
Level1 struct {
|
|
ID uint
|
|
Value string
|
|
Level2ID uint
|
|
}
|
|
Level2 struct {
|
|
ID uint
|
|
Level1s []Level1
|
|
Level3ID uint
|
|
}
|
|
Level3 struct {
|
|
ID uint
|
|
Name string
|
|
Level2s []Level2
|
|
}
|
|
)
|
|
DB.Migrator().DropTable(&Level3{}, &Level2{}, &Level1{})
|
|
if err := DB.AutoMigrate(&Level3{}, &Level2{}, &Level1{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want := make([]Level3, 2)
|
|
want[0] = Level3{
|
|
Level2s: []Level2{
|
|
{
|
|
Level1s: []Level1{
|
|
{Value: "value1"},
|
|
{Value: "value2"},
|
|
},
|
|
},
|
|
{
|
|
Level1s: []Level1{
|
|
{Value: "value3"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
if err := DB.Create(&want[0]).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want[1] = Level3{
|
|
Level2s: []Level2{
|
|
{
|
|
Level1s: []Level1{
|
|
{Value: "value3"},
|
|
{Value: "value4"},
|
|
},
|
|
},
|
|
{
|
|
Level1s: []Level1{
|
|
{Value: "value5"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
if err := DB.Create(&want[1]).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
var got []Level3
|
|
if err := DB.Preload("Level2s.Level1s").Find(&got).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Errorf("got %s; want %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
}
|
|
|
|
func TestNestedPreload7(t *testing.T) {
|
|
type (
|
|
Level1 struct {
|
|
ID uint
|
|
Value string
|
|
Level2ID uint
|
|
}
|
|
Level2 struct {
|
|
ID uint
|
|
Level1 Level1
|
|
Level3ID uint
|
|
}
|
|
Level3 struct {
|
|
ID uint
|
|
Name string
|
|
Level2s []Level2
|
|
}
|
|
)
|
|
DB.Migrator().DropTable(&Level3{}, &Level2{}, &Level1{})
|
|
if err := DB.AutoMigrate(&Level3{}, &Level2{}, &Level1{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want := make([]Level3, 2)
|
|
want[0] = Level3{
|
|
Level2s: []Level2{
|
|
{Level1: Level1{Value: "value1"}},
|
|
{Level1: Level1{Value: "value2"}},
|
|
},
|
|
}
|
|
if err := DB.Create(&want[0]).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want[1] = Level3{
|
|
Level2s: []Level2{
|
|
{Level1: Level1{Value: "value3"}},
|
|
{Level1: Level1{Value: "value4"}},
|
|
},
|
|
}
|
|
if err := DB.Create(&want[1]).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
var got []Level3
|
|
if err := DB.Preload("Level2s.Level1").Find(&got).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Errorf("got %s; want %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
}
|
|
|
|
func TestNestedPreload8(t *testing.T) {
|
|
type (
|
|
Level1 struct {
|
|
ID uint
|
|
Value string
|
|
Level2ID uint
|
|
}
|
|
Level2 struct {
|
|
ID uint
|
|
Level1s []Level1
|
|
Level3ID uint
|
|
}
|
|
Level3 struct {
|
|
ID uint
|
|
Name string
|
|
Level2 Level2
|
|
}
|
|
)
|
|
DB.Migrator().DropTable(&Level3{}, &Level2{}, &Level1{})
|
|
if err := DB.AutoMigrate(&Level3{}, &Level2{}, &Level1{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want := make([]Level3, 2)
|
|
want[0] = Level3{
|
|
Level2: Level2{
|
|
Level1s: []Level1{
|
|
{Value: "value1"},
|
|
{Value: "value2"},
|
|
},
|
|
},
|
|
}
|
|
if err := DB.Create(&want[0]).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
want[1] = Level3{
|
|
Level2: Level2{
|
|
Level1s: []Level1{
|
|
{Value: "value3"},
|
|
{Value: "value4"},
|
|
},
|
|
},
|
|
}
|
|
if err := DB.Create(&want[1]).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
var got []Level3
|
|
if err := DB.Preload("Level2.Level1s").Find(&got).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Errorf("got %s; want %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
}
|
|
|
|
func TestNestedPreload9(t *testing.T) {
|
|
type (
|
|
Level0 struct {
|
|
ID uint
|
|
Value string
|
|
Level1ID uint
|
|
}
|
|
Level1 struct {
|
|
ID uint
|
|
Value string
|
|
Level2ID uint
|
|
Level2_1ID uint
|
|
Level0s []Level0 `json:",omitempty"`
|
|
}
|
|
Level2 struct {
|
|
ID uint
|
|
Level1s []Level1
|
|
Level3ID uint
|
|
}
|
|
Level2_1 struct {
|
|
ID uint
|
|
Level1s []Level1 `json:",omitempty"`
|
|
Level3ID uint
|
|
}
|
|
Level3 struct {
|
|
ID uint
|
|
Name string
|
|
Level2 Level2
|
|
Level2_1 Level2_1
|
|
}
|
|
)
|
|
DB.Migrator().DropTable(&Level3{}, &Level2{}, &Level1{}, &Level2_1{}, &Level0{})
|
|
if err := DB.AutoMigrate(&Level3{}, &Level2{}, &Level1{}, &Level2_1{}, &Level0{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want := make([]Level3, 2)
|
|
want[0] = Level3{
|
|
Level2: Level2{
|
|
Level1s: []Level1{
|
|
{Value: "value1"},
|
|
{Value: "value2"},
|
|
},
|
|
},
|
|
Level2_1: Level2_1{
|
|
Level1s: []Level1{
|
|
{
|
|
Value: "value1-1",
|
|
Level0s: []Level0{{Value: "Level0-1"}},
|
|
},
|
|
{
|
|
Value: "value2-2",
|
|
Level0s: []Level0{{Value: "Level0-2"}},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
if err := DB.Create(&want[0]).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
want[1] = Level3{
|
|
Level2: Level2{
|
|
Level1s: []Level1{
|
|
{Value: "value3"},
|
|
{Value: "value4"},
|
|
},
|
|
},
|
|
Level2_1: Level2_1{
|
|
Level1s: []Level1{
|
|
{
|
|
Value: "value3-3",
|
|
Level0s: []Level0{},
|
|
},
|
|
{
|
|
Value: "value4-4",
|
|
Level0s: []Level0{},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
if err := DB.Create(&want[1]).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
var got []Level3
|
|
if err := DB.Preload("Level2").Preload("Level2.Level1s").Preload("Level2_1").Preload("Level2_1.Level1s").Preload("Level2_1.Level1s.Level0s").Find(&got).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if string(toJSONString(got)) != string(toJSONString(want)) {
|
|
t.Errorf("got %s; want %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
}
|
|
|
|
type LevelA1 struct {
|
|
ID uint
|
|
Value string
|
|
}
|
|
|
|
type LevelA2 struct {
|
|
ID uint
|
|
Value string
|
|
LevelA3s []*LevelA3 `json:",omitempty"`
|
|
}
|
|
|
|
type LevelA3 struct {
|
|
ID uint
|
|
Value string
|
|
LevelA1ID sql.NullInt64
|
|
LevelA1 *LevelA1
|
|
LevelA2ID sql.NullInt64
|
|
LevelA2 *LevelA2
|
|
}
|
|
|
|
func TestNestedPreload10(t *testing.T) {
|
|
DB.Migrator().DropTable(&LevelA3{}, &LevelA2{}, &LevelA1{})
|
|
if err := DB.AutoMigrate(&LevelA1{}, &LevelA2{}, &LevelA3{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
levelA1 := &LevelA1{Value: "foo"}
|
|
if err := DB.Save(levelA1).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want := []*LevelA2{
|
|
{
|
|
Value: "bar",
|
|
LevelA3s: []*LevelA3{
|
|
{
|
|
Value: "qux",
|
|
LevelA1: levelA1,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Value: "bar 2",
|
|
LevelA3s: []*LevelA3{},
|
|
},
|
|
}
|
|
for _, levelA2 := range want {
|
|
if err := DB.Save(levelA2).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
var got []*LevelA2
|
|
if err := DB.Preload("LevelA3s.LevelA1").Find(&got).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(toJSONString(got), toJSONString(want)) {
|
|
t.Errorf("got %s; want %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
}
|
|
|
|
type LevelB1 struct {
|
|
ID uint
|
|
Value string
|
|
LevelB3s []*LevelB3
|
|
}
|
|
|
|
type LevelB2 struct {
|
|
ID uint
|
|
Value string
|
|
}
|
|
|
|
type LevelB3 struct {
|
|
ID uint
|
|
Value string
|
|
LevelB1ID sql.NullInt64
|
|
LevelB1 *LevelB1
|
|
LevelB2s []*LevelB2 `gorm:"many2many:levelb1_levelb3_levelb2s" json:",omitempty"`
|
|
}
|
|
|
|
func TestNestedPreload11(t *testing.T) {
|
|
DB.Migrator().DropTable(&LevelB3{}, &LevelB2{}, &LevelB1{})
|
|
if err := DB.AutoMigrate(&LevelB1{}, &LevelB2{}, &LevelB3{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
levelB1 := &LevelB1{Value: "foo"}
|
|
if err := DB.Create(levelB1).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
levelB3 := &LevelB3{
|
|
Value: "bar",
|
|
LevelB1ID: sql.NullInt64{Valid: true, Int64: int64(levelB1.ID)},
|
|
LevelB2s: []*LevelB2{},
|
|
}
|
|
if err := DB.Create(levelB3).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
levelB1.LevelB3s = []*LevelB3{levelB3}
|
|
|
|
want := []*LevelB1{levelB1}
|
|
var got []*LevelB1
|
|
if err := DB.Preload("LevelB3s.LevelB2s").Find(&got).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(toJSONString(got), toJSONString(want)) {
|
|
t.Errorf("got %s; want %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
}
|
|
|
|
type LevelC1 struct {
|
|
ID uint
|
|
Value string
|
|
LevelC2ID uint
|
|
}
|
|
|
|
type LevelC2 struct {
|
|
ID uint
|
|
Value string
|
|
LevelC1 LevelC1
|
|
}
|
|
|
|
type LevelC3 struct {
|
|
ID uint
|
|
Value string
|
|
LevelC2ID uint
|
|
LevelC2 LevelC2
|
|
}
|
|
|
|
func TestNestedPreload12(t *testing.T) {
|
|
DB.Migrator().DropTable(&LevelC3{}, &LevelC2{}, &LevelC1{})
|
|
if err := DB.AutoMigrate(&LevelC1{}, &LevelC2{}, &LevelC3{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
level2 := LevelC2{
|
|
Value: "c2",
|
|
LevelC1: LevelC1{
|
|
Value: "c1",
|
|
},
|
|
}
|
|
DB.Create(&level2)
|
|
|
|
want := []LevelC3{
|
|
{
|
|
Value: "c3-1",
|
|
LevelC2: level2,
|
|
}, {
|
|
Value: "c3-2",
|
|
LevelC2: level2,
|
|
},
|
|
}
|
|
|
|
for i := range want {
|
|
if err := DB.Create(&want[i]).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
var got []LevelC3
|
|
if err := DB.Preload("LevelC2").Preload("LevelC2.LevelC1").Find(&got).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Errorf("got %s; want %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
}
|
|
|
|
func TestManyToManyPreloadWithMultiPrimaryKeys(t *testing.T) {
|
|
if name := DB.Dialector.Name(); name == "sqlite" || name == "sqlserver" {
|
|
t.Skip("skip sqlite, sqlserver due to it doesn't support multiple primary keys with auto increment")
|
|
}
|
|
|
|
type (
|
|
Level1 struct {
|
|
ID uint `gorm:"primary_key;"`
|
|
LanguageCode string `gorm:"primary_key"`
|
|
Value string
|
|
}
|
|
Level2 struct {
|
|
ID uint `gorm:"primary_key;"`
|
|
LanguageCode string `gorm:"primary_key"`
|
|
Value string
|
|
Level1s []Level1 `gorm:"many2many:levels;"`
|
|
}
|
|
)
|
|
|
|
DB.Migrator().DropTable(&Level2{}, &Level1{})
|
|
DB.Migrator().DropTable("levels")
|
|
|
|
if err := DB.AutoMigrate(&Level2{}, &Level1{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want := Level2{Value: "Bob", LanguageCode: "ru", Level1s: []Level1{
|
|
{Value: "ru", LanguageCode: "ru"},
|
|
{Value: "en", LanguageCode: "en"},
|
|
}}
|
|
if err := DB.Save(&want).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want2 := Level2{Value: "Tom", LanguageCode: "zh", Level1s: []Level1{
|
|
{Value: "zh", LanguageCode: "zh"},
|
|
{Value: "de", LanguageCode: "de"},
|
|
}}
|
|
if err := DB.Save(&want2).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
var got Level2
|
|
if err := DB.Preload("Level1s").Find(&got, "value = ?", "Bob").Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
return
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Errorf("got %s; want %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
|
|
var got2 Level2
|
|
if err := DB.Preload("Level1s").Find(&got2, "value = ?", "Tom").Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got2, want2) {
|
|
t.Errorf("got %s; want %s", toJSONString(got2), toJSONString(want2))
|
|
}
|
|
|
|
var got3 []Level2
|
|
if err := DB.Preload("Level1s").Find(&got3, "value IN (?)", []string{"Bob", "Tom"}).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got3, []Level2{got, got2}) {
|
|
t.Errorf("got %s; want %s", toJSONString(got3), toJSONString([]Level2{got, got2}))
|
|
}
|
|
|
|
var got4 []Level2
|
|
if err := DB.Preload("Level1s", "value IN (?)", []string{"zh", "ru"}).Find(&got4, "value IN (?)", []string{"Bob", "Tom"}).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
var ruLevel1 Level1
|
|
var zhLevel1 Level1
|
|
DB.First(&ruLevel1, "value = ?", "ru")
|
|
DB.First(&zhLevel1, "value = ?", "zh")
|
|
|
|
got.Level1s = []Level1{ruLevel1}
|
|
got2.Level1s = []Level1{zhLevel1}
|
|
if !reflect.DeepEqual(got4, []Level2{got, got2}) {
|
|
t.Errorf("got %s; want %s", toJSONString(got4), toJSONString([]Level2{got, got2}))
|
|
}
|
|
|
|
if err := DB.Preload("Level1s").Find(&got4, "value IN (?)", []string{"non-existing"}).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestManyToManyPreloadForNestedPointer(t *testing.T) {
|
|
type (
|
|
Level1 struct {
|
|
ID uint
|
|
Value string
|
|
}
|
|
Level2 struct {
|
|
ID uint
|
|
Value string
|
|
Level1s []*Level1 `gorm:"many2many:levels;"`
|
|
}
|
|
Level3 struct {
|
|
ID uint
|
|
Value string
|
|
Level2ID sql.NullInt64
|
|
Level2 *Level2
|
|
}
|
|
)
|
|
|
|
DB.Migrator().DropTable(&Level3{}, &Level2{}, &Level1{})
|
|
DB.Migrator().DropTable("levels")
|
|
|
|
if err := DB.AutoMigrate(&Level3{}, &Level2{}, &Level1{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want := Level3{
|
|
Value: "Bob",
|
|
Level2: &Level2{
|
|
Value: "Foo",
|
|
Level1s: []*Level1{
|
|
{Value: "ru"},
|
|
{Value: "en"},
|
|
},
|
|
},
|
|
}
|
|
if err := DB.Save(&want).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want2 := Level3{
|
|
Value: "Tom",
|
|
Level2: &Level2{
|
|
Value: "Bar",
|
|
Level1s: []*Level1{
|
|
{Value: "zh"},
|
|
{Value: "de"},
|
|
},
|
|
},
|
|
}
|
|
if err := DB.Save(&want2).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
var got Level3
|
|
if err := DB.Preload("Level2.Level1s").Find(&got, "value = ?", "Bob").Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Errorf("got %s; want %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
|
|
var got2 Level3
|
|
if err := DB.Preload("Level2.Level1s").Find(&got2, "value = ?", "Tom").Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got2, want2) {
|
|
t.Errorf("got %s; want %s", toJSONString(got2), toJSONString(want2))
|
|
}
|
|
|
|
var got3 []Level3
|
|
if err := DB.Preload("Level2.Level1s").Find(&got3, "value IN (?)", []string{"Bob", "Tom"}).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got3, []Level3{got, got2}) {
|
|
t.Errorf("got %s; want %s", toJSONString(got3), toJSONString([]Level3{got, got2}))
|
|
}
|
|
|
|
var got4 []Level3
|
|
if err := DB.Preload("Level2.Level1s", "value IN (?)", []string{"zh", "ru"}).Find(&got4, "value IN (?)", []string{"Bob", "Tom"}).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
var got5 Level3
|
|
DB.Preload("Level2.Level1s").Find(&got5, "value = ?", "bogus")
|
|
|
|
var ruLevel1 Level1
|
|
var zhLevel1 Level1
|
|
DB.First(&ruLevel1, "value = ?", "ru")
|
|
DB.First(&zhLevel1, "value = ?", "zh")
|
|
|
|
got.Level2.Level1s = []*Level1{&ruLevel1}
|
|
got2.Level2.Level1s = []*Level1{&zhLevel1}
|
|
if !reflect.DeepEqual(got4, []Level3{got, got2}) {
|
|
t.Errorf("got %s; want %s", toJSONString(got4), toJSONString([]Level3{got, got2}))
|
|
}
|
|
}
|
|
|
|
func TestNestedManyToManyPreload(t *testing.T) {
|
|
type (
|
|
Level1 struct {
|
|
ID uint
|
|
Value string
|
|
}
|
|
Level2 struct {
|
|
ID uint
|
|
Value string
|
|
Level1s []*Level1 `gorm:"many2many:level1_level2;"`
|
|
}
|
|
Level3 struct {
|
|
ID uint
|
|
Value string
|
|
Level2s []Level2 `gorm:"many2many:level2_level3;"`
|
|
}
|
|
)
|
|
|
|
DB.Migrator().DropTable(&Level3{}, &Level2{}, &Level1{}, "level1_level2", "level2_level3")
|
|
|
|
if err := DB.AutoMigrate(&Level3{}, &Level2{}, &Level1{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want := Level3{
|
|
Value: "Level3",
|
|
Level2s: []Level2{
|
|
{
|
|
Value: "Bob",
|
|
Level1s: []*Level1{
|
|
{Value: "ru"},
|
|
{Value: "en"},
|
|
},
|
|
}, {
|
|
Value: "Tom",
|
|
Level1s: []*Level1{
|
|
{Value: "zh"},
|
|
{Value: "de"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
if err := DB.Save(&want).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
var got Level3
|
|
if err := DB.Preload("Level2s").Preload("Level2s.Level1s").Find(&got, "value = ?", "Level3").Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Errorf("got %s; want %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
|
|
if err := DB.Preload("Level2s.Level1s").First(&got, "value = ?", "not_found").Error; err != gorm.ErrRecordNotFound {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestNestedManyToManyPreload2(t *testing.T) {
|
|
type (
|
|
Level1 struct {
|
|
ID uint
|
|
Value string
|
|
}
|
|
Level2 struct {
|
|
ID uint
|
|
Value string
|
|
Level1s []*Level1 `gorm:"many2many:level1_level2;"`
|
|
}
|
|
Level3 struct {
|
|
ID uint
|
|
Value string
|
|
Level2ID sql.NullInt64
|
|
Level2 *Level2
|
|
}
|
|
)
|
|
|
|
DB.Migrator().DropTable(&Level3{}, &Level2{}, &Level1{})
|
|
DB.Migrator().DropTable("level1_level2")
|
|
|
|
if err := DB.AutoMigrate(&Level3{}, &Level2{}, &Level1{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want := Level3{
|
|
Value: "Level3",
|
|
Level2: &Level2{
|
|
Value: "Bob",
|
|
Level1s: []*Level1{
|
|
{Value: "ru"},
|
|
{Value: "en"},
|
|
},
|
|
},
|
|
}
|
|
|
|
if err := DB.Save(&want).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
var got Level3
|
|
if err := DB.Preload("Level2.Level1s").Find(&got, "value = ?", "Level3").Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Errorf("got %s; want %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
|
|
if err := DB.Preload("Level2.Level1s").First(&got, "value = ?", "not_found").Error; err != gorm.ErrRecordNotFound {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestNestedManyToManyPreload3(t *testing.T) {
|
|
type (
|
|
Level1 struct {
|
|
ID uint
|
|
Value string
|
|
}
|
|
Level2 struct {
|
|
ID uint
|
|
Value string
|
|
Level1s []*Level1 `gorm:"many2many:level1_level2;"`
|
|
}
|
|
Level3 struct {
|
|
ID uint
|
|
Value string
|
|
Level2ID sql.NullInt64
|
|
Level2 *Level2
|
|
}
|
|
)
|
|
|
|
DB.Migrator().DropTable(&Level3{}, &Level2{}, &Level1{}, "level1_level2")
|
|
|
|
if err := DB.AutoMigrate(&Level3{}, &Level2{}, &Level1{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
level1Zh := &Level1{Value: "zh"}
|
|
level1Ru := &Level1{Value: "ru"}
|
|
level1En := &Level1{Value: "en"}
|
|
|
|
level21 := &Level2{
|
|
Value: "Level2-1",
|
|
Level1s: []*Level1{level1Zh, level1Ru},
|
|
}
|
|
|
|
level22 := &Level2{
|
|
Value: "Level2-2",
|
|
Level1s: []*Level1{level1Zh, level1En},
|
|
}
|
|
|
|
wants := []*Level3{
|
|
{
|
|
Value: "Level3-1",
|
|
Level2: level21,
|
|
},
|
|
{
|
|
Value: "Level3-2",
|
|
Level2: level22,
|
|
},
|
|
{
|
|
Value: "Level3-3",
|
|
Level2: level21,
|
|
},
|
|
}
|
|
|
|
for _, want := range wants {
|
|
if err := DB.Save(want).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
var gots []*Level3
|
|
if err := DB.Preload("Level2.Level1s", func(db *gorm.DB) *gorm.DB {
|
|
return db.Order("level1.id ASC")
|
|
}).Find(&gots).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(gots, wants) {
|
|
t.Errorf("got %s; want %s", toJSONString(gots), toJSONString(wants))
|
|
}
|
|
}
|
|
|
|
func TestNestedManyToManyPreload3ForStruct(t *testing.T) {
|
|
type (
|
|
Level1 struct {
|
|
ID uint
|
|
Value string
|
|
}
|
|
Level2 struct {
|
|
ID uint
|
|
Value string
|
|
Level1s []Level1 `gorm:"many2many:level1_level2;"`
|
|
}
|
|
Level3 struct {
|
|
ID uint
|
|
Value string
|
|
Level2ID sql.NullInt64
|
|
Level2 Level2
|
|
}
|
|
)
|
|
|
|
DB.Migrator().DropTable(&Level3{}, &Level2{}, &Level1{})
|
|
DB.Migrator().DropTable("level1_level2")
|
|
|
|
if err := DB.AutoMigrate(&Level3{}, &Level2{}, &Level1{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
level1Zh := Level1{Value: "zh"}
|
|
level1Ru := Level1{Value: "ru"}
|
|
level1En := Level1{Value: "en"}
|
|
|
|
level21 := Level2{
|
|
Value: "Level2-1",
|
|
Level1s: []Level1{level1Zh, level1Ru},
|
|
}
|
|
|
|
level22 := Level2{
|
|
Value: "Level2-2",
|
|
Level1s: []Level1{level1Zh, level1En},
|
|
}
|
|
|
|
wants := []*Level3{
|
|
{
|
|
Value: "Level3-1",
|
|
Level2: level21,
|
|
},
|
|
{
|
|
Value: "Level3-2",
|
|
Level2: level22,
|
|
},
|
|
{
|
|
Value: "Level3-3",
|
|
Level2: level21,
|
|
},
|
|
}
|
|
|
|
for _, want := range wants {
|
|
if err := DB.Save(want).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
var gots []*Level3
|
|
if err := DB.Preload("Level2.Level1s", func(db *gorm.DB) *gorm.DB {
|
|
return db.Order("level1.id ASC")
|
|
}).Find(&gots).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(gots, wants) {
|
|
t.Errorf("got %s; want %s", toJSONString(gots), toJSONString(wants))
|
|
}
|
|
}
|
|
|
|
func TestNestedManyToManyPreload4(t *testing.T) {
|
|
type (
|
|
Level4 struct {
|
|
ID uint
|
|
Value string
|
|
Level3ID uint
|
|
}
|
|
Level3 struct {
|
|
ID uint
|
|
Value string
|
|
Level4s []*Level4
|
|
}
|
|
Level2 struct {
|
|
ID uint
|
|
Value string
|
|
Level3s []*Level3 `gorm:"many2many:level2_level3;"`
|
|
}
|
|
Level1 struct {
|
|
ID uint
|
|
Value string
|
|
Level2s []*Level2 `gorm:"many2many:level1_level2;"`
|
|
}
|
|
)
|
|
|
|
DB.Migrator().DropTable("level1_level2", "level2_level3")
|
|
DB.Migrator().DropTable(&Level3{}, &Level2{}, &Level1{}, &Level4{})
|
|
|
|
dummy := Level1{
|
|
Value: "Level1",
|
|
Level2s: []*Level2{{
|
|
Value: "Level2",
|
|
Level3s: []*Level3{{
|
|
Value: "Level3",
|
|
Level4s: []*Level4{{
|
|
Value: "Level4",
|
|
}},
|
|
}},
|
|
}},
|
|
}
|
|
|
|
if err := DB.AutoMigrate(&Level4{}, &Level3{}, &Level2{}, &Level1{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err := DB.Save(&dummy).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
var level1 Level1
|
|
if err := DB.Preload("Level2s").Preload("Level2s.Level3s").Preload("Level2s.Level3s.Level4s").First(&level1).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestManyToManyPreloadForPointer(t *testing.T) {
|
|
type (
|
|
Level1 struct {
|
|
ID uint
|
|
Value string
|
|
}
|
|
Level2 struct {
|
|
ID uint
|
|
Value string
|
|
Level1s []*Level1 `gorm:"many2many:levels;"`
|
|
}
|
|
)
|
|
|
|
DB.Migrator().DropTable("levels", &Level2{}, &Level1{})
|
|
|
|
if err := DB.AutoMigrate(&Level2{}, &Level1{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want := Level2{Value: "Bob", Level1s: []*Level1{
|
|
{Value: "ru"},
|
|
{Value: "en"},
|
|
}}
|
|
if err := DB.Save(&want).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want2 := Level2{Value: "Tom", Level1s: []*Level1{
|
|
{Value: "zh"},
|
|
{Value: "de"},
|
|
}}
|
|
if err := DB.Save(&want2).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
var got Level2
|
|
if err := DB.Preload("Level1s").Find(&got, "value = ?", "Bob").Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Errorf("got %s; want %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
|
|
var got2 Level2
|
|
if err := DB.Preload("Level1s").Find(&got2, "value = ?", "Tom").Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got2, want2) {
|
|
t.Errorf("got %s; want %s", toJSONString(got2), toJSONString(want2))
|
|
}
|
|
|
|
var got3 []Level2
|
|
if err := DB.Preload("Level1s").Find(&got3, "value IN (?)", []string{"Bob", "Tom"}).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got3, []Level2{got, got2}) {
|
|
t.Errorf("got %s; want %s", toJSONString(got3), toJSONString([]Level2{got, got2}))
|
|
}
|
|
|
|
var got4 []Level2
|
|
if err := DB.Preload("Level1s", "value IN (?)", []string{"zh", "ru"}).Find(&got4, "value IN (?)", []string{"Bob", "Tom"}).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
var got5 Level2
|
|
DB.Preload("Level1s").First(&got5, "value = ?", "bogus")
|
|
|
|
var ruLevel1 Level1
|
|
var zhLevel1 Level1
|
|
DB.First(&ruLevel1, "value = ?", "ru")
|
|
DB.First(&zhLevel1, "value = ?", "zh")
|
|
|
|
got.Level1s = []*Level1{&ruLevel1}
|
|
got2.Level1s = []*Level1{&zhLevel1}
|
|
if !reflect.DeepEqual(got4, []Level2{got, got2}) {
|
|
t.Errorf("got %s; want %s", toJSONString(got4), toJSONString([]Level2{got, got2}))
|
|
}
|
|
}
|
|
|
|
func TestNilPointerSlice(t *testing.T) {
|
|
type (
|
|
Level3 struct {
|
|
ID uint
|
|
Value string
|
|
}
|
|
Level2 struct {
|
|
ID uint
|
|
Value string
|
|
Level3ID uint
|
|
Level3 *Level3
|
|
}
|
|
Level1 struct {
|
|
ID uint
|
|
Value string
|
|
Level2ID *uint
|
|
Level2 *Level2
|
|
}
|
|
)
|
|
|
|
DB.Migrator().DropTable(&Level3{}, &Level2{}, &Level1{})
|
|
|
|
if err := DB.AutoMigrate(&Level3{}, &Level2{}, &Level1{}); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want := Level1{
|
|
Value: "Bob",
|
|
Level2: &Level2{
|
|
Value: "en",
|
|
Level3: &Level3{
|
|
Value: "native",
|
|
},
|
|
},
|
|
}
|
|
if err := DB.Save(&want).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want2 := Level1{
|
|
Value: "Tom",
|
|
Level2: nil,
|
|
}
|
|
if err := DB.Save(&want2).Error; err != nil {
|
|
t.Fatalf("Got error %v", err)
|
|
}
|
|
|
|
var got []Level1
|
|
if err := DB.Preload("Level2").Preload("Level2.Level3").Find(&got).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if len(got) != 2 {
|
|
t.Errorf("got %v items, expected 2", len(got))
|
|
}
|
|
|
|
if !reflect.DeepEqual(got[0], want) && !reflect.DeepEqual(got[1], want) {
|
|
t.Errorf("got %s; want array containing %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
|
|
if !reflect.DeepEqual(got[0], want2) && !reflect.DeepEqual(got[1], want2) {
|
|
t.Errorf("got %s; want array containing %s", toJSONString(got), toJSONString(want2))
|
|
}
|
|
}
|
|
|
|
func TestNilPointerSlice2(t *testing.T) {
|
|
type (
|
|
Level4 struct {
|
|
ID uint
|
|
}
|
|
Level3 struct {
|
|
ID uint
|
|
Level4ID sql.NullInt64 `sql:"index"`
|
|
Level4 *Level4
|
|
}
|
|
Level2 struct {
|
|
ID uint
|
|
Level3s []*Level3 `gorm:"many2many:level2_level3s"`
|
|
}
|
|
Level1 struct {
|
|
ID uint
|
|
Level2ID sql.NullInt64 `sql:"index"`
|
|
Level2 *Level2
|
|
}
|
|
)
|
|
|
|
DB.Migrator().DropTable(&Level3{}, &Level2{}, &Level1{}, &Level4{})
|
|
|
|
if err := DB.AutoMigrate(new(Level4), new(Level3), new(Level2), new(Level1)); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want := new(Level1)
|
|
if err := DB.Save(want).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
got := new(Level1)
|
|
err := DB.Preload("Level2.Level3s.Level4").Last(&got).Error
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Errorf("got %s; want %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
}
|
|
|
|
func TestPrefixedPreloadDuplication(t *testing.T) {
|
|
type (
|
|
Level4 struct {
|
|
ID uint
|
|
Name string
|
|
Level3ID uint
|
|
}
|
|
Level3 struct {
|
|
ID uint
|
|
Name string
|
|
Level4s []*Level4 `json:",omitempty"`
|
|
}
|
|
Level2 struct {
|
|
ID uint
|
|
Name string
|
|
Level3ID sql.NullInt64 `sql:"index"`
|
|
Level3 *Level3
|
|
}
|
|
Level1 struct {
|
|
ID uint
|
|
Name string
|
|
Level2ID sql.NullInt64 `sql:"index"`
|
|
Level2 *Level2
|
|
}
|
|
)
|
|
|
|
DB.Migrator().DropTable(&Level3{}, &Level2{}, &Level1{}, &Level4{})
|
|
|
|
if err := DB.AutoMigrate(new(Level3), new(Level4), new(Level2), new(Level1)); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
lvl := &Level3{}
|
|
if err := DB.Save(lvl).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
sublvl1 := &Level4{Level3ID: lvl.ID}
|
|
if err := DB.Save(sublvl1).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
sublvl2 := &Level4{Level3ID: lvl.ID}
|
|
if err := DB.Save(sublvl2).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
lvl.Level4s = []*Level4{sublvl1, sublvl2}
|
|
|
|
want1 := Level1{
|
|
Level2: &Level2{
|
|
Level3: lvl,
|
|
},
|
|
}
|
|
if err := DB.Save(&want1).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want2 := Level1{
|
|
Level2: &Level2{
|
|
Level3: lvl,
|
|
},
|
|
}
|
|
if err := DB.Save(&want2).Error; err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
want := []Level1{want1, want2}
|
|
|
|
var got []Level1
|
|
err := DB.Preload("Level2.Level3.Level4s").Find(&got).Error
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Errorf("got %s; want %s", toJSONString(got), toJSONString(want))
|
|
}
|
|
}
|
|
|
|
func TestPreloadManyToManyCallbacks(t *testing.T) {
|
|
type (
|
|
Level2 struct {
|
|
ID uint
|
|
Name string
|
|
}
|
|
Level1 struct {
|
|
ID uint
|
|
Name string
|
|
Level2s []Level2 `gorm:"many2many:level1_level2s"`
|
|
}
|
|
)
|
|
|
|
DB.Migrator().DropTable("level1_level2s", &Level2{}, &Level1{})
|
|
|
|
if err := DB.AutoMigrate(new(Level1), new(Level2)); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
lvl := Level1{
|
|
Name: "l1",
|
|
Level2s: []Level2{
|
|
{Name: "l2-1"}, {Name: "l2-2"},
|
|
},
|
|
}
|
|
DB.Save(&lvl)
|
|
|
|
called := 0
|
|
|
|
DB.Callback().Query().After("gorm:query").Register("TestPreloadManyToManyCallbacks", func(_ *gorm.DB) {
|
|
called = called + 1
|
|
})
|
|
|
|
DB.Preload("Level2s").First(&Level1{}, "id = ?", lvl.ID)
|
|
|
|
if called != 3 {
|
|
t.Errorf("Wanted callback to be called 3 times but got %d", called)
|
|
}
|
|
}
|