55 lines
1.3 KiB
Go
55 lines
1.3 KiB
Go
package postgres
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
_ "github.com/jackc/pgx/v5/stdlib"
|
|
"github.com/jmoiron/sqlx"
|
|
"github.com/saveinmed/backend-go/internal/config"
|
|
)
|
|
|
|
func TestDatabaseConnection(t *testing.T) {
|
|
if os.Getenv("SKIP_DB_TEST") != "" {
|
|
t.Skip("Skipping database tests")
|
|
}
|
|
|
|
// Simple .env loader for testing purposes
|
|
// Try loading from project root (3 levels up from this file)
|
|
if content, err := os.ReadFile("../../../.env"); err == nil {
|
|
for _, line := range strings.Split(string(content), "\n") {
|
|
parts := strings.SplitN(line, "=", 2)
|
|
if len(parts) == 2 {
|
|
os.Setenv(strings.TrimSpace(parts[0]), strings.TrimSpace(parts[1]))
|
|
}
|
|
}
|
|
}
|
|
|
|
cfg := config.Load()
|
|
|
|
// Create a context with timeout
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
db, err := sqlx.ConnectContext(ctx, "pgx", cfg.DatabaseURL)
|
|
if err != nil {
|
|
t.Fatalf("Failed to connect to database: %v", err)
|
|
}
|
|
defer db.Close()
|
|
|
|
if err := db.PingContext(ctx); err != nil {
|
|
t.Fatalf("Failed to ping database: %v", err)
|
|
}
|
|
|
|
var result int
|
|
if err := db.QueryRowContext(ctx, "SELECT 1").Scan(&result); err != nil {
|
|
t.Fatalf("Failed to execute query: %v", err)
|
|
}
|
|
|
|
if result != 1 {
|
|
t.Errorf("Expected 1, got %d", result)
|
|
}
|
|
}
|