mirror of
https://github.com/Mezeporta/Erupe.git
synced 2026-03-22 07:32:32 +01:00
f2f5696a22
Cover the three repository methods added in the previous commit: - CharacterRepo.LoadSaveData: normal, new-character, and not-found cases - EventRepo.GetEventQuests: empty, multi-row, ordering, tx commit/rollback - UserRepo.BanUser: permanent, temporary, and both upsert directions
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)
|
|
}
|
|
}
|