Update table snapshot tests.

Opentrons · Jul 29, 2024 · 6b96456 · 6b96456
1 parent c66e65b
commit 6b96456
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/robot-server/tests/persistence/test_tables.py b/robot-server/tests/persistence/test_tables.py
@@ -33,8 +33,9 @@
         id VARCHAR NOT NULL,
         created_at DATETIME NOT NULL,
         protocol_key VARCHAR,
-        protocol_kind VARCHAR,
-        PRIMARY KEY (id)
+        protocol_kind VARCHAR(14) NOT NULL,
+        PRIMARY KEY (id),
+        CONSTRAINT protocolkindsqlenum CHECK (protocol_kind IN ('standard', 'quick-transfer'))
     )
     """,
     """
@@ -114,6 +115,9 @@
     CREATE UNIQUE INDEX ix_run_run_id_index_in_run ON run_command (run_id, index_in_run)
     """,
     """
+    CREATE INDEX ix_protocol_protocol_kind ON protocol (protocol_kind)
+    """,
+    """
     CREATE TABLE data_files (
         id VARCHAR NOT NULL,
         name VARCHAR NOT NULL,