From b907500a477e6040523c942451b874606d09a86b Mon Sep 17 00:00:00 2001
From: Jinzhu <wosmvp@gmail.com>
Date: Sun, 3 Jan 2016 16:18:51 +0800
Subject: [PATCH] Add test for many2many relations with customized foreign keys

---
 multi_primary_keys_test.go | 129 +++++++++++++++++++++++++++++++++++--
 1 file changed, 124 insertions(+), 5 deletions(-)

diff --git a/multi_primary_keys_test.go b/multi_primary_keys_test.go
index 061822ef..27997fb7 100644
--- a/multi_primary_keys_test.go
+++ b/multi_primary_keys_test.go
@@ -8,17 +8,19 @@ import (
 )
 
 type Blog struct {
-	ID      uint   `gorm:"primary_key"`
-	Locale  string `gorm:"primary_key"`
-	Subject string
-	Body    string
-	Tags    []Tag `gorm:"many2many:blog_tags;"`
+	ID         uint   `gorm:"primary_key"`
+	Locale     string `gorm:"primary_key"`
+	Subject    string
+	Body       string
+	Tags       []Tag `gorm:"many2many:blog_tags;"`
+	SharedTags []Tag `gorm:"many2many:shared_blog_tags;ForeignKey:id;AssociationForeignKey:id"`
 }
 
 type Tag struct {
 	ID     uint   `gorm:"primary_key"`
 	Locale string `gorm:"primary_key"`
 	Value  string
+	Blogs  []*Blog `gorm:"many2many:"blogs_tags`
 }
 
 func compareTags(tags []Tag, contents []string) bool {
@@ -114,3 +116,120 @@ func TestManyToManyWithMultiPrimaryKeys(t *testing.T) {
 		}
 	}
 }
+
+func TestManyToManyWithCustomizedForeignKeys(t *testing.T) {
+	if dialect := os.Getenv("GORM_DIALECT"); dialect != "" && dialect != "sqlite" {
+		DB.DropTable(&Blog{}, &Tag{})
+		DB.DropTable("shared_blog_tags")
+		DB.CreateTable(&Blog{}, &Tag{})
+		blog := Blog{
+			Locale:  "ZH",
+			Subject: "subject",
+			Body:    "body",
+			SharedTags: []Tag{
+				{Locale: "ZH", Value: "tag1"},
+				{Locale: "ZH", Value: "tag2"},
+			},
+		}
+		DB.Save(&blog)
+
+		blog2 := Blog{
+			ID:     blog.ID,
+			Locale: "ZH",
+		}
+		DB.Save(&blog2)
+
+		if !compareTags(blog.SharedTags, []string{"tag1", "tag2"}) {
+			t.Errorf("Blog should has two tags")
+		}
+
+		// Append
+		var tag3 = &Tag{Locale: "ZH", Value: "tag3"}
+		DB.Model(&blog).Association("SharedTags").Append([]*Tag{tag3})
+		if !compareTags(blog.SharedTags, []string{"tag1", "tag2", "tag3"}) {
+			t.Errorf("Blog should has three tags after Append")
+		}
+
+		if DB.Model(&blog).Association("SharedTags").Count() != 3 {
+			t.Errorf("Blog should has three tags after Append")
+		}
+
+		if DB.Model(&blog2).Association("SharedTags").Count() != 3 {
+			t.Errorf("Blog should has three tags after Append")
+		}
+
+		var tags []Tag
+		DB.Model(&blog).Related(&tags, "SharedTags")
+		if !compareTags(tags, []string{"tag1", "tag2", "tag3"}) {
+			t.Errorf("Should find 3 tags with Related")
+		}
+
+		DB.Model(&blog2).Related(&tags, "SharedTags")
+		if !compareTags(tags, []string{"tag1", "tag2", "tag3"}) {
+			t.Errorf("Should find 3 tags with Related")
+		}
+
+		var blog1 Blog
+		DB.Preload("SharedTags").Find(&blog1)
+		if !compareTags(blog1.SharedTags, []string{"tag1", "tag2", "tag3"}) {
+			t.Errorf("Preload many2many relations")
+		}
+
+		var tag4 = &Tag{Locale: "ZH", Value: "tag4"}
+		DB.Model(&blog2).Association("SharedTags").Append(tag4)
+
+		DB.Model(&blog).Related(&tags, "SharedTags")
+		if !compareTags(tags, []string{"tag1", "tag2", "tag3", "tag4"}) {
+			t.Errorf("Should find 3 tags with Related")
+		}
+
+		DB.Model(&blog2).Related(&tags, "SharedTags")
+		if !compareTags(tags, []string{"tag1", "tag2", "tag3", "tag4"}) {
+			t.Errorf("Should find 3 tags with Related")
+		}
+
+		// Replace
+		var tag5 = &Tag{Locale: "ZH", Value: "tag5"}
+		var tag6 = &Tag{Locale: "ZH", Value: "tag6"}
+		DB.Model(&blog2).Association("SharedTags").Replace(tag5, tag6)
+		var tags2 []Tag
+		DB.Model(&blog).Related(&tags2, "SharedTags")
+		if !compareTags(tags2, []string{"tag5", "tag6"}) {
+			t.Errorf("Should find 2 tags after Replace")
+		}
+
+		DB.Model(&blog2).Related(&tags2, "SharedTags")
+		if !compareTags(tags2, []string{"tag5", "tag6"}) {
+			t.Errorf("Should find 2 tags after Replace")
+		}
+
+		if DB.Model(&blog).Association("SharedTags").Count() != 2 {
+			t.Errorf("Blog should has three tags after Replace")
+		}
+
+		// Delete
+		DB.Model(&blog).Association("SharedTags").Delete(tag5)
+		var tags3 []Tag
+		DB.Model(&blog).Related(&tags3, "SharedTags")
+		if !compareTags(tags3, []string{"tag6"}) {
+			t.Errorf("Should find 1 tags after Delete")
+		}
+
+		if DB.Model(&blog).Association("SharedTags").Count() != 1 {
+			t.Errorf("Blog should has three tags after Delete")
+		}
+
+		DB.Model(&blog2).Association("SharedTags").Delete(tag3)
+		var tags4 []Tag
+		DB.Model(&blog).Related(&tags4, "SharedTags")
+		if !compareTags(tags4, []string{"tag6"}) {
+			t.Errorf("Tag should not be deleted when Delete with a unrelated tag")
+		}
+
+		// Clear
+		DB.Model(&blog2).Association("SharedTags").Clear()
+		if DB.Model(&blog).Association("SharedTags").Count() != 0 {
+			t.Errorf("All tags should be cleared")
+		}
+	}
+}