clawbot
e50a79ced9
Some checks failed
check / check (push) Has been cancelled
Rewrites retention_days=0 to the RetentionForeverDays sentinel (365 * 1000) in Webhook.BeforeSave, so the GORM column default cannot win the race. The reaper skips retain-forever webhooks before building any query. Also bounds the reaper's cutoff arithmetic: a time.Duration is int64 nanoseconds, so day counts above MaxFiniteRetentionDays (106751) overflowed and wrapped the cutoff into the future, where created_at < cutoff matched every row and the sweep deleted everything. parseRetentionDays now rejects finite values above the ceiling, and retentionCutoff saturates so rows written by older versions cannot reach it either. Views render RetentionLabel() rather than the raw sentinel.
103 lines
1.9 KiB
Go
103 lines
1.9 KiB
Go
package database_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"go.uber.org/fx/fxtest"
|
|
"sneak.berlin/go/webhooker/internal/config"
|
|
"sneak.berlin/go/webhooker/internal/database"
|
|
"sneak.berlin/go/webhooker/internal/globals"
|
|
"sneak.berlin/go/webhooker/internal/logger"
|
|
)
|
|
|
|
const (
|
|
// testAppname is the Globals.Appname used in tests.
|
|
testAppname = "webhooker-test"
|
|
// testVersion is the Globals.Version used in tests.
|
|
testVersion = "test"
|
|
// testContentType is the event content type used in tests.
|
|
testContentType = "application/json"
|
|
// testWebhookName is the Webhook.Name used in tests.
|
|
testWebhookName = "test-webhook"
|
|
// testForeverLabel is Webhook.RetentionLabel for a retain-forever
|
|
// webhook.
|
|
testForeverLabel = "forever"
|
|
)
|
|
|
|
func setupTestDB(
|
|
t *testing.T,
|
|
) (*database.Database, *fxtest.Lifecycle) {
|
|
t.Helper()
|
|
|
|
lc := fxtest.NewLifecycle(t)
|
|
|
|
g := &globals.Globals{
|
|
Appname: testAppname,
|
|
Version: testVersion,
|
|
}
|
|
|
|
l, err := logger.New(
|
|
lc,
|
|
logger.LoggerParams{Globals: g},
|
|
)
|
|
if err != nil {
|
|
t.Fatalf("Failed to create logger: %v", err)
|
|
}
|
|
|
|
c := &config.Config{
|
|
DataDir: t.TempDir(),
|
|
Environment: "dev",
|
|
}
|
|
|
|
db, err := database.New(lc, database.DatabaseParams{
|
|
Config: c,
|
|
Logger: l,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("Failed to create database: %v", err)
|
|
}
|
|
|
|
return db, lc
|
|
}
|
|
|
|
func TestDatabaseConnection(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
db, lc := setupTestDB(t)
|
|
ctx := context.Background()
|
|
|
|
err := lc.Start(ctx)
|
|
if err != nil {
|
|
t.Fatalf("Failed to connect to database: %v", err)
|
|
}
|
|
|
|
defer func() {
|
|
stopErr := lc.Stop(ctx)
|
|
if stopErr != nil {
|
|
t.Errorf(
|
|
"Failed to stop lifecycle: %v",
|
|
stopErr,
|
|
)
|
|
}
|
|
}()
|
|
|
|
if db.DB() == nil {
|
|
t.Error("Expected non-nil database connection")
|
|
}
|
|
|
|
var result int
|
|
|
|
err = db.DB().Raw("SELECT 1").Scan(&result).Error
|
|
if err != nil {
|
|
t.Fatalf("Failed to execute test query: %v", err)
|
|
}
|
|
|
|
if result != 1 {
|
|
t.Errorf(
|
|
"Expected query result to be 1, got %d",
|
|
result,
|
|
)
|
|
}
|
|
}
|