From 3628a21ece9279f4d18a71b42a99899618c373d8 Mon Sep 17 00:00:00 2001
From: Jinzhu <wosmvp@gmail.com>
Date: Sat, 13 Nov 2021 13:55:17 +0800
Subject: [PATCH] Update migrate_test.go

---
 tests/migrate_test.go | 44 ++++++++++++++++++++++++++++++++++---------
 1 file changed, 35 insertions(+), 9 deletions(-)

diff --git a/tests/migrate_test.go b/tests/migrate_test.go
index 670e0500..789a5e45 100644
--- a/tests/migrate_test.go
+++ b/tests/migrate_test.go
@@ -6,7 +6,6 @@ import (
 	"testing"
 	"time"
 
-	"github.com/juliangruber/go-intersect"
 	"gorm.io/gorm"
 	. "gorm.io/gorm/utils/tests"
 )
@@ -24,14 +23,22 @@ func TestMigrate(t *testing.T) {
 	if err := DB.AutoMigrate(allModels...); err != nil {
 		t.Fatalf("Failed to auto migrate, but got error %v", err)
 	}
-	tableList, tableErr := DB.Migrator().GetTables()
-	if tableErr != nil {
-		t.Fatalf("Failed to get database all tables, but got error %v", tableErr)
-	}
-	//get auto Migrator create tables and databases tables intersection
-	intersectList := intersect.Simple(tableList, []string{"users", "accounts", "pets", "companies", "toys", "languages"})
-	if len(intersectList) != len(allModels) {
-		t.Fatalf("Failed to get auto Migrator create tables and databases tables intersection nums  not eq allModels %d intersectList %d", len(allModels), len(intersectList))
+
+	if tables, err := DB.Migrator().GetTables(); err != nil {
+		t.Fatalf("Failed to get database all tables, but got error %v", err)
+	} else {
+		for _, t1 := range []string{"users", "accounts", "pets", "companies", "toys", "languages"} {
+			hasTable := false
+			for _, t2 := range tables {
+				if t2 == t1 {
+					hasTable = true
+					break
+				}
+			}
+			if !hasTable {
+				t.Fatalf("Failed to get table %v when GetTables", t1)
+			}
+		}
 	}
 
 	for _, m := range allModels {
@@ -63,6 +70,25 @@ func TestMigrate(t *testing.T) {
 	}
 }
 
+func TestAutoMigrateSelfReferential(t *testing.T) {
+	type MigratePerson struct {
+		ID        uint
+		Name      string
+		ManagerID *uint
+		Manager   *MigratePerson
+	}
+
+	DB.Migrator().DropTable(&MigratePerson{})
+
+	if err := DB.AutoMigrate(&MigratePerson{}); err != nil {
+		t.Fatalf("Failed to auto migrate, but got error %v", err)
+	}
+
+	if !DB.Migrator().HasConstraint("migrate_people", "fk_migrate_people_manager") {
+		t.Fatalf("Failed to find has one constraint between people and managers")
+	}
+}
+
 func TestSmartMigrateColumn(t *testing.T) {
 	fullSupported := map[string]bool{"mysql": true, "postgres": true}[DB.Dialector.Name()]