Erupe/server/channelserver/repo_rengoku_test.go

package channelserver

import (
	dbutil "erupe-ce/common/db"
	"testing"

	"github.com/jmoiron/sqlx"
)

func setupRengokuRepo(t *testing.T) (*RengokuRepository, *sqlx.DB, uint32, uint32) {
	t.Helper()
	db := SetupTestDB(t)
	userID := CreateTestUser(t, db, "rengoku_test_user")
	charID := CreateTestCharacter(t, db, userID, "RengokuChar")
	guildID := CreateTestGuild(t, db, charID, "RengokuGuild")
	repo := NewRengokuRepository(dbutil.Wrap(db))
	t.Cleanup(func() { TeardownTestDB(t, db) })
	return repo, db, charID, guildID
}

func TestRepoRengokuUpsertScoreNew(t *testing.T) {
	repo, db, charID, _ := setupRengokuRepo(t)

	if err := repo.UpsertScore(charID, 10, 500, 5, 200); err != nil {
		t.Fatalf("UpsertScore failed: %v", err)
	}

	var stagesMp, pointsMp, stagesSp, pointsSp uint32
	if err := db.QueryRow("SELECT max_stages_mp, max_points_mp, max_stages_sp, max_points_sp FROM rengoku_score WHERE character_id=$1", charID).Scan(&stagesMp, &pointsMp, &stagesSp, &pointsSp); err != nil {
		t.Fatalf("Verification query failed: %v", err)
	}
	if stagesMp != 10 || pointsMp != 500 || stagesSp != 5 || pointsSp != 200 {
		t.Errorf("Expected 10/500/5/200, got %d/%d/%d/%d", stagesMp, pointsMp, stagesSp, pointsSp)
	}
}

func TestRepoRengokuUpsertScoreUpdate(t *testing.T) {
	repo, db, charID, _ := setupRengokuRepo(t)

	if err := repo.UpsertScore(charID, 10, 500, 5, 200); err != nil {
		t.Fatalf("First UpsertScore failed: %v", err)
	}
	if err := repo.UpsertScore(charID, 20, 1000, 15, 800); err != nil {
		t.Fatalf("Second UpsertScore failed: %v", err)
	}

	var stagesMp, pointsMp uint32
	if err := db.QueryRow("SELECT max_stages_mp, max_points_mp FROM rengoku_score WHERE character_id=$1", charID).Scan(&stagesMp, &pointsMp); err != nil {
		t.Fatalf("Verification query failed: %v", err)
	}
	if stagesMp != 20 || pointsMp != 1000 {
		t.Errorf("Expected 20/1000 after update, got %d/%d", stagesMp, pointsMp)
	}
}

func TestRepoRengokuGetRankingGlobal(t *testing.T) {
	repo, _, charID, _ := setupRengokuRepo(t)

	if err := repo.UpsertScore(charID, 10, 500, 5, 200); err != nil {
		t.Fatalf("UpsertScore failed: %v", err)
	}

	// Leaderboard 0 = max_stages_mp (global)
	scores, err := repo.GetRanking(0, 0)
	if err != nil {
		t.Fatalf("GetRanking failed: %v", err)
	}
	if len(scores) != 1 {
		t.Fatalf("Expected 1 score, got: %d", len(scores))
	}
	if scores[0].Score != 10 {
		t.Errorf("Expected score=10, got: %d", scores[0].Score)
	}
	if scores[0].Name != "RengokuChar" {
		t.Errorf("Expected name='RengokuChar', got: %q", scores[0].Name)
	}
}

func TestRepoRengokuGetRankingGuildFiltered(t *testing.T) {
	repo, db, charID, guildID := setupRengokuRepo(t)

	if err := repo.UpsertScore(charID, 10, 500, 5, 200); err != nil {
		t.Fatalf("UpsertScore failed: %v", err)
	}

	// Create another character in a different guild
	user2 := CreateTestUser(t, db, "rengoku_user2")
	char2 := CreateTestCharacter(t, db, user2, "RengokuChar2")
	CreateTestGuild(t, db, char2, "OtherGuild")
	if err := repo.UpsertScore(char2, 20, 1000, 15, 800); err != nil {
		t.Fatalf("UpsertScore char2 failed: %v", err)
	}

	// Leaderboard 2 = max_stages_mp (guild-filtered)
	scores, err := repo.GetRanking(2, guildID)
	if err != nil {
		t.Fatalf("GetRanking failed: %v", err)
	}
	if len(scores) != 1 {
		t.Fatalf("Expected 1 guild-filtered score, got: %d", len(scores))
	}
	if scores[0].Name != "RengokuChar" {
		t.Errorf("Expected 'RengokuChar' in guild ranking, got: %q", scores[0].Name)
	}
}

func TestRepoRengokuGetRankingPointsLeaderboard(t *testing.T) {
	repo, _, charID, _ := setupRengokuRepo(t)

	if err := repo.UpsertScore(charID, 10, 500, 5, 200); err != nil {
		t.Fatalf("UpsertScore failed: %v", err)
	}

	// Leaderboard 1 = max_points_mp (global)
	scores, err := repo.GetRanking(1, 0)
	if err != nil {
		t.Fatalf("GetRanking failed: %v", err)
	}
	if len(scores) != 1 {
		t.Fatalf("Expected 1 score, got: %d", len(scores))
	}
	if scores[0].Score != 500 {
		t.Errorf("Expected score=500 for points leaderboard, got: %d", scores[0].Score)
	}
}

func TestRepoRengokuGetRankingSPLeaderboard(t *testing.T) {
	repo, _, charID, _ := setupRengokuRepo(t)

	if err := repo.UpsertScore(charID, 10, 500, 5, 200); err != nil {
		t.Fatalf("UpsertScore failed: %v", err)
	}

	// Leaderboard 4 = max_stages_sp (global)
	scores, err := repo.GetRanking(4, 0)
	if err != nil {
		t.Fatalf("GetRanking failed: %v", err)
	}
	if len(scores) != 1 {
		t.Fatalf("Expected 1 score, got: %d", len(scores))
	}
	if scores[0].Score != 5 {
		t.Errorf("Expected score=5 for SP stages leaderboard, got: %d", scores[0].Score)
	}
}