github.com/goravel/framework@v1.13.9/database/gorm/test_models.go

package gorm

import (
	"errors"

	ormcontract "github.com/goravel/framework/contracts/database/orm"
	"github.com/goravel/framework/database/orm"
)

type contextKey int

const testContextKey contextKey = 0

type User struct {
	orm.Model
	orm.SoftDeletes
	Name    string
	Avatar  string
	Address *Address
	Books   []*Book
	House   *House   `gorm:"polymorphic:Houseable"`
	Phones  []*Phone `gorm:"polymorphic:Phoneable"`
	Roles   []*Role  `gorm:"many2many:role_user"`
	age     int
}

func (u *User) DispatchesEvents() map[ormcontract.EventType]func(ormcontract.Event) error {
	return map[ormcontract.EventType]func(ormcontract.Event) error{
		ormcontract.EventCreating: func(event ormcontract.Event) error {
			name := event.GetAttribute("name")
			if name != nil {
				if name.(string) == "event_creating_name" {
					event.SetAttribute("avatar", "event_creating_avatar")
				}
				if name.(string) == "event_creating_FirstOrCreate_name" {
					event.SetAttribute("avatar", "event_creating_FirstOrCreate_avatar")
				}
				if name.(string) == "event_creating_IsDirty_name" {
					if event.IsDirty("name") {
						event.SetAttribute("avatar", "event_creating_IsDirty_avatar")
					}
				}
				if name.(string) == "event_context" {
					val := event.Context().Value(testContextKey)
					event.SetAttribute("avatar", val.(string))
				}
				if name.(string) == "event_query" {
					_ = event.Query().Create(&User{Name: "event_query1"})
				}
			}

			return nil
		},
		ormcontract.EventCreated: func(event ormcontract.Event) error {
			name := event.GetAttribute("name")
			if name != nil {
				if name.(string) == "event_created_name" {
					event.SetAttribute("avatar", "event_created_avatar")
				}
				if name.(string) == "event_created_FirstOrCreate_name" {
					event.SetAttribute("avatar", "event_created_FirstOrCreate_avatar")
				}
			}

			return nil
		},
		ormcontract.EventSaving: func(event ormcontract.Event) error {
			name := event.GetAttribute("name")
			if name != nil {
				if name.(string) == "event_saving_create_name" {
					event.SetAttribute("avatar", "event_saving_create_avatar")
				}
				if name.(string) == "event_saving_save_name" {
					event.SetAttribute("avatar", "event_saving_save_avatar")
				}
				if name.(string) == "event_saving_FirstOrCreate_name" {
					event.SetAttribute("avatar", "event_saving_FirstOrCreate_avatar")
				}
				if name.(string) == "event_save_without_name" {
					event.SetAttribute("avatar", "event_save_without_avatar")
				}
				if name.(string) == "event_save_quietly_name" {
					event.SetAttribute("avatar", "event_save_quietly_avatar")
				}
				if name.(string) == "event_saving_IsDirty_name" {
					if event.IsDirty("name") {
						event.SetAttribute("avatar", "event_saving_IsDirty_avatar")
					}
				}
			}

			avatar := event.GetAttribute("avatar")
			if avatar != nil && avatar.(string) == "event_saving_single_update_avatar" {
				event.SetAttribute("avatar", "event_saving_single_update_avatar1")
			}

			return nil
		},
		ormcontract.EventSaved: func(event ormcontract.Event) error {
			name := event.GetAttribute("name")
			if name != nil {
				if name.(string) == "event_saved_create_name" {
					event.SetAttribute("avatar", "event_saved_create_avatar")
				}
				if name.(string) == "event_saved_save_name" {
					event.SetAttribute("avatar", "event_saved_save_avatar")
				}
				if name.(string) == "event_saved_FirstOrCreate_name" {
					event.SetAttribute("avatar", "event_saved_FirstOrCreate_avatar")
				}
				if name.(string) == "event_save_without_name" {
					event.SetAttribute("avatar", "event_saved_without_avatar")
				}
				if name.(string) == "event_save_quietly_name" {
					event.SetAttribute("avatar", "event_saved_quietly_avatar")
				}
			}

			avatar := event.GetAttribute("avatar")
			if avatar != nil && avatar.(string) == "event_saved_map_update_avatar" {
				event.SetAttribute("avatar", "event_saved_map_update_avatar1")
			}

			return nil
		},
		ormcontract.EventUpdating: func(event ormcontract.Event) error {
			name := event.GetAttribute("name")
			if name != nil {
				if name.(string) == "event_updating_create_name" {
					event.SetAttribute("avatar", "event_updating_create_avatar")
				}
				if name.(string) == "event_updating_save_name" {
					event.SetAttribute("avatar", "event_updating_save_avatar")
				}
				if name.(string) == "event_updating_single_update_IsDirty_name1" {
					if event.IsDirty("name") {
						name := event.GetAttribute("name")
						if name != "event_updating_single_update_IsDirty_name1" {
							return errors.New("error")
						}

						event.SetAttribute("avatar", "event_updating_single_update_IsDirty_avatar")
					}
				}
				if name.(string) == "event_updating_map_update_IsDirty_name1" {
					if event.IsDirty("name") {
						name := event.GetAttribute("name")
						if name != "event_updating_map_update_IsDirty_name1" {
							return errors.New("error")
						}

						event.SetAttribute("avatar", "event_updating_map_update_IsDirty_avatar")
					}
				}
				if name.(string) == "event_updating_model_update_IsDirty_name1" {
					if event.IsDirty("name") {
						name := event.GetAttribute("name")
						if name != "event_updating_model_update_IsDirty_name1" {
							return errors.New("error")
						}
						event.SetAttribute("avatar", "event_updating_model_update_IsDirty_avatar")
					}
				}
			}

			avatar := event.GetAttribute("avatar")
			if avatar != nil {
				if avatar.(string) == "event_updating_save_avatar" {
					event.SetAttribute("avatar", "event_updating_save_avatar1")
				}
				if avatar.(string) == "event_updating_model_update_avatar" {
					event.SetAttribute("avatar", "event_updating_model_update_avatar1")
				}
			}

			return nil
		},
		ormcontract.EventUpdated: func(event ormcontract.Event) error {
			name := event.GetAttribute("name")
			if name != nil {
				if name.(string) == "event_updated_create_name" {
					event.SetAttribute("avatar", "event_updated_create_avatar")
				}
				if name.(string) == "event_updated_save_name" {
					event.SetAttribute("avatar", "event_updated_save_avatar")
				}
			}

			avatar := event.GetAttribute("avatar")
			if avatar != nil {
				if avatar.(string) == "event_updated_save_avatar" {
					event.SetAttribute("avatar", "event_updated_save_avatar1")
				}
				if avatar.(string) == "event_updated_model_update_avatar" {
					event.SetAttribute("avatar", "event_updated_model_update_avatar1")
				}
			}

			return nil
		},
		ormcontract.EventDeleting: func(event ormcontract.Event) error {
			name := event.GetAttribute("name")
			if name != nil && name.(string) == "event_deleting_name" {
				return errors.New("deleting error")
			}

			return nil
		},
		ormcontract.EventDeleted: func(event ormcontract.Event) error {
			name := event.GetAttribute("name")
			if name != nil && name.(string) == "event_deleted_name" {
				return errors.New("deleted error")
			}

			return nil
		},
		ormcontract.EventForceDeleting: func(event ormcontract.Event) error {
			name := event.GetAttribute("name")
			if name != nil && name.(string) == "event_force_deleting_name" {
				return errors.New("force deleting error")
			}

			return nil
		},
		ormcontract.EventForceDeleted: func(event ormcontract.Event) error {
			name := event.GetAttribute("name")
			if name != nil && name.(string) == "event_force_deleted_name" {
				return errors.New("force deleted error")
			}

			return nil
		},
		ormcontract.EventRetrieved: func(event ormcontract.Event) error {
			name := event.GetAttribute("name")
			if name != nil && name.(string) == "event_retrieved_name" {
				event.SetAttribute("name", "event_retrieved_name1")
			}

			return nil
		},
	}
}

