99 lines
2.7 KiB
Go
99 lines
2.7 KiB
Go
package tests
|
|
|
|
import (
|
|
"acc-server-manager/local/model"
|
|
"acc-server-manager/local/utl/jwt"
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
// GenerateTestToken creates a JWT token for testing purposes
|
|
func GenerateTestToken() (string, error) {
|
|
// Create test user
|
|
user := &model.User{
|
|
ID: uuid.New(),
|
|
Username: "test_user",
|
|
RoleID: uuid.New(),
|
|
}
|
|
|
|
// Use the environment JWT_SECRET for consistency with middleware
|
|
testSecret := os.Getenv("JWT_SECRET")
|
|
if testSecret == "" {
|
|
// Fallback to a test secret if env var is not set
|
|
testSecret = "test-secret-that-is-at-least-32-bytes-long-for-security"
|
|
}
|
|
jwtHandler := jwt.NewJWTHandler(testSecret)
|
|
|
|
// Generate JWT token
|
|
token, err := jwtHandler.GenerateToken(user.ID.String())
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to generate test token: %w", err)
|
|
}
|
|
|
|
return token, nil
|
|
}
|
|
|
|
// MustGenerateTestToken generates a test token and panics if it fails
|
|
// This is useful for test setup where failing to generate a token is a fatal error
|
|
func MustGenerateTestToken() string {
|
|
token, err := GenerateTestToken()
|
|
if err != nil {
|
|
panic(fmt.Sprintf("Failed to generate test token: %v", err))
|
|
}
|
|
return token
|
|
}
|
|
|
|
// GenerateTestTokenWithExpiry creates a JWT token with a specific expiry time
|
|
func GenerateTestTokenWithExpiry(expiryTime time.Time) (string, error) {
|
|
// Use the environment JWT_SECRET for consistency with middleware
|
|
testSecret := os.Getenv("JWT_SECRET")
|
|
if testSecret == "" {
|
|
// Fallback to a test secret if env var is not set
|
|
testSecret = "test-secret-that-is-at-least-32-bytes-long-for-security"
|
|
}
|
|
jwtHandler := jwt.NewJWTHandler(testSecret)
|
|
|
|
// Create test user
|
|
user := &model.User{
|
|
ID: uuid.New(),
|
|
Username: "test_user",
|
|
RoleID: uuid.New(),
|
|
}
|
|
|
|
// Generate JWT token with custom expiry
|
|
token, err := jwtHandler.GenerateTokenWithExpiry(user, expiryTime)
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to generate test token with expiry: %w", err)
|
|
}
|
|
|
|
return token, nil
|
|
}
|
|
|
|
// AddAuthHeader adds a test auth token to the request headers
|
|
// This is a convenience method for tests that need to authenticate requests
|
|
func AddAuthHeader(headers map[string]string) (map[string]string, error) {
|
|
token, err := GenerateTestToken()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if headers == nil {
|
|
headers = make(map[string]string)
|
|
}
|
|
headers["Authorization"] = "Bearer " + token
|
|
|
|
return headers, nil
|
|
}
|
|
|
|
// MustAddAuthHeader adds a test auth token to the request headers and panics if it fails
|
|
func MustAddAuthHeader(headers map[string]string) map[string]string {
|
|
result, err := AddAuthHeader(headers)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("Failed to add auth header: %v", err))
|
|
}
|
|
return result
|
|
}
|