OpenFrontIO/tests/TeamAssignment.test.ts

import { TeamName, PlayerType, PlayerInfo } from "../src/core/game/Game";
import { assignTeams } from "../src/core/game/TeamAssignment";

describe("assignTeams", () => {
  const createPlayer = (id: string, clan?: string): PlayerInfo => {
    const name = clan ? `[${clan}]Player ${id}` : `Player ${id}`;
    return new PlayerInfo(
      "🏳️", // flag
      name,
      PlayerType.Human,
      null, // clientID (null for testing)
      id,
      null, // nation (null for testing)
    );
  };

  it("should assign players to teams when no clans are present", () => {
    const players = [
      createPlayer("1"),
      createPlayer("2"),
      createPlayer("3"),
      createPlayer("4"),
    ];

    const result = assignTeams(players);

    // Check that players are assigned alternately
    expect(result.get(players[0])).toEqual({ name: TeamName.Red });
    expect(result.get(players[1])).toEqual({ name: TeamName.Blue });
    expect(result.get(players[2])).toEqual({ name: TeamName.Red });
    expect(result.get(players[3])).toEqual({ name: TeamName.Blue });
  });

  it("should keep clan members together on the same team", () => {
    const players = [
      createPlayer("1", "CLANA"),
      createPlayer("2", "CLANA"),
      createPlayer("3", "CLANB"),
      createPlayer("4", "CLANB"),
    ];

    const result = assignTeams(players);

    // Check that clan members are on the same team
    expect(result.get(players[0])).toEqual({ name: TeamName.Red });
    expect(result.get(players[1])).toEqual({ name: TeamName.Red });
    expect(result.get(players[2])).toEqual({ name: TeamName.Blue });
    expect(result.get(players[3])).toEqual({ name: TeamName.Blue });
  });

  it("should handle mixed clan and non-clan players", () => {
    const players = [
      createPlayer("1", "CLANA"),
      createPlayer("2", "CLANA"),
      createPlayer("3"),
      createPlayer("4"),
    ];

    const result = assignTeams(players);

    // Check that clan members are together and non-clan players balance teams
    expect(result.get(players[0])).toEqual({ name: TeamName.Red });
    expect(result.get(players[1])).toEqual({ name: TeamName.Red });
    expect(result.get(players[2])).toEqual({ name: TeamName.Blue });
    expect(result.get(players[3])).toEqual({ name: TeamName.Blue });
  });

  it("should kick players when teams are full", () => {
    const players = [
      createPlayer("1", "CLANA"),
      createPlayer("2", "CLANA"),
      createPlayer("3", "CLANA"),
      createPlayer("4", "CLANA"),
      createPlayer("5", "CLANB"),
      createPlayer("6", "CLANB"),
    ];

    const result = assignTeams(players);

    // Check that players are kicked when teams are full
    expect(result.get(players[0])).toEqual({ name: TeamName.Red });
    expect(result.get(players[1])).toEqual({ name: TeamName.Red });
    expect(result.get(players[2])).toEqual({ name: TeamName.Red });

    expect(result.get(players[3])).toEqual("kicked");

    expect(result.get(players[4])).toEqual({ name: TeamName.Blue });
    expect(result.get(players[5])).toEqual({ name: TeamName.Blue });
  });

  it("should handle empty player list", () => {
    const result = assignTeams([]);
    expect(result.size).toBe(0);
  });

  it("should handle single player", () => {
    const players = [createPlayer("1")];
    const result = assignTeams(players);
    expect(result.get(players[0])).toEqual({ name: TeamName.Red });
  });

  it("should handle multiple clans with different sizes", () => {
    const players = [
      createPlayer("1", "CLANA"),
      createPlayer("2", "CLANA"),
      createPlayer("3", "CLANA"),
      createPlayer("4", "CLANB"),
      createPlayer("5", "CLANB"),
      createPlayer("6", "CLANC"),
    ];

    const result = assignTeams(players);

    // Check that larger clans are assigned first
    expect(result.get(players[0])).toEqual({ name: TeamName.Red });
    expect(result.get(players[1])).toEqual({ name: TeamName.Red });
    expect(result.get(players[2])).toEqual({ name: TeamName.Red });
    expect(result.get(players[3])).toEqual({ name: TeamName.Blue });
    expect(result.get(players[4])).toEqual({ name: TeamName.Blue });
    expect(result.get(players[5])).toEqual({ name: TeamName.Blue });
  });
});