forked from mirror/gorm
117 lines
2.9 KiB
Go
117 lines
2.9 KiB
Go
package gorm_test
|
|
|
|
import (
|
|
"os"
|
|
"reflect"
|
|
"sort"
|
|
"testing"
|
|
)
|
|
|
|
type Blog struct {
|
|
ID uint `gorm:"primary_key"`
|
|
Locale string `gorm:"primary_key"`
|
|
Subject string
|
|
Body string
|
|
Tags []Tag `gorm:"many2many:blog_tags;"`
|
|
}
|
|
|
|
type Tag struct {
|
|
ID uint `gorm:"primary_key"`
|
|
Locale string `gorm:"primary_key"`
|
|
Value string
|
|
}
|
|
|
|
func compareTags(tags []Tag, contents []string) bool {
|
|
var tagContents []string
|
|
for _, tag := range tags {
|
|
tagContents = append(tagContents, tag.Value)
|
|
}
|
|
sort.Strings(tagContents)
|
|
sort.Strings(contents)
|
|
return reflect.DeepEqual(tagContents, contents)
|
|
}
|
|
|
|
func TestManyToManyWithMultiPrimaryKeys(t *testing.T) {
|
|
if dialect := os.Getenv("GORM_DIALECT"); dialect != "" && dialect != "sqlite" {
|
|
DB.DropTable(&Blog{}, &Tag{})
|
|
DB.DropTable("blog_tags")
|
|
DB.CreateTable(&Blog{}, &Tag{})
|
|
blog := Blog{
|
|
Locale: "ZH",
|
|
Subject: "subject",
|
|
Body: "body",
|
|
Tags: []Tag{
|
|
{Locale: "ZH", Value: "tag1"},
|
|
{Locale: "ZH", Value: "tag2"},
|
|
},
|
|
}
|
|
|
|
DB.Save(&blog)
|
|
if !compareTags(blog.Tags, []string{"tag1", "tag2"}) {
|
|
t.Errorf("Blog should has two tags")
|
|
}
|
|
|
|
// Append
|
|
var tag3 = &Tag{Locale: "ZH", Value: "tag3"}
|
|
DB.Model(&blog).Association("Tags").Append([]*Tag{tag3})
|
|
if !compareTags(blog.Tags, []string{"tag1", "tag2", "tag3"}) {
|
|
t.Errorf("Blog should has three tags after Append")
|
|
}
|
|
|
|
if DB.Model(&blog).Association("Tags").Count() != 3 {
|
|
t.Errorf("Blog should has three tags after Append")
|
|
}
|
|
|
|
var tags []Tag
|
|
DB.Model(&blog).Related(&tags, "Tags")
|
|
if !compareTags(tags, []string{"tag1", "tag2", "tag3"}) {
|
|
t.Errorf("Should find 3 tags with Related")
|
|
}
|
|
|
|
var blog1 Blog
|
|
DB.Preload("Tags").Find(&blog1)
|
|
if !compareTags(blog1.Tags, []string{"tag1", "tag2", "tag3"}) {
|
|
t.Errorf("Preload many2many relations")
|
|
}
|
|
|
|
// Replace
|
|
var tag5 = &Tag{Locale: "ZH", Value: "tag5"}
|
|
var tag6 = &Tag{Locale: "ZH", Value: "tag6"}
|
|
DB.Model(&blog).Association("Tags").Replace(tag5, tag6)
|
|
var tags2 []Tag
|
|
DB.Model(&blog).Related(&tags2, "Tags")
|
|
if !compareTags(tags2, []string{"tag5", "tag6"}) {
|
|
t.Errorf("Should find 2 tags after Replace")
|
|
}
|
|
|
|
if DB.Model(&blog).Association("Tags").Count() != 2 {
|
|
t.Errorf("Blog should has three tags after Replace")
|
|
}
|
|
|
|
// Delete
|
|
DB.Model(&blog).Association("Tags").Delete(tag5)
|
|
var tags3 []Tag
|
|
DB.Model(&blog).Related(&tags3, "Tags")
|
|
if !compareTags(tags3, []string{"tag6"}) {
|
|
t.Errorf("Should find 1 tags after Delete")
|
|
}
|
|
|
|
if DB.Model(&blog).Association("Tags").Count() != 1 {
|
|
t.Errorf("Blog should has three tags after Delete")
|
|
}
|
|
|
|
DB.Model(&blog).Association("Tags").Delete(tag3)
|
|
var tags4 []Tag
|
|
DB.Model(&blog).Related(&tags4, "Tags")
|
|
if !compareTags(tags4, []string{"tag6"}) {
|
|
t.Errorf("Tag should not be deleted when Delete with a unrelated tag")
|
|
}
|
|
|
|
// Clear
|
|
DB.Model(&blog).Association("Tags").Clear()
|
|
if DB.Model(&blog).Association("Tags").Count() != 0 {
|
|
t.Errorf("All tags should be cleared")
|
|
}
|
|
}
|
|
}
|