82 lines
1.7 KiB
Go
82 lines
1.7 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"git.eeqj.de/sneak/webhooker/internal/config"
|
|
"git.eeqj.de/sneak/webhooker/internal/globals"
|
|
"git.eeqj.de/sneak/webhooker/internal/logger"
|
|
"go.uber.org/fx/fxtest"
|
|
)
|
|
|
|
func TestDatabaseConnection(t *testing.T) {
|
|
// Set up test dependencies
|
|
lc := fxtest.NewLifecycle(t)
|
|
|
|
// Create globals
|
|
globals.Appname = "webhooker-test"
|
|
globals.Version = "test"
|
|
globals.Buildarch = "test"
|
|
|
|
g, err := globals.New(lc)
|
|
if err != nil {
|
|
t.Fatalf("Failed to create globals: %v", err)
|
|
}
|
|
|
|
// Create logger
|
|
l, err := logger.New(lc, logger.LoggerParams{Globals: g})
|
|
if err != nil {
|
|
t.Fatalf("Failed to create logger: %v", err)
|
|
}
|
|
|
|
// Create config
|
|
c, err := config.New(lc, config.ConfigParams{
|
|
Globals: g,
|
|
Logger: l,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("Failed to create config: %v", err)
|
|
}
|
|
|
|
// Set test database URL
|
|
c.DBURL = "file:test.db?cache=shared&mode=rwc"
|
|
|
|
// Create database
|
|
db, err := New(lc, DatabaseParams{
|
|
Config: c,
|
|
Logger: l,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("Failed to create database: %v", err)
|
|
}
|
|
|
|
// Start lifecycle (this will trigger the connection)
|
|
ctx := context.Background()
|
|
err = lc.Start(ctx)
|
|
if err != nil {
|
|
t.Fatalf("Failed to connect to database: %v", err)
|
|
}
|
|
defer func() {
|
|
if stopErr := lc.Stop(ctx); stopErr != nil {
|
|
t.Errorf("Failed to stop lifecycle: %v", stopErr)
|
|
}
|
|
}()
|
|
|
|
// Verify we can get the DB instance
|
|
if db.DB() == nil {
|
|
t.Error("Expected non-nil database connection")
|
|
}
|
|
|
|
// Test that we can perform a simple query
|
|
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)
|
|
}
|
|
}
|