From 6c812d6dd454a9226ffe32226ba855a07a4e5df5 Mon Sep 17 00:00:00 2001
From: tamarzanzouri <tamar.zanzouri@opentrons.com>
Date: Wed, 3 Apr 2024 21:59:57 -0400
Subject: [PATCH] added tests for papi and json play and stop in the middle of
 a run

---
 .../runs/test_play_stop_papi.tavern.yaml      | 114 ++++++++++++++++++
 .../runs/test_play_stop_v6.tavern.yaml        | 114 ++++++++++++++++++
 .../protocols/wait_for_resume_stop_papi.py    |  13 ++
 .../protocols/wait_for_resume_stop_v6.json    |  36 ++++++
 4 files changed, 277 insertions(+)
 create mode 100644 robot-server/tests/integration/http_api/runs/test_play_stop_papi.tavern.yaml
 create mode 100644 robot-server/tests/integration/http_api/runs/test_play_stop_v6.tavern.yaml
 create mode 100644 robot-server/tests/integration/protocols/wait_for_resume_stop_papi.py
 create mode 100644 robot-server/tests/integration/protocols/wait_for_resume_stop_v6.json

diff --git a/robot-server/tests/integration/http_api/runs/test_play_stop_papi.tavern.yaml b/robot-server/tests/integration/http_api/runs/test_play_stop_papi.tavern.yaml
new file mode 100644
index 00000000000..ec1e97b1bd4
--- /dev/null
+++ b/robot-server/tests/integration/http_api/runs/test_play_stop_papi.tavern.yaml
@@ -0,0 +1,114 @@
+test_name: Test python protocol run commands are failed when stopped.
+
+marks:
+  - usefixtures:
+      - ot2_server_base_url
+stages:
+  - name: Upload a python protocol
+    request:
+      url: '{ot2_server_base_url}/protocols'
+      method: POST
+      files:
+        files: 'tests/integration/protocols/wait_for_resume_stop_papi.py'
+    response:
+      status_code: 201
+      save:
+        json:
+          protocol_id: data.id
+
+  - name: Create run from protocol
+    request:
+      url: '{ot2_server_base_url}/runs'
+      method: POST
+      json:
+        data:
+          protocolId: '{protocol_id}'
+    response:
+      status_code: 201
+      save:
+        json:
+          run_id: data.id
+
+  - name: Play the run
+    request:
+      url: '{ot2_server_base_url}/runs/{run_id}/actions'
+      method: POST
+      json:
+        data:
+          actionType: play
+    response:
+      status_code: 201
+
+  - name: Stop the run
+    request:
+      url: '{ot2_server_base_url}/runs/{run_id}/actions'
+      method: POST
+      json:
+        data:
+          actionType: stop
+    response:
+      status_code: 201
+
+  - name: Wait for the run to complete
+    max_retries: 10
+    delay_after: 0.2
+    request:
+      url: '{ot2_server_base_url}/runs/{run_id}'
+      method: GET
+    response:
+      status_code: 200
+      strict:
+        - json:off
+      json:
+        data:
+          status: stopped
+
+  - name: Get run commands
+    max_retries: 10
+    delay_after: 0.2
+    request:
+      url: '{ot2_server_base_url}/runs/{run_id}/commands'
+      method: GET
+    response:
+      status_code: 200
+      strict:
+        - json:off
+      json:
+        data:
+          - id: !anystr
+            key: !anystr
+            commandType: home
+            createdAt: !anystr
+            startedAt: !anystr
+            completedAt: !anystr
+            status: succeeded
+            params: {}
+            notes: []
+          - id: !anystr
+            key: !anystr
+            commandType: home
+            createdAt: !anystr
+            startedAt: !anystr
+            completedAt: !anystr
+            status: succeeded
+            params: { }
+            notes: [ ]
+          - id: !anystr
+            key: !anystr
+            commandType: waitForResume
+            createdAt: !anystr
+            startedAt: !anystr
+            completedAt: !anystr
+            status: failed
+            params: { }
+            notes: [ ]
+            error:
+              createdAt: !anystr
+              detail: 'Run was cancelled'
+              errorCode: '4000'
+              errorInfo: { }
+              errorType: 'RunStoppedError'
+              id: !anystr
+              wrappedErrors: [ ]
+
+
diff --git a/robot-server/tests/integration/http_api/runs/test_play_stop_v6.tavern.yaml b/robot-server/tests/integration/http_api/runs/test_play_stop_v6.tavern.yaml
new file mode 100644
index 00000000000..73ce4c02721
--- /dev/null
+++ b/robot-server/tests/integration/http_api/runs/test_play_stop_v6.tavern.yaml
@@ -0,0 +1,114 @@
+test_name: Test a JSONv6 run can be paused and then cancelled.
+
+marks:
+  - usefixtures:
+      - ot2_server_base_url
+stages:
+  - name: Upload a JSONv6 protocol
+    request:
+      url: '{ot2_server_base_url}/protocols'
+      method: POST
+      files:
+        files: 'tests/integration/protocols/wait_for_resume_stop_v6.json'
+    response:
+      status_code: 201
+      save:
+        json:
+          protocol_id: data.id
+
+  - name: Create run from protocol
+    request:
+      url: '{ot2_server_base_url}/runs'
+      method: POST
+      json:
+        data:
+          protocolId: '{protocol_id}'
+    response:
+      status_code: 201
+      save:
+        json:
+          run_id: data.id
+
+  - name: Play the run
+    request:
+      url: '{ot2_server_base_url}/runs/{run_id}/actions'
+      method: POST
+      json:
+        data:
+          actionType: play
+    response:
+      status_code: 201
+
+  - name: Stop the run
+    request:
+      url: '{ot2_server_base_url}/runs/{run_id}/actions'
+      method: POST
+      json:
+        data:
+          actionType: stop
+    response:
+      status_code: 201
+
+  - name: Wait for the run to complete
+    max_retries: 10
+    delay_after: 0.2
+    request:
+      url: '{ot2_server_base_url}/runs/{run_id}'
+      method: GET
+    response:
+      status_code: 200
+      strict:
+        - json:off
+      json:
+        data:
+          status: stopped
+
+  - name: Get run commands
+    max_retries: 10
+    delay_after: 0.2
+    request:
+      url: '{ot2_server_base_url}/runs/{run_id}/commands'
+      method: GET
+    response:
+      status_code: 200
+      strict:
+        - json:off
+      json:
+        data:
+          - id: !anystr
+            key: !anystr
+            commandType: home
+            createdAt: !anystr
+            startedAt: !anystr
+            completedAt: !anystr
+            status: succeeded
+            params: {}
+            notes: []
+          - id: !anystr
+            key: !anystr
+            commandType: home
+            createdAt: !anystr
+            startedAt: !anystr
+            completedAt: !anystr
+            status: succeeded
+            params: { }
+            notes: [ ]
+          - id: !anystr
+            key: !anystr
+            commandType: waitForResume
+            createdAt: !anystr
+            startedAt: !anystr
+            completedAt: !anystr
+            status: failed
+            params: { }
+            notes: [ ]
+            error:
+              createdAt: !anystr
+              detail: 'Run was cancelled'
+              errorCode: '4000'
+              errorInfo: { }
+              errorType: 'RunStoppedError'
+              id: !anystr
+              wrappedErrors: [ ]
+
+
diff --git a/robot-server/tests/integration/protocols/wait_for_resume_stop_papi.py b/robot-server/tests/integration/protocols/wait_for_resume_stop_papi.py
new file mode 100644
index 00000000000..8ef7bc4cd1c
--- /dev/null
+++ b/robot-server/tests/integration/protocols/wait_for_resume_stop_papi.py
@@ -0,0 +1,13 @@
+from opentrons.protocol_api import ProtocolContext
+
+metadata = {
+    "protocolName": "stop while waiting test",
+    "author": "Opentrons <protocols@opentrons.com>",
+    "apiLevel": "2.15",
+}
+
+
+def run(ctx: ProtocolContext) -> None:
+    ctx.home()
+    ctx.pause()
+    ctx.set_rail_lights(on=True)
diff --git a/robot-server/tests/integration/protocols/wait_for_resume_stop_v6.json b/robot-server/tests/integration/protocols/wait_for_resume_stop_v6.json
new file mode 100644
index 00000000000..5598f2f5779
--- /dev/null
+++ b/robot-server/tests/integration/protocols/wait_for_resume_stop_v6.json
@@ -0,0 +1,36 @@
+{
+  "$otSharedSchema": "#/protocol/schemas/6",
+  "schemaVersion": 6,
+  "metadata": {
+    "protocolName": "Simple test protocol",
+    "author": "engineering <engineering@opentrons.com>",
+    "description": "A short test protocol",
+    "created": 1223131231,
+    "tags": ["unitTest"]
+  },
+  "robot": {
+    "model": "OT-2 Standard",
+    "deckId": "ot2_standard"
+  },
+  "pipettes": {},
+  "modules": {},
+  "labware": {},
+  "liquids": {},
+  "labwareDefinitions": {},
+  "commands": [
+    {
+      "commandType": "home",
+      "params": {}
+    },
+    {
+      "commandType": "waitForResume",
+      "params": {
+      }
+    },
+    {
+      "commandType": "home",
+      "params": {}
+    }
+  ],
+  "commandAnnotations": []
+}