mirror of
https://github.com/Mezeporta/Erupe.git
synced 2026-03-22 07:32:32 +01:00
101 lines
2.9 KiB
Go
101 lines
2.9 KiB
Go
package api
|
|
|
|
import (
|
|
"database/sql"
|
|
"testing"
|
|
|
|
_config "erupe-ce/config"
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
)
|
|
|
|
// MockDB provides a mock database for testing
|
|
type MockDB struct {
|
|
QueryRowFunc func(query string, args ...interface{}) *sql.Row
|
|
QueryFunc func(query string, args ...interface{}) (*sql.Rows, error)
|
|
ExecFunc func(query string, args ...interface{}) (sql.Result, error)
|
|
QueryRowContext func(ctx interface{}, query string, args ...interface{}) *sql.Row
|
|
GetContext func(ctx interface{}, dest interface{}, query string, args ...interface{}) error
|
|
SelectContext func(ctx interface{}, dest interface{}, query string, args ...interface{}) error
|
|
}
|
|
|
|
// NewTestLogger creates a logger for testing
|
|
func NewTestLogger(t *testing.T) *zap.Logger {
|
|
logger, err := zap.NewDevelopment()
|
|
if err != nil {
|
|
t.Fatalf("Failed to create test logger: %v", err)
|
|
}
|
|
return logger
|
|
}
|
|
|
|
// NewTestConfig creates a default test configuration
|
|
func NewTestConfig() *_config.Config {
|
|
return &_config.Config{
|
|
API: _config.API{
|
|
Port: 8000,
|
|
PatchServer: "http://localhost:8080",
|
|
Banners: []_config.APISignBanner{},
|
|
Messages: []_config.APISignMessage{},
|
|
Links: []_config.APISignLink{},
|
|
},
|
|
Screenshots: _config.ScreenshotsOptions{
|
|
Enabled: true,
|
|
OutputDir: "/tmp/screenshots",
|
|
UploadQuality: 85,
|
|
},
|
|
DebugOptions: _config.DebugOptions{
|
|
MaxLauncherHR: false,
|
|
},
|
|
GameplayOptions: _config.GameplayOptions{
|
|
MezFesSoloTickets: 100,
|
|
MezFesGroupTickets: 50,
|
|
MezFesDuration: 604800, // 1 week
|
|
MezFesSwitchMinigame: false,
|
|
},
|
|
LoginNotices: []string{"Welcome to Erupe!"},
|
|
HideLoginNotice: false,
|
|
}
|
|
}
|
|
|
|
// NewTestAPIServer creates an API server for testing with a real database
|
|
func NewTestAPIServer(t *testing.T, db *sqlx.DB) *APIServer {
|
|
logger := NewTestLogger(t)
|
|
cfg := NewTestConfig()
|
|
config := &Config{
|
|
Logger: logger,
|
|
DB: db,
|
|
ErupeConfig: cfg,
|
|
}
|
|
return NewAPIServer(config)
|
|
}
|
|
|
|
// CleanupTestData removes test data from the database
|
|
func CleanupTestData(t *testing.T, db *sqlx.DB, userID uint32) {
|
|
// Delete characters associated with the user
|
|
_, err := db.Exec("DELETE FROM characters WHERE user_id = $1", userID)
|
|
if err != nil {
|
|
t.Logf("Error cleaning up characters: %v", err)
|
|
}
|
|
|
|
// Delete sign sessions for the user
|
|
_, err = db.Exec("DELETE FROM sign_sessions WHERE user_id = $1", userID)
|
|
if err != nil {
|
|
t.Logf("Error cleaning up sign_sessions: %v", err)
|
|
}
|
|
|
|
// Delete the user
|
|
_, err = db.Exec("DELETE FROM users WHERE id = $1", userID)
|
|
if err != nil {
|
|
t.Logf("Error cleaning up users: %v", err)
|
|
}
|
|
}
|
|
|
|
// GetTestDBConnection returns a test database connection (requires database to be running)
|
|
func GetTestDBConnection(t *testing.T) *sqlx.DB {
|
|
// This function would need to connect to a test database
|
|
// For now, it's a placeholder that returns nil
|
|
// In practice, you'd use a test database container or mock
|
|
return nil
|
|
}
|