Charon/backend/internal/models/notification_test.go

package models

import (
	"testing"

	"github.com/stretchr/testify/assert"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
)

func TestNotification_BeforeCreate(t *testing.T) {
	db, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{})
	assert.NoError(t, err)
	_ = db.AutoMigrate(&Notification{})

	// Case 1: ID is empty, should be generated
	n1 := &Notification{Title: "Test", Message: "Test Message"}
	err = db.Create(n1).Error
	assert.NoError(t, err)
	assert.NotEmpty(t, n1.ID)

	// Case 2: ID is provided, should be kept
	id := "123e4567-e89b-12d3-a456-426614174000"
	n2 := &Notification{ID: id, Title: "Test 2", Message: "Test Message 2"}
	err = db.Create(n2).Error
	assert.NoError(t, err)
	assert.Equal(t, id, n2.ID)
}

func TestNotificationConfig_BeforeCreate(t *testing.T) {
	db, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{})
	assert.NoError(t, err)
	_ = db.AutoMigrate(&NotificationConfig{})

	// Case 1: ID is empty, should be generated
	nc1 := &NotificationConfig{Enabled: true, MinLogLevel: "error"}
	err = db.Create(nc1).Error
	assert.NoError(t, err)
	assert.NotEmpty(t, nc1.ID)

	// Case 2: ID is provided, should be kept
	id := "custom-config-id"
	nc2 := &NotificationConfig{ID: id, Enabled: false, MinLogLevel: "warn"}
	err = db.Create(nc2).Error
	assert.NoError(t, err)
	assert.Equal(t, id, nc2.ID)
}