Workflow to test app + script to test workflow

Author: luminousmining

.github/workflows/build_docker.yml

@@ -0,0 +1,27 @@
+name: build_docker
+
+on:
+  workflow_dispatch:
+    inputs:
+      version:
+        description: 'Image version (e.g. 1.0.0)'
+        required: true
+        type: string
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Build Docker image
+        run: docker build -t pool-simulator:${{ inputs.version }} .
+
+      - name: Save image as tar
+        run: docker save pool-simulator:${{ inputs.version }} -o pool-simulator-${{ inputs.version }}.tar
+
+      - name: Upload artifact
+        uses: actions/upload-artifact@v4
+        with:
+          name: pool-simulator-${{ inputs.version }}
+          path: pool-simulator-${{ inputs.version }}.tar

.github/workflows/test_install.yml

@@ -0,0 +1,18 @@
+name: test_install
+
+on:
+  push:
+    branches: [main]
+
+jobs:
+  install:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+
+      - uses: actions/setup-python@v5
+        with:
+          python-version: '3.11'
+
+      - name: Install requirements
+        run: pip install -r requirements.txt

.github/workflows/test_workflow.yml

@@ -0,0 +1,29 @@
+name: test_workflow
+
+on:
+  push:
+    branches: [main]
+
+jobs:
+  workflow:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+
+      - uses: actions/setup-python@v5
+        with:
+          python-version: '3.11'
+
+      - name: Install requirements
+        run: pip install -r requirements.txt
+
+      - name: Start stratum workflow server
+        run: |
+          python main.py \
+            --algo workflow \
+            --workflow-file workflow.example.json \
+            --workflow-name workflow_sample_1 \
+            --port 7878 &
+
+      - name: Run workflow client test
+        run: python scripts/test_workflow_client.py

Dockerfile

@@ -0,0 +1,6 @@
+FROM python:3.11-slim
+WORKDIR /app
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+COPY . .
+ENTRYPOINT ["python", "main.py"]

scripts/test_workflow_client.py

@@ -0,0 +1,60 @@
+import json
+import socket
+import sys
+import time
+
+HOST = '127.0.0.1'
+PORT = 7878
+EXPECTED_STEPS = 3
+TIMEOUT = 10
+
+
+def wait_for_server(timeout: int) -> None:
+    deadline = time.monotonic() + timeout
+    while time.monotonic() < deadline:
+        try:
+            s = socket.create_connection((HOST, PORT), timeout=1)
+            s.close()
+            return
+        except OSError:
+            time.sleep(0.2)
+    print(f'Server did not open on {HOST}:{PORT} within {timeout}s', file=sys.stderr)
+    sys.exit(1)
+
+
+def run() -> None:
+    wait_for_server(TIMEOUT)
+
+    sock = socket.create_connection((HOST, PORT), timeout=TIMEOUT)
+    steps_done = 0
+    buf = ''
+
+    try:
+        while steps_done < EXPECTED_STEPS:
+            chunk = sock.recv(4096).decode()
+            if not chunk:
+                print('Connection closed unexpectedly', file=sys.stderr)
+                sys.exit(1)
+            buf += chunk
+            lines = buf.split('\n')
+            buf = lines[-1]
+            for line in lines[:-1]:
+                line = line.strip()
+                if not line:
+                    continue
+                msg = json.loads(line)
+                msg_id = msg.get('id')
+                if msg_id is None:
+                    continue
+                response = json.dumps({'id': msg_id, 'result': True}) + '\n'
+                sock.sendall(response.encode())
+                steps_done += 1
+                print(f'step {steps_done}/{EXPECTED_STEPS} ok (id={msg_id})')
+    finally:
+        sock.close()
+
+    print(f'{steps_done} steps completed successfully')
+
+
+if __name__ == '__main__':
+    run()