63cebf07ab
- Updated file permissions in certificate_service_test.go and log_service_test.go to use octal notation. - Added a new doc.go file to document the services package. - Enhanced error handling in docker_service.go, log_service.go, notification_service.go, proxyhost_service.go, remoteserver_service.go, update_service.go, and uptime_service.go by logging errors when closing resources. - Improved log_service.go to simplify log file processing and deduplication. - Introduced CRUD tests for notification templates in notification_service_template_test.go. - Removed the obsolete python_compile_check.sh script. - Updated notification_service.go to improve template management functions. - Added tests for uptime service notifications in uptime_service_notification_test.go.
31 lines
951 B
Go
31 lines
951 B
Go
package handlers
|
|
|
|
import (
|
|
crand "crypto/rand"
|
|
"fmt"
|
|
"math/big"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// OpenTestDB creates a SQLite in-memory DB unique per test and applies
|
|
// a busy timeout and WAL journal mode to reduce SQLITE locking during parallel tests.
|
|
func OpenTestDB(t *testing.T) *gorm.DB {
|
|
t.Helper()
|
|
// Append a timestamp/random suffix to ensure uniqueness even across parallel runs
|
|
dsnName := strings.ReplaceAll(t.Name(), "/", "_")
|
|
// Use crypto/rand for suffix generation in tests to avoid static analysis warnings
|
|
n, _ := crand.Int(crand.Reader, big.NewInt(10000))
|
|
uniqueSuffix := fmt.Sprintf("%d%d", time.Now().UnixNano(), n.Int64())
|
|
dsn := fmt.Sprintf("file:%s_%s?mode=memory&cache=shared&_journal_mode=WAL&_busy_timeout=5000", dsnName, uniqueSuffix)
|
|
db, err := gorm.Open(sqlite.Open(dsn), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatalf("failed to open test db: %v", err)
|
|
}
|
|
return db
|
|
}
|