type Role struct {
	orm.Model
	Name  string
	Users []*User `gorm:"many2many:role_user"`
}

type Address struct {
	orm.Model
	UserID   uint
	Name     string
	Province string
	User     *User
}

type Book struct {
	orm.Model
	UserID uint
	Name   string
	User   *User
	Author *Author
}

type Author struct {
	orm.Model
	BookID uint
	Name   string
}

type House struct {
	orm.Model
	Name          string
	HouseableID   uint
	HouseableType string
}

func (h *House) Factory() string {
	return "house"
}

type Phone struct {
	orm.Model
	Name          string
	PhoneableID   uint
	PhoneableType string
}

type Product struct {
	orm.Model
	orm.SoftDeletes
	Name string
}

func (p *Product) Connection() string {
	return "postgresql"
}

type Review struct {
	orm.Model
	orm.SoftDeletes
	Body string
}

func (r *Review) Connection() string {
	return ""
}

type People struct {
	orm.Model
	orm.SoftDeletes
	Body string
}

func (p *People) Connection() string {
	return "dummy"
}

type Person struct {
	orm.Model
	orm.SoftDeletes
	Name string
}

func (p *Person) Connection() string {
	return "dummy"
}

type Box struct {
	orm.Model
	orm.SoftDeletes
	Name string
}

func (p *Box) Connection() string {
	return "mysql"
}