mirror of https://github.com/go-gorm/gorm.git
57 lines
1.4 KiB
Go
57 lines
1.4 KiB
Go
|
package gorm_test
|
||
|
|
|
||
|
|
import "testing"
|
||
|
|
|
||
|
|
type Cat struct {
|
||
|
|
Id int
|
||
|
|
Name string
|
||
|
|
Toy Toy `gorm:"polymorphic:Owner;"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type Dog struct {
|
||
|
|
Id int
|
||
|
|
Name string
|
||
|
|
Toys []Toy `gorm:"polymorphic:Owner;"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type Toy struct {
|
||
|
|
Id int
|
||
|
|
Name string
|
||
|
|
OwnerId int
|
||
|
|
OwnerType string
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPolymorphic(t *testing.T) {
|
||
|
|
DB.AutoMigrate(&Cat{})
|
||
|
|
DB.AutoMigrate(&Dog{})
|
||
|
|
DB.AutoMigrate(&Toy{})
|
||
|
|
|
||
|
|
cat := Cat{Name: "Mr. Bigglesworth", Toy: Toy{Name: "cat nip"}}
|
||
|
|
dog := Dog{Name: "Pluto", Toys: []Toy{Toy{Name: "orange ball"}, Toy{Name: "yellow ball"}}}
|
||
|
|
DB.Save(&cat).Save(&dog)
|
||
|
|
|
||
|
|
var catToys []Toy
|
||
|
|
if DB.Model(&cat).Related(&catToys, "Toy").RecordNotFound() {
|
||
|
|
t.Errorf("Did not find any has one polymorphic association")
|
||
|
|
} else if len(catToys) != 1 {
|
||
|
|
t.Errorf("Should have found only one polymorphic has one association")
|
||
|
|
} else if catToys[0].Name != cat.Toy.Name {
|
||
|
|
t.Errorf("Should have found the proper has one polymorphic association")
|
||
|
|
}
|
||
|
|
|
||
|
|
var dogToys []Toy
|
||
|
|
if DB.Model(&dog).Related(&dogToys, "Toys").RecordNotFound() {
|
||
|
|
t.Errorf("Did not find any polymorphic has many associations")
|
||
|
|
} else if len(dogToys) != len(dog.Toys) {
|
||
|
|
t.Errorf("Should have found all polymorphic has many associations")
|
||
|
|
}
|
||
|
|
|
||
|
|
if DB.Model(&cat).Association("Toy").Count() != 1 {
|
||
|
|
t.Errorf("Should return one polymorphic has one association")
|
||
|
|
}
|
||
|
|
|
||
|
|
if DB.Model(&dog).Association("Toys").Count() != 2 {
|
||
|
|
t.Errorf("Should return two polymorphic has many associations")
|
||
|
|
}
|
||
|
|
}
|