package main

import (
	"testing"

	"gorm.io/gorm"
	"gorm.io/gorm/clause"
)

type User struct {
	gorm.Model
	Name      string
	Age       uint
}

type UserMixin struct {
	UserID uint
	User User `gorm:"foreignKey:UserID"`
}

type Company struct {
	gorm.Model
	UserMixin
	ID   int
	Name string
}

func TestGORM(t *testing.T) {
	user := User{Name: "jinzhu"}

	DB.Create(&user)

	company := Company{Name: "jinzhu's company", UserMixin: UserMixin{UserID: user.ID}}

	DB.Create(&company)

	var companyResult Company
	if err := DB.Preload(clause.Associations).First(&companyResult, company.ID).Error; err != nil {
		t.Errorf("Failed, got error: %v", err)
	}
}