56 lines
999 B
Go
56 lines
999 B
Go
|
package models
|
||
|
|
|
||
|
|
import (
|
||
|
|
"gorm.io/gorm"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestBaseModel_BeforeCreate(t *testing.T) {
|
||
|
|
type fields struct {
|
||
|
|
ID string
|
||
|
|
CreatedAt time.Time
|
||
|
|
UpdatedAt time.Time
|
||
|
|
DeletedAt gorm.DeletedAt
|
||
|
|
}
|
||
|
|
type args struct {
|
||
|
|
db *gorm.DB
|
||
|
|
}
|
||
|
|
tests := []struct {
|
||
|
|
name string
|
||
|
|
fields fields
|
||
|
|
args args
|
||
|
|
wantErr bool
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "Test 1: Prefilled ID",
|
||
|
|
fields: fields{
|
||
|
|
ID: "1",
|
||
|
|
},
|
||
|
|
args: args{db: nil},
|
||
|
|
wantErr: false,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "Test 1: Autogenerate ID",
|
||
|
|
fields: fields{
|
||
|
|
ID: "",
|
||
|
|
},
|
||
|
|
args: args{db: nil},
|
||
|
|
wantErr: false,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for _, tt := range tests {
|
||
|
|
t.Run(tt.name, func(t *testing.T) {
|
||
|
|
base := &BaseModel{
|
||
|
|
ID: tt.fields.ID,
|
||
|
|
CreatedAt: tt.fields.CreatedAt,
|
||
|
|
UpdatedAt: tt.fields.UpdatedAt,
|
||
|
|
DeletedAt: tt.fields.DeletedAt,
|
||
|
|
}
|
||
|
|
if err := base.BeforeCreate(tt.args.db); (err != nil) != tt.wantErr {
|
||
|
|
t.Errorf("BeforeCreate() error = %v, wantErr %v", err, tt.wantErr)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|