123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366 |
- package gorm_test
- import (
- "reflect"
- "sort"
- "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 Hamster struct {
- Id int
- Name string
- PreferredToy Toy `gorm:"polymorphic:Owner;polymorphic_value:hamster_preferred"`
- OtherToy Toy `gorm:"polymorphic:Owner;polymorphic_value:hamster_other"`
- }
- type Toy struct {
- Id int
- Name string
- OwnerId int
- OwnerType string
- }
- var compareToys = func(toys []Toy, contents []string) bool {
- var toyContents []string
- for _, toy := range toys {
- toyContents = append(toyContents, toy.Name)
- }
- sort.Strings(toyContents)
- sort.Strings(contents)
- return reflect.DeepEqual(toyContents, contents)
- }
- func TestPolymorphic(t *testing.T) {
- cat := Cat{Name: "Mr. Bigglesworth", Toy: Toy{Name: "cat toy"}}
- dog := Dog{Name: "Pluto", Toys: []Toy{{Name: "dog toy 1"}, {Name: "dog toy 2"}}}
- DB.Save(&cat).Save(&dog)
- if DB.Model(&cat).Association("Toy").Count() != 1 {
- t.Errorf("Cat's toys count should be 1")
- }
- if DB.Model(&dog).Association("Toys").Count() != 2 {
- t.Errorf("Dog's toys count should be 2")
- }
- // Query
- 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")
- }
- var catToy Toy
- DB.Model(&cat).Association("Toy").Find(&catToy)
- if catToy.Name != cat.Toy.Name {
- t.Errorf("Should find has one polymorphic association")
- }
- var dogToys1 []Toy
- DB.Model(&dog).Association("Toys").Find(&dogToys1)
- if !compareToys(dogToys1, []string{"dog toy 1", "dog toy 2"}) {
- t.Errorf("Should find has many polymorphic association")
- }
- // Append
- DB.Model(&cat).Association("Toy").Append(&Toy{
- Name: "cat toy 2",
- })
- var catToy2 Toy
- DB.Model(&cat).Association("Toy").Find(&catToy2)
- if catToy2.Name != "cat toy 2" {
- t.Errorf("Should update has one polymorphic association with Append")
- }
- if DB.Model(&cat).Association("Toy").Count() != 1 {
- t.Errorf("Cat's toys count should be 1 after Append")
- }
- if DB.Model(&dog).Association("Toys").Count() != 2 {
- t.Errorf("Should return two polymorphic has many associations")
- }
- DB.Model(&dog).Association("Toys").Append(&Toy{
- Name: "dog toy 3",
- })
- var dogToys2 []Toy
- DB.Model(&dog).Association("Toys").Find(&dogToys2)
- if !compareToys(dogToys2, []string{"dog toy 1", "dog toy 2", "dog toy 3"}) {
- t.Errorf("Dog's toys should be updated with Append")
- }
- if DB.Model(&dog).Association("Toys").Count() != 3 {
- t.Errorf("Should return three polymorphic has many associations")
- }
- // Replace
- DB.Model(&cat).Association("Toy").Replace(&Toy{
- Name: "cat toy 3",
- })
- var catToy3 Toy
- DB.Model(&cat).Association("Toy").Find(&catToy3)
- if catToy3.Name != "cat toy 3" {
- t.Errorf("Should update has one polymorphic association with Replace")
- }
- if DB.Model(&cat).Association("Toy").Count() != 1 {
- t.Errorf("Cat's toys count should be 1 after Replace")
- }
- if DB.Model(&dog).Association("Toys").Count() != 3 {
- t.Errorf("Should return three polymorphic has many associations")
- }
- DB.Model(&dog).Association("Toys").Replace(&Toy{
- Name: "dog toy 4",
- }, []Toy{
- {Name: "dog toy 5"}, {Name: "dog toy 6"}, {Name: "dog toy 7"},
- })
- var dogToys3 []Toy
- DB.Model(&dog).Association("Toys").Find(&dogToys3)
- if !compareToys(dogToys3, []string{"dog toy 4", "dog toy 5", "dog toy 6", "dog toy 7"}) {
- t.Errorf("Dog's toys should be updated with Replace")
- }
- if DB.Model(&dog).Association("Toys").Count() != 4 {
- t.Errorf("Should return three polymorphic has many associations")
- }
- // Delete
- DB.Model(&cat).Association("Toy").Delete(&catToy2)
- var catToy4 Toy
- DB.Model(&cat).Association("Toy").Find(&catToy4)
- if catToy4.Name != "cat toy 3" {
- t.Errorf("Should not update has one polymorphic association when Delete a unrelated Toy")
- }
- if DB.Model(&cat).Association("Toy").Count() != 1 {
- t.Errorf("Cat's toys count should be 1")
- }
- if DB.Model(&dog).Association("Toys").Count() != 4 {
- t.Errorf("Dog's toys count should be 4")
- }
- DB.Model(&cat).Association("Toy").Delete(&catToy3)
- if !DB.Model(&cat).Related(&Toy{}, "Toy").RecordNotFound() {
- t.Errorf("Toy should be deleted with Delete")
- }
- if DB.Model(&cat).Association("Toy").Count() != 0 {
- t.Errorf("Cat's toys count should be 0 after Delete")
- }
- if DB.Model(&dog).Association("Toys").Count() != 4 {
- t.Errorf("Dog's toys count should not be changed when delete cat's toy")
- }
- DB.Model(&dog).Association("Toys").Delete(&dogToys2)
- if DB.Model(&dog).Association("Toys").Count() != 4 {
- t.Errorf("Dog's toys count should not be changed when delete unrelated toys")
- }
- DB.Model(&dog).Association("Toys").Delete(&dogToys3)
- if DB.Model(&dog).Association("Toys").Count() != 0 {
- t.Errorf("Dog's toys count should be deleted with Delete")
- }
- // Clear
- DB.Model(&cat).Association("Toy").Append(&Toy{
- Name: "cat toy 2",
- })
- if DB.Model(&cat).Association("Toy").Count() != 1 {
- t.Errorf("Cat's toys should be added with Append")
- }
- DB.Model(&cat).Association("Toy").Clear()
- if DB.Model(&cat).Association("Toy").Count() != 0 {
- t.Errorf("Cat's toys should be cleared with Clear")
- }
- DB.Model(&dog).Association("Toys").Append(&Toy{
- Name: "dog toy 8",
- })
- if DB.Model(&dog).Association("Toys").Count() != 1 {
- t.Errorf("Dog's toys should be added with Append")
- }
- DB.Model(&dog).Association("Toys").Clear()
- if DB.Model(&dog).Association("Toys").Count() != 0 {
- t.Errorf("Dog's toys should be cleared with Clear")
- }
- }
- func TestNamedPolymorphic(t *testing.T) {
- hamster := Hamster{Name: "Mr. Hammond", PreferredToy: Toy{Name: "bike"}, OtherToy: Toy{Name: "treadmill"}}
- DB.Save(&hamster)
- hamster2 := Hamster{}
- DB.Preload("PreferredToy").Preload("OtherToy").Find(&hamster2, hamster.Id)
- if hamster2.PreferredToy.Id != hamster.PreferredToy.Id || hamster2.PreferredToy.Name != hamster.PreferredToy.Name {
- t.Errorf("Hamster's preferred toy couldn't be preloaded")
- }
- if hamster2.OtherToy.Id != hamster.OtherToy.Id || hamster2.OtherToy.Name != hamster.OtherToy.Name {
- t.Errorf("Hamster's other toy couldn't be preloaded")
- }
- // clear to omit Toy.Id in count
- hamster2.PreferredToy = Toy{}
- hamster2.OtherToy = Toy{}
- if DB.Model(&hamster2).Association("PreferredToy").Count() != 1 {
- t.Errorf("Hamster's preferred toy count should be 1")
- }
- if DB.Model(&hamster2).Association("OtherToy").Count() != 1 {
- t.Errorf("Hamster's other toy count should be 1")
- }
- // Query
- var hamsterToys []Toy
- if DB.Model(&hamster).Related(&hamsterToys, "PreferredToy").RecordNotFound() {
- t.Errorf("Did not find any has one polymorphic association")
- } else if len(hamsterToys) != 1 {
- t.Errorf("Should have found only one polymorphic has one association")
- } else if hamsterToys[0].Name != hamster.PreferredToy.Name {
- t.Errorf("Should have found the proper has one polymorphic association")
- }
- if DB.Model(&hamster).Related(&hamsterToys, "OtherToy").RecordNotFound() {
- t.Errorf("Did not find any has one polymorphic association")
- } else if len(hamsterToys) != 1 {
- t.Errorf("Should have found only one polymorphic has one association")
- } else if hamsterToys[0].Name != hamster.OtherToy.Name {
- t.Errorf("Should have found the proper has one polymorphic association")
- }
- hamsterToy := Toy{}
- DB.Model(&hamster).Association("PreferredToy").Find(&hamsterToy)
- if hamsterToy.Name != hamster.PreferredToy.Name {
- t.Errorf("Should find has one polymorphic association")
- }
- hamsterToy = Toy{}
- DB.Model(&hamster).Association("OtherToy").Find(&hamsterToy)
- if hamsterToy.Name != hamster.OtherToy.Name {
- t.Errorf("Should find has one polymorphic association")
- }
- // Append
- DB.Model(&hamster).Association("PreferredToy").Append(&Toy{
- Name: "bike 2",
- })
- DB.Model(&hamster).Association("OtherToy").Append(&Toy{
- Name: "treadmill 2",
- })
- hamsterToy = Toy{}
- DB.Model(&hamster).Association("PreferredToy").Find(&hamsterToy)
- if hamsterToy.Name != "bike 2" {
- t.Errorf("Should update has one polymorphic association with Append")
- }
- hamsterToy = Toy{}
- DB.Model(&hamster).Association("OtherToy").Find(&hamsterToy)
- if hamsterToy.Name != "treadmill 2" {
- t.Errorf("Should update has one polymorphic association with Append")
- }
- if DB.Model(&hamster2).Association("PreferredToy").Count() != 1 {
- t.Errorf("Hamster's toys count should be 1 after Append")
- }
- if DB.Model(&hamster2).Association("OtherToy").Count() != 1 {
- t.Errorf("Hamster's toys count should be 1 after Append")
- }
- // Replace
- DB.Model(&hamster).Association("PreferredToy").Replace(&Toy{
- Name: "bike 3",
- })
- DB.Model(&hamster).Association("OtherToy").Replace(&Toy{
- Name: "treadmill 3",
- })
- hamsterToy = Toy{}
- DB.Model(&hamster).Association("PreferredToy").Find(&hamsterToy)
- if hamsterToy.Name != "bike 3" {
- t.Errorf("Should update has one polymorphic association with Replace")
- }
- hamsterToy = Toy{}
- DB.Model(&hamster).Association("OtherToy").Find(&hamsterToy)
- if hamsterToy.Name != "treadmill 3" {
- t.Errorf("Should update has one polymorphic association with Replace")
- }
- if DB.Model(&hamster2).Association("PreferredToy").Count() != 1 {
- t.Errorf("hamster's toys count should be 1 after Replace")
- }
- if DB.Model(&hamster2).Association("OtherToy").Count() != 1 {
- t.Errorf("hamster's toys count should be 1 after Replace")
- }
- // Clear
- DB.Model(&hamster).Association("PreferredToy").Append(&Toy{
- Name: "bike 2",
- })
- DB.Model(&hamster).Association("OtherToy").Append(&Toy{
- Name: "treadmill 2",
- })
- if DB.Model(&hamster).Association("PreferredToy").Count() != 1 {
- t.Errorf("Hamster's toys should be added with Append")
- }
- if DB.Model(&hamster).Association("OtherToy").Count() != 1 {
- t.Errorf("Hamster's toys should be added with Append")
- }
- DB.Model(&hamster).Association("PreferredToy").Clear()
- if DB.Model(&hamster2).Association("PreferredToy").Count() != 0 {
- t.Errorf("Hamster's preferred toy should be cleared with Clear")
- }
- if DB.Model(&hamster2).Association("OtherToy").Count() != 1 {
- t.Errorf("Hamster's other toy should be still available")
- }
- DB.Model(&hamster).Association("OtherToy").Clear()
- if DB.Model(&hamster).Association("OtherToy").Count() != 0 {
- t.Errorf("Hamster's other toy should be cleared with Clear")
- }
- }
|