otter-space-sdk/pkg/models/orm_test.go

package models

import (
	"testing"
	"time"

	"gorm.io/gorm"
)

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[PostID]{
				ID:        PostID(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)
			}
		})
	}
}