package domain

import (
	"testing"
	"time"
)

func testFormation(t *testing.T, count int) MatchFormation {
	t.Helper()
	now := time.Unix(1000, 0)
	players := make([]Candidate, count)
	for i := range players {
		players[i] = Candidate{TicketID: string(rune('a' + i)), PlayerID: string(rune('p' + i)), ProtocolVersion: 1, Rating: 1500, EnqueuedAt: now, PredictedRTT: map[string]float64{"EU": 40}}
	}
	selection, err := SelectCandidates(players[0], players[1:], count, now)
	if err != nil {
		t.Fatal(err)
	}
	teams, err := PartitionTeams(selection.Players)
	if err != nil {
		t.Fatal(err)
	}
	return MatchFormation{Selection: selection, Teams: teams}
}

func TestPrepareProposalBuildsCasualLineupBeforeCreatingProposal(t *testing.T) {
	prepared, err := PrepareProposal("proposal-casual-123456", Casual, testFormation(t, 2), nil, RankedArena{}, time.Unix(1000, 0))
	if err != nil {
		t.Fatal(err)
	}
	if prepared.Proposal.Playlist != Casual || len(prepared.Proposal.Participants) != 2 || len(prepared.CasualLineup) != 6 {
		t.Fatalf("prepared casual proposal = %+v", prepared)
	}
	if prepared.Proposal.Region != "EU" || prepared.Proposal.Protocol != 1 || prepared.Proposal.Participants[0].Slot == prepared.Proposal.Participants[1].Slot || prepared.Proposal.Participants[0].Team == prepared.Proposal.Participants[1].Team {
		t.Fatalf("prepared proposal did not retain deterministic topology: %+v", prepared.Proposal)
	}
	humans := 0
	teams := map[int]bool{}
	for _, slot := range prepared.CasualLineup {
		if !slot.IsBot {
			humans++
			teams[slot.Team] = true
		}
	}
	if humans != 2 || len(teams) != 2 {
		t.Fatalf("casual lineup humans/teams = %d/%v", humans, teams)
	}
}

func TestPrepareProposalRejectsInvalidRankedMetadataAndAcceptsVerifiedSix(t *testing.T) {
	formation := testFormation(t, 6)
	participants := make([]RankedParticipant, 6)
	for i, player := range formation.Selection.Players {
		participants[i] = RankedParticipant{PlayerID: player.PlayerID, SteamID: "steam-" + player.PlayerID}
	}
	if _, err := PrepareProposal("proposal-ranked-123456", Ranked, formation, participants, DefaultRankedArena(), time.Unix(1000, 0)); err != nil {
		t.Fatal(err)
	}
	participants[0].IsBot = true
	if _, err := PrepareProposal("proposal-ranked-654321", Ranked, formation, participants, DefaultRankedArena(), time.Unix(1000, 0)); err == nil {
		t.Fatal("ranked bot metadata accepted")
	}
	participants[0].IsBot = false
	participants[0].PlayerID = "unknown"
	if _, err := PrepareProposal("proposal-ranked-000000", Ranked, formation, participants, DefaultRankedArena(), time.Unix(1000, 0)); err == nil {
		t.Fatal("ranked unknown player metadata accepted")
	}
}