Infinite match series testing

Author: VirxEC

tests/atba/atba.bot.toml

@@ -14,7 +14,7 @@ root_dir = ""
 run_command = "..\\..\\venv\\Scripts\\python atba.py"
 # The command RLBot will call to start your bot on linux
 # If this isn't set, RLBot may attempt to run the Windows command under Wine
-run_command_linux = "../../venv/bin/python atba.py"
+run_command_linux = "../../.venv/bin/python atba.py"
 
 [details]
 description = "Made possible by RLBot"

tests/atba/atba.py

@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 import math
-from typing import Optional
 
 from rlbot import flat
 from rlbot.managers import Bot
@@ -67,9 +66,9 @@ def __init__(self):
 
     def initialize(self):
         self.logger.info("Initializing agent!")
-
-        num_boost_pads = len(self.field_info.boost_pads)
-        self.logger.info(f"There are {num_boost_pads} boost pads on the field.")
+        self.logger.info(
+            f"There are {len(self.field_info.boost_pads)} boost pads and {len(self.field_info.goals)} goals on the field."
+        )
 
         if self.rendering:
             self.renderer.begin_rendering("custom one-time rendering group")
@@ -83,16 +82,6 @@ def initialize(self):
             )
             self.renderer.end_rendering()
 
-    def handle_match_comm(
-        self,
-        index: int,
-        team: int,
-        content: bytes,
-        display: Optional[str],
-        team_only: bool,
-    ):
-        self.logger.info(f"Received match communication from index {index}! {display}")
-
     def get_output(self, packet: flat.GamePacket) -> flat.ControllerState:
         if self.rendering:
             self.test_rendering(packet)
@@ -110,10 +99,10 @@ def get_output(self, packet: flat.GamePacket) -> flat.ControllerState:
         if self.state_setting:
             self.test_state_setting(packet)
 
-        if self.match_comms:
+        if self.match_comms and packet.match_info.match_phase == flat.MatchPhase.Active:
             # Limit packet spam
             if packet.match_info.frame_num - self.last_send >= 360:
-                self.send_match_comm(b"", "Hello world!", True)
+                self.send_match_comm(b"", "Hello world!", False)
                 self.last_send = packet.match_info.frame_num
 
         ball_location = Vector2(packet.balls[0].physics.location)
@@ -145,7 +134,7 @@ def test_state_setting(self, packet: flat.GamePacket):
                     i: flat.DesiredCarState(
                         flat.DesiredPhysics(rotation=flat.RotatorPartial(yaw=0))
                     )
-                    for i, car in enumerate(packet.players)
+                    for i in range(len(packet.players))
                 },
             )
 

tests/many_match.py

@@ -0,0 +1,84 @@
+from pathlib import Path
+from time import sleep
+
+from rlbot import flat
+from rlbot.config import load_player_config
+from rlbot.managers import MatchManager
+
+DIR = Path(__file__).parent
+
+BOT_PATH = DIR / "atba/atba.bot.toml"
+RLBOT_SERVER_FOLDER = DIR / "../../core/RLBotCS/bin/Release/"
+
+num_comms = set()
+
+
+def handle_match_comm(comm: flat.MatchComm):
+    global num_comms
+    if comm.team < 2:
+        num_comms.add(comm.index)
+
+
+if __name__ == "__main__":
+    match_manager = MatchManager(RLBOT_SERVER_FOLDER)
+    match_manager.rlbot_interface.match_comm_handlers.append(handle_match_comm)
+    match_manager.ensure_server_started()
+    match_manager.connect(
+        wants_match_communications=True,
+        wants_ball_predictions=False,
+        close_between_matches=False,
+    )
+
+    current_map = -1
+
+    blue_bot = load_player_config(BOT_PATH, 0)
+    orange_bot = load_player_config(BOT_PATH, 1)
+
+    match_settings = flat.MatchConfiguration(
+        launcher=flat.Launcher.Steam,
+        auto_start_agents=True,
+        wait_for_agents=True,
+        existing_match_behavior=flat.ExistingMatchBehavior.Restart,
+        game_map_upk="Stadium_P",
+        instant_start=True,
+        enable_state_setting=True,
+        player_configurations=[
+            blue_bot,
+            blue_bot,
+            blue_bot,
+            blue_bot,
+            blue_bot,
+            orange_bot,
+            orange_bot,
+            orange_bot,
+            orange_bot,
+            orange_bot,
+        ],
+    )
+
+    num_games = 0
+    paused = False
+
+    while not paused:
+        num_games += 1
+        print(f"Starting match # {num_games}")
+
+        match_manager.start_match(match_settings, ensure_server_started=False)
+        # when calling start_match, by default it will wait for the first packet
+        assert match_manager.packet is not None
+
+        sleep(2)
+        num_comms.clear()
+        while len(num_comms) < 10:
+            # give an extra 5 seconds for the match to start before calling it a failure
+            if (
+                match_manager.packet.match_info.match_phase == flat.MatchPhase.Active
+                and match_manager.packet.match_info.game_time_remaining < 60 * 4 + 55
+            ):
+                match_manager.set_game_state(commands=["Pause"])
+                paused = True
+                break
+            sleep(1)
+
+    print("Failed to start match. Paused and exiting.")
+    match_manager.disconnect()