Use context manager pattern in test

Author: NicEastvillage

tests/run_match.py

@@ -10,14 +10,10 @@
 RLBOT_SERVER_FOLDER = DIR / "../../core/RLBotCS/bin/Release/"
 
 if __name__ == "__main__":
-    match_manager = MatchManager(RLBOT_SERVER_FOLDER)
+    with MatchManager(RLBOT_SERVER_FOLDER) as man:
+        man.start_match(MATCH_CONFIG_PATH)
+        assert man.packet is not None
 
-    match_manager.start_match(MATCH_CONFIG_PATH)
-    assert match_manager.packet is not None
-
-    try:
         # wait for the match to end
-        while match_manager.packet.match_info.match_phase != flat.MatchPhase.Ended:
+        while man.packet.match_info.match_phase != flat.MatchPhase.Ended:
             sleep(1.0)
-    finally:
-        match_manager.shut_down()

tests/run_only.py

@@ -14,13 +14,11 @@
         match_config_path = Path(sys.argv[1])
     assert match_config_path.exists(), f"Match config not found: {match_config_path}"
 
-    # start the match
-    match_manager = MatchManager(RLBOT_SERVER_FOLDER)
-    match_manager.start_match(match_config_path, False)
+    with MatchManager(RLBOT_SERVER_FOLDER) as man:
+        man.start_match(match_config_path, False)
 
-    # wait
-    input("\nPress enter to end the match: ")
+        # Wait for input
+        input("\nPress enter to end the match: ")
 
-    # end the match and disconnect
-    match_manager.stop_match()
-    match_manager.disconnect()
+        # End the match
+        man.stop_match()