Add scheduled stress test workflow

Signed-off-by: Casper Beyer <casper@synadia.com>

Author: caspervonb

.github/workflows/stress.yml

@@ -0,0 +1,63 @@
+name: stress
+on:
+    schedule:
+        - cron: "0 0 * * *"
+    workflow_dispatch:
+        inputs:
+            repeat_count:
+                description: "Number of times to repeat each test"
+                required: false
+                default: "100"
+            parallel_workers:
+                description: "Number of parallel workers (-n auto or specific number)"
+                required: false
+                default: "auto"
+
+jobs:
+    stress-test:
+        name: ${{ matrix.project }} stress test (python-${{ matrix.python-version }}, nats-server-${{ matrix.nats-server-version }}, ${{ matrix.os }})
+        runs-on: ${{ matrix.os }}
+        strategy:
+            fail-fast: false
+            matrix:
+                python-version: ["3.11", "3.12", "3.13"]
+                os: ["ubuntu-latest", "macos-latest", "windows-latest"]
+                nats-server-version: ["latest"]
+                project: ["nats-core", "nats-server"]
+        steps:
+            - name: Checkout repository
+              uses: actions/checkout@v5
+
+            - name: Set up Go
+              uses: actions/setup-go@v6
+              with:
+                  go-version: "stable"
+
+            - name: Install NATS Server
+              run: |
+                  go install github.com/nats-io/nats-server/v2@${{ matrix.nats-server-version }}
+              shell: bash
+
+            - name: Set up Python ${{ matrix.python-version }}
+              uses: actions/setup-python@v6
+              with:
+                  python-version: ${{ matrix.python-version }}
+
+            - name: Install uv
+              uses: astral-sh/setup-uv@v6
+
+            - name: Install dependencies
+              run: uv sync --dev
+              working-directory: ${{ matrix.project }}
+
+            - name: Run stress tests for ${{ matrix.project }}
+              run: uv run pytest -v -n ${{ github.event.inputs.parallel_workers || 'auto' }} --count=${{ github.event.inputs.repeat_count || '100' }}
+              working-directory: ${{ matrix.project }}
+
+            - name: Upload failure artifacts
+              if: failure()
+              uses: actions/upload-artifact@v4
+              with:
+                  name: stress-test-failures-${{ matrix.os }}-py${{ matrix.python-version }}
+                  path: ${{ matrix.project }}/.pytest_cache/
+                  retention-days: 7

nats-core/pyproject.toml

@@ -43,6 +43,7 @@ dev = [
   "pytest-asyncio>=0.21.0",
   "pytest-cov>=7.0.0",
   "pytest-xdist>=3.0.0",
+  "pytest-repeat>=0.9.0",
   "pytest-benchmark>=5.2.1",
 ]
 

nats-server/pyproject.toml

@@ -33,6 +33,16 @@ Source = "https://github.com/nats-io/nats.py"
 [tool.setuptools.packages.find]
 where = ["src"]
 
+[dependency-groups]
+dev = [
+  "pytest>=7.0.0",
+  "pytest-asyncio>=0.21.0",
+  "pytest-cov>=7.0.0",
+  "pytest-xdist>=3.0.0",
+  "pytest-repeat>=0.9.0",
+  "pytest-benchmark>=5.2.1",
+]
+
 [tool.pytest.ini_options]
 asyncio_default_fixture_loop_scope = "function"
 asyncio_mode = "auto"

nats-server/tests/test_server.py

@@ -10,6 +10,7 @@
 from urllib.parse import urlparse
 
 import pytest
+
 from nats.server import Server, ServerCluster, ServerError, run, run_cluster
 
 
@@ -73,6 +74,7 @@ async def fetch_server_info(server: Server) -> ServerInfo:
     return info
 
 
+@pytest.mark.repeat(1)
 async def test_run_with_default_port():
     """Test the run function with the default port (4222)."""
     # Start with default port (no port specified)
@@ -174,6 +176,7 @@ async def test_run_with_jetstream(jetstream, tmp_path):
     await server.shutdown()
 
 
+@pytest.mark.repeat(1)
 async def test_run_with_port_from_config():
     """Test the run function with port specified in config file."""
     # Start server with config file that sets port to 4567

pyproject.toml

@@ -4,6 +4,7 @@ dev = [
     "pytest-asyncio>=0.21.0",
     "pytest-cov>=7.0.0",
     "pytest-xdist>=3.0.0",
+    "pytest-repeat>=0.9.0",
     "ty>=0.0.0a6",
     "ruff>=0.1.0",
     "codespell>=2.2.0",