Enable configuration for executor pool size

pytest_agent/__main__.py

@@ -1,3 +1,4 @@
+import os
 import subprocess
 
 import click
@@ -23,9 +24,19 @@ def cli():
     type=int,
     help="Specify port of host machine on which service is going to listen",
 )
-def start_pytest_agent(host: str, port: int):
+@click.option(
+    "--workers",
+    default=2,
+    type=int,
+    help="Defines max number of workers that can process tests simultaneously",
+)
+def start_pytest_agent(host: str, port: int, workers: int):
+    environment = os.environ.copy()
+    environment["MAX_WORKER_PROCESSES"] = str(workers)
+
     subprocess.call(
-        ["uvicorn", "pytest_agent.api:api", "--host", host, "--port", str(port)]
+        ["uvicorn", "pytest_agent.api:api", "--host", host, "--port", str(port)],
+        env=environment
     )
 
 

pytest_agent/settings.py

@@ -1,5 +1,11 @@
 """
 Defines configuration constants.
 """
+import os
+import logging
+
+LOGGER = logging.getLogger("uvicorn")
+
 SQLALCHEMY_DATABASE_URL = "sqlite:////tmp/pytest_agent.db"
-MAX_WORKER_PROCESSES = 4
+MAX_WORKER_PROCESSES = int(os.getenv("MAX_WORKER_PROCESSES", 2))
+LOGGER.warning(f"Pytest agent starting with {MAX_WORKER_PROCESSES} workers processes")