popertots
44cac94f64
- Fix TestRequireRoleMiddleware by chaining RequireAuth before RequireRole (role context requirement) - Remove unused 'strings' import from testdb.go - Create crussell_test database in Docker setup - Tests now properly initialize authentication context for role-based tests Result: handlers test suite passes (13/13 tests) Remaining failures in admin/auth/bookings/portfolio/scheduling/services/user packages need further investigation (environment setup, database constraints, endpoint initialization)
144 lines
4.4 KiB
Go
144 lines
4.4 KiB
Go
//go:build test
|
|
// +build test
|
|
|
|
package fixtures
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
"golang.org/x/crypto/bcrypt"
|
|
)
|
|
|
|
func CreateTestAdminUser(pool *pgxpool.Pool) (string, error) {
|
|
return createTestUser(pool, "Admin", "User", "admin@test.com", "admin")
|
|
}
|
|
|
|
func CreateTestUser(pool *pgxpool.Pool) (string, error) {
|
|
return createTestUser(pool, "Test", "User", "user@test.com", "verified_email")
|
|
}
|
|
|
|
func createTestUser(pool *pgxpool.Pool, firstName, lastName, email, role string) (string, error) {
|
|
passwordHash, err := bcrypt.GenerateFromPassword([]byte("testpassword123"), bcrypt.DefaultCost)
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to hash password: %w", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
var userID string
|
|
err = pool.QueryRow(ctx, `
|
|
INSERT INTO users (n_first_name, n_last_name, email, password_hash, account_role, account_type)
|
|
VALUES ($1, $2, $3, $4, $5, 'email')
|
|
RETURNING id
|
|
`, firstName, lastName, email, string(passwordHash), role).Scan(&userID)
|
|
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to create user: %w", err)
|
|
}
|
|
|
|
return userID, nil
|
|
}
|
|
|
|
func CreateTestService(pool *pgxpool.Pool) (string, error) {
|
|
ctx := context.Background()
|
|
var serviceID string
|
|
err := pool.QueryRow(ctx, `
|
|
INSERT INTO services (name, description, price, duration_minutes, is_active, patch_test_duration_hours, minimum_age_required)
|
|
VALUES ($1, $2, $3, $4, $5, $6, $7)
|
|
RETURNING id
|
|
`, "Test Service", "A test service for unit tests", 50.00, 60, true, 0, 16).Scan(&serviceID)
|
|
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to create service: %w", err)
|
|
}
|
|
|
|
return serviceID, nil
|
|
}
|
|
|
|
func CreateTestServiceWithPatchTest(pool *pgxpool.Pool) (string, error) {
|
|
ctx := context.Background()
|
|
var serviceID string
|
|
err := pool.QueryRow(ctx, `
|
|
INSERT INTO services (name, description, price, duration_minutes, is_active, patch_test_duration_hours, minimum_age_required)
|
|
VALUES ($1, $2, $3, $4, $5, $6, $7)
|
|
RETURNING id
|
|
`, "Test Patch Test Service", "A test service requiring patch test", 75.00, 90, true, 48, 18).Scan(&serviceID)
|
|
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to create service: %w", err)
|
|
}
|
|
|
|
return serviceID, nil
|
|
}
|
|
|
|
func CreateTestBooking(pool *pgxpool.Pool, userID, serviceID string) (string, error) {
|
|
ctx := context.Background()
|
|
var bookingID string
|
|
err := pool.QueryRow(ctx, `
|
|
INSERT INTO bookings (user_id, start_time, status, notes)
|
|
VALUES ($1, $2, $3, $4)
|
|
RETURNING id
|
|
`, userID, "2099-12-31 10:00:00+00", "pending", "Test booking").Scan(&bookingID)
|
|
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to create booking: %w", err)
|
|
}
|
|
|
|
_, err = pool.Exec(ctx, `
|
|
INSERT INTO booking_services (booking_id, service_id)
|
|
VALUES ($1, $2)
|
|
`, bookingID, serviceID)
|
|
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to link service to booking: %w", err)
|
|
}
|
|
|
|
return bookingID, nil
|
|
}
|
|
|
|
func CreateTestVerifiedUser(pool *pgxpool.Pool) (string, error) {
|
|
return createTestUser(pool, "Verified", "User", "verified@test.com", "verified_email")
|
|
}
|
|
|
|
func CreateTestUnverifiedUser(pool *pgxpool.Pool) (string, error) {
|
|
return createTestUser(pool, "Unverified", "User", "unverified@test.com", "unverified_email")
|
|
}
|
|
|
|
func CreateTestGuestUser(pool *pgxpool.Pool) (string, error) {
|
|
return createTestUser(pool, "Guest", "User", "guest@test.com", "guest")
|
|
}
|
|
|
|
func DeleteUser(pool *pgxpool.Pool, userID string) error {
|
|
ctx := context.Background()
|
|
_, err := pool.Exec(ctx, "DELETE FROM users WHERE id = $1", userID)
|
|
return err
|
|
}
|
|
|
|
func DeleteService(pool *pgxpool.Pool, serviceID string) error {
|
|
ctx := context.Background()
|
|
_, err := pool.Exec(ctx, "DELETE FROM services WHERE id = $1", serviceID)
|
|
return err
|
|
}
|
|
|
|
func DeleteBooking(pool *pgxpool.Pool, bookingID string) error {
|
|
ctx := context.Background()
|
|
_, err := pool.Exec(ctx, "DELETE FROM bookings WHERE id = $1", bookingID)
|
|
return err
|
|
}
|
|
|
|
// SafeDeleteUser wraps DeleteUser and returns error (for tests that care about cleanup failure)
|
|
func SafeDeleteUser(db *pgxpool.Pool, userID string) error {
|
|
return DeleteUser(db, userID)
|
|
}
|
|
|
|
// SafeDeleteService wraps DeleteService and returns error (for tests that care about cleanup failure)
|
|
func SafeDeleteService(db *pgxpool.Pool, serviceID string) error {
|
|
return DeleteService(db, serviceID)
|
|
}
|
|
|
|
// SafeDeleteBooking wraps DeleteBooking and returns error (for tests that care about cleanup failure)
|
|
func SafeDeleteBooking(db *pgxpool.Pool, bookingID string) error {
|
|
return DeleteBooking(db, bookingID)
|
|
}
|