mirror of
https://github.com/wesnoth/wesnoth
synced 2025-04-27 16:32:46 +00:00
e7c2105c48
Plus a bunch of changes which were necessary for the script to work: * The "simulate lobby activity" plugin now exits when the server is shut down. * The plugin now uses wesnoth.random() for random number generation. Math.random() uses a fixed seed, which would make all the clients perform the exact same actions. * Exposed wesnoth.random() to plugins to allow the change above. * --nogui no longer implies --wconsole on Windows. With implied --wconsole the clients attached themselves to the standard output of the Python script, which made it impossible to see the output of the script itself.
60 lines
2.0 KiB
Python
60 lines
2.0 KiB
Python
# This script launches wesnothd, followed by 20 copies of wesnoth running the simulate-lobby-activity.lua plugin.
|
|
# The idea is to use the script to simulate a high amount of lobby traffic, e.g. for performance testing.
|
|
|
|
import subprocess
|
|
from subprocess import DEVNULL
|
|
import os
|
|
import os.path
|
|
import sys
|
|
import time
|
|
|
|
PORT = 56321
|
|
NUM_CLIENTS = 20
|
|
EXIT_WAIT_TIME = 20.0
|
|
|
|
def is_in_path(filename):
|
|
for d in os.get_exec_path():
|
|
if os.path.exists(os.path.join(d, filename)):
|
|
return True
|
|
return False
|
|
|
|
if (os.name == "nt" and not is_in_path("SDL2.dll")):
|
|
# Launching Wesnoth is not going to succeed
|
|
sys.exit("Error: SDL2.dll is not in PATH. This suggests that you haven't added the external\\dll directory to your PATH.")
|
|
|
|
print("Launching processes... ", end="")
|
|
|
|
# Change the working directory to the parent directory of the directory where this script resides
|
|
os.chdir(os.path.dirname(os.path.dirname(__file__)))
|
|
|
|
# Wesnoth restarts itself on launch if OMP_WAIT_POLICY isn't set. That's problematic because it makes it impossible to poll
|
|
# when the client processes terminate. Thus, set OMP_WAIT_POLICY.
|
|
os.environ["OMP_WAIT_POLICY"] = "PASSIVE"
|
|
|
|
# Launch the server
|
|
server = subprocess.Popen(("wesnothd", "-p", str(PORT)), -1, None, DEVNULL, DEVNULL, DEVNULL)
|
|
|
|
# Launch the clients
|
|
clients = set()
|
|
for i in range(NUM_CLIENTS):
|
|
clients.add(subprocess.Popen(("wesnoth", "--plugin=simulate-lobby-activity.lua", "--server=localhost:%d" % PORT, "--username=%d" % i, "--nogui"),
|
|
-1, None, DEVNULL, DEVNULL, DEVNULL))
|
|
|
|
input("done.\nPress Enter when you want to terminate all processes.")
|
|
|
|
server.terminate()
|
|
|
|
print("Waiting for clients to terminate...")
|
|
waiting_start_time = time.monotonic()
|
|
while len(clients) > 0 and time.monotonic() < waiting_start_time + EXIT_WAIT_TIME:
|
|
time.sleep(1.0)
|
|
clients_copy = list(clients)
|
|
for c in clients_copy:
|
|
if c.poll() != None:
|
|
# The process has terminated, remove it from the set.
|
|
clients.remove(c)
|
|
|
|
# Make sure that we get rid of the remaining processes
|
|
for c in clients:
|
|
c.kill()
|