Skip to content
This repository was archived by the owner on Oct 23, 2023. It is now read-only.

Commit 594764f

Browse files
committed
Fixed tests
1 parent 10f06f9 commit 594764f

File tree

4 files changed

+28
-18
lines changed

4 files changed

+28
-18
lines changed

python/args.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -133,6 +133,10 @@ def add_server_group(parser: argparse.ArgumentParser):
133133
action="store",
134134
type=int,
135135
help="Port to listen on")
136+
group.add_argument(
137+
"--server-verbose",
138+
action="store_true",
139+
help="Verbose logging for the server")
136140

137141

138142
def add_worker_group(parser: argparse.ArgumentParser):
@@ -172,6 +176,10 @@ def add_worker_group(parser: argparse.ArgumentParser):
172176
action="store",
173177
type=int,
174178
help="Maximum number of pending requests")
179+
group.add_argument(
180+
"--worker-verbose",
181+
action="store_true",
182+
help="Verbose logging for the worker")
175183

176184

177185
def add_execution_group(parser: argparse.ArgumentParser):

python/config.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,12 +16,14 @@ class Config:
1616
"server", "run_server", "run_worker", "storedir", "tempdir",
1717
"cache_size"
1818
],
19-
"server":
20-
["server_logfile", "server_pidfile", "server_address", "server_port"],
19+
"server": [
20+
"server_logfile", "server_pidfile", "server_address", "server_port",
21+
"server_verbose"
22+
],
2123
"worker": [
2224
"worker_logfile", "worker_pidfile", "worker_keep_sandboxes",
2325
"worker_name", "worker_num_cores", "worker_port", "worker_address",
24-
"worker_pending_requests"
26+
"worker_pending_requests", "worker_verbose"
2527
],
2628
"execution": ["exclusive", "extra_time", "copy_exe"],
2729
"ioi": ["detailed_checker"],
@@ -62,6 +64,7 @@ def __init__(self):
6264
self.server_pidfile = None # type: Optional[str]
6365
self.server_address = None # type: Optional[str]
6466
self.server_port = 7070
67+
self.server_verbose = False
6568

6669
# worker group
6770
self.worker_logfile = "/tmp/task-maker-worker.log"
@@ -72,6 +75,7 @@ def __init__(self):
7275
self.worker_port = 7070
7376
self.worker_address = "127.0.0.1"
7477
self.worker_pending_requests = None # type: Optional[int]
78+
self.worker_verbose = False
7579

7680
# execution group
7781
self.exclusive = False

python/manager.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,8 @@ def spawn_server(config: Config):
4848
args += ["--address", config.server_address]
4949
if config.server_port is not None:
5050
args += ["--port", str(config.server_port)]
51+
if config.server_verbose:
52+
args += ["--verbose"]
5153
spawn_backend("server", args, not config.run_server)
5254

5355

@@ -73,6 +75,8 @@ def spawn_worker(config: Config):
7375
args += ["--server", config.worker_address]
7476
if config.worker_pending_requests is not None:
7577
args += ["--pending-requests", str(config.worker_pending_requests)]
78+
if config.worker_verbose:
79+
args += ["--verbose"]
7680
spawn_backend("worker", args, not config.run_worker)
7781

7882

python/tests/test.py

Lines changed: 9 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -40,21 +40,15 @@ def run_tests(task_name: str, file: str):
4040
config.cache = CacheMode.NOTHING
4141
config.task_dir = task_path
4242
config.dry_run = True
43-
config.server_args = \
44-
"--store-dir='{}/files' " \
45-
"--temp-dir='{}/temp' " \
46-
"--pidfile='{}/server.pid' " \
47-
"--logfile={}/server.log " \
48-
"--verbose " \
49-
"--port=7070".format(temp_dir, temp_dir, temp_dir, temp_dir)
50-
config.worker_args = \
51-
"--store-dir='{}/files' " \
52-
"--temp-dir='{}/temp' " \
53-
"--pidfile='{}/worker.pid' " \
54-
"--logfile={}/worker.log " \
55-
"--verbose " \
56-
"--name=local " \
57-
"--server=127.0.0.1:7070".format(temp_dir, temp_dir, temp_dir, temp_dir)
43+
config.storedir = temp_dir + "/files"
44+
config.tempdir = temp_dir + "/temp"
45+
config.server_pidfile = temp_dir + "/server.pid"
46+
config.server_logfile = temp_dir + "/server.log"
47+
config.server_verbose = True
48+
config.server_port = 7070
49+
config.worker_pidfile = temp_dir + "/worker.pid"
50+
config.worker_logfile = temp_dir + "/worker.log"
51+
config.worker_verbose = True
5852
global interface
5953
setup(config)
6054
ret = run(config)

0 commit comments

Comments
 (0)