From acbb1aa4164304ba3318a583acad140fa03a3c02 Mon Sep 17 00:00:00 2001
From: amit lissack <amit@opentrons.com>
Date: Mon, 4 Jan 2021 11:43:29 -0500
Subject: [PATCH] command model tests.

---
 .../tests/service/session/models/__init__.py  |   0
 .../service/session/models/test_command.py    | 112 ++++++++++++++++++
 2 files changed, 112 insertions(+)
 create mode 100644 robot-server/tests/service/session/models/__init__.py
 create mode 100644 robot-server/tests/service/session/models/test_command.py

diff --git a/robot-server/tests/service/session/models/__init__.py b/robot-server/tests/service/session/models/__init__.py
new file mode 100644
index 00000000000..e69de29bb2d
diff --git a/robot-server/tests/service/session/models/test_command.py b/robot-server/tests/service/session/models/test_command.py
new file mode 100644
index 00000000000..4b7df3a8ece
--- /dev/null
+++ b/robot-server/tests/service/session/models/test_command.py
@@ -0,0 +1,112 @@
+from datetime import datetime
+import pytest
+from pydantic import ValidationError
+
+from robot_server.service.session.models import command, command_definitions
+
+
+@pytest.mark.parametrize(
+    argnames="command_def",
+    argvalues=[
+        command_definitions.ProtocolCommand.start_run,
+        command_definitions.CalibrationCommand.move_to_deck,
+        command_definitions.CheckCalibrationCommand.compare_point,
+    ])
+def test_empty(command_def: command_definitions.CommandDefinition):
+    """Test creation of empty command request and response."""
+    request = command.CommandRequest.parse_obj({
+        "data": {
+            "command": command_def.value,
+            "data": {}
+        }
+    })
+    assert request.data.command == command_def
+    assert request.data.data == command.EmptyModel()
+
+    dt = datetime(2000, 1, 1)
+
+    response = request.data.make_response(
+        identifier="id",
+        status=command.CommandStatus.executed,
+        created_at=dt,
+        started_at=None,
+        completed_at=None,
+        result=None
+    )
+
+    assert response.command == command_def
+    assert response.data == command.EmptyModel()
+    assert response.id == "id"
+    assert response.createdAt == dt
+    assert response.startedAt is None
+    assert response.completedAt is None
+    assert response.result is None
+
+
+def test_not_empty():
+    """Test that command request with data and result are created properly."""
+    request = command.CommandRequest.parse_obj({
+        "data": {
+            "command": "equipment.loadInstrument",
+            "data": {
+                "instrumentName": "p10_single",
+                "mount": "left"
+            }
+        }
+    })
+    assert request.data.command == \
+           command_definitions.EquipmentCommand.load_instrument
+    assert request.data.data == command.LoadInstrumentRequestData(
+        instrumentName="p10_single",
+        mount=command.Mount.left
+    )
+
+    dt = datetime(2000, 1, 1)
+
+    response = request.data.make_response(
+        identifier="id",
+        status=command.CommandStatus.executed,
+        created_at=dt,
+        started_at=None,
+        completed_at=None,
+        result=command.LoadInstrumentResponseData(
+            instrumentId=command.IdentifierType("123")
+        )
+    )
+
+    assert response.command == \
+           command_definitions.EquipmentCommand.load_instrument
+    assert response.data == command.LoadInstrumentRequestData(
+        instrumentName="p10_single",
+        mount=command.Mount.left
+    )
+    assert response.id == "id"
+    assert response.createdAt == dt
+    assert response.startedAt is None
+    assert response.completedAt is None
+    assert response.result == command.LoadInstrumentResponseData(
+        instrumentId=command.IdentifierType("123")
+    )
+
+
+@pytest.mark.parametrize(
+    argnames="command_def",
+    argvalues=[
+        command_definitions.EquipmentCommand.load_labware,
+        command_definitions.EquipmentCommand.load_instrument,
+        command_definitions.PipetteCommand.aspirate,
+        command_definitions.PipetteCommand.dispense,
+        command_definitions.PipetteCommand.drop_tip,
+        command_definitions.PipetteCommand.pick_up_tip,
+        command_definitions.CalibrationCommand.jog,
+        command_definitions.CalibrationCommand.set_has_calibration_block
+    ])
+def test_requires_data(command_def: command_definitions.CommandDefinition):
+    """Test creation of command requiring data will fail with empty body."""
+    with pytest.raises(ValidationError):
+        command.CommandRequest.parse_obj({
+            "data": {
+                "command": command_def.value,
+                "data": {}
+            }
+        })