diff --git a/server/api/test_helpers.go b/server/api/test_helpers.go index 25ea16e7d..520ec8bfd 100644 --- a/server/api/test_helpers.go +++ b/server/api/test_helpers.go @@ -1,25 +1,12 @@ 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() @@ -58,43 +45,3 @@ func NewTestConfig() *_config.Config { } } -// 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 -}