mirror of
https://github.com/Mezeporta/Erupe.git
synced 2026-03-22 15:43:49 +01:00
ba7ec122f8
An MMO server without multiplayer defeats the purpose. PostgreSQL is the right choice and Docker Compose already solves the setup pain. This reverts the common/db wrapper, SQLite schema, config Driver field, modernc.org/sqlite dependency, and all repo type changes while keeping the dashboard, wizard, and CI improvements from the previous commit.
111 lines
3.0 KiB
Go
111 lines
3.0 KiB
Go
package channelserver
|
|
|
|
import (
|
|
"database/sql"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
)
|
|
|
|
func setupUserRepo(t *testing.T) (*UserRepository, *sqlx.DB, uint32) {
|
|
t.Helper()
|
|
db := SetupTestDB(t)
|
|
userID := CreateTestUser(t, db, "user_repo_test")
|
|
repo := NewUserRepository(db)
|
|
t.Cleanup(func() { TeardownTestDB(t, db) })
|
|
return repo, db, userID
|
|
}
|
|
|
|
func TestBanUserPermanent(t *testing.T) {
|
|
repo, db, userID := setupUserRepo(t)
|
|
|
|
if err := repo.BanUser(userID, nil); err != nil {
|
|
t.Fatalf("BanUser (permanent) failed: %v", err)
|
|
}
|
|
|
|
// Verify ban exists with NULL expires
|
|
var gotUserID uint32
|
|
var expires sql.NullTime
|
|
err := db.QueryRow("SELECT user_id, expires FROM bans WHERE user_id=$1", userID).Scan(&gotUserID, &expires)
|
|
if err != nil {
|
|
t.Fatalf("Verification query failed: %v", err)
|
|
}
|
|
if gotUserID != userID {
|
|
t.Errorf("Expected user_id %d, got: %d", userID, gotUserID)
|
|
}
|
|
if expires.Valid {
|
|
t.Errorf("Expected NULL expires for permanent ban, got: %v", expires.Time)
|
|
}
|
|
}
|
|
|
|
func TestBanUserTemporary(t *testing.T) {
|
|
repo, db, userID := setupUserRepo(t)
|
|
|
|
expiry := time.Now().Add(24 * time.Hour).Truncate(time.Microsecond)
|
|
if err := repo.BanUser(userID, &expiry); err != nil {
|
|
t.Fatalf("BanUser (temporary) failed: %v", err)
|
|
}
|
|
|
|
var gotUserID uint32
|
|
var got time.Time
|
|
err := db.QueryRow("SELECT user_id, expires FROM bans WHERE user_id=$1", userID).Scan(&gotUserID, &got)
|
|
if err != nil {
|
|
t.Fatalf("Verification query failed: %v", err)
|
|
}
|
|
if gotUserID != userID {
|
|
t.Errorf("Expected user_id %d, got: %d", userID, gotUserID)
|
|
}
|
|
if !got.Equal(expiry) {
|
|
t.Errorf("Expected expires %v, got: %v", expiry, got)
|
|
}
|
|
}
|
|
|
|
func TestBanUserUpsertPermanentToTemporary(t *testing.T) {
|
|
repo, db, userID := setupUserRepo(t)
|
|
|
|
// First: permanent ban
|
|
if err := repo.BanUser(userID, nil); err != nil {
|
|
t.Fatalf("BanUser (permanent) failed: %v", err)
|
|
}
|
|
|
|
// Upsert: change to temporary
|
|
expiry := time.Now().Add(1 * time.Hour).Truncate(time.Microsecond)
|
|
if err := repo.BanUser(userID, &expiry); err != nil {
|
|
t.Fatalf("BanUser (upsert to temporary) failed: %v", err)
|
|
}
|
|
|
|
var got time.Time
|
|
err := db.QueryRow("SELECT expires FROM bans WHERE user_id=$1", userID).Scan(&got)
|
|
if err != nil {
|
|
t.Fatalf("Verification query failed: %v", err)
|
|
}
|
|
if !got.Equal(expiry) {
|
|
t.Errorf("Expected expires %v after upsert, got: %v", expiry, got)
|
|
}
|
|
}
|
|
|
|
func TestBanUserUpsertTemporaryToPermanent(t *testing.T) {
|
|
repo, db, userID := setupUserRepo(t)
|
|
|
|
// First: temporary ban
|
|
expiry := time.Now().Add(1 * time.Hour).Truncate(time.Microsecond)
|
|
if err := repo.BanUser(userID, &expiry); err != nil {
|
|
t.Fatalf("BanUser (temporary) failed: %v", err)
|
|
}
|
|
|
|
// Upsert: change to permanent
|
|
if err := repo.BanUser(userID, nil); err != nil {
|
|
t.Fatalf("BanUser (upsert to permanent) failed: %v", err)
|
|
}
|
|
|
|
var expires sql.NullTime
|
|
err := db.QueryRow("SELECT expires FROM bans WHERE user_id=$1", userID).Scan(&expires)
|
|
if err != nil {
|
|
t.Fatalf("Verification query failed: %v", err)
|
|
}
|
|
if expires.Valid {
|
|
t.Errorf("Expected NULL expires after upsert to permanent, got: %v", expires.Time)
|
|
}
|
|
}
|