test(priming-group): add mocked unit tests for create/get/delete priming group

Author: brandon-wada

test/unit/test_priming_groups.py

@@ -9,6 +9,8 @@
 """
 
 import time
+from datetime import datetime
+from unittest.mock import MagicMock, patch
 
 import pytest
 from groundlight import ExperimentalApi
@@ -51,6 +53,126 @@ def test_list_priming_groups_returns_list(gl_experimental: ExperimentalApi):
     assert all(isinstance(g, PrimingGroup) for g in groups)
 
 
+# ---------------------------------------------------------------------------
+# create / get / delete  (mocked)
+# ---------------------------------------------------------------------------
+
+_MOCK_PIPELINE = {
+    "id": "pipe_mock0000000000000001",
+    "pipeline_config": "never-review",
+    "is_active_pipeline": True,
+    "is_edge_pipeline": False,
+    "is_unclear_pipeline": False,
+    "is_oodd_pipeline": False,
+    "is_enabled": True,
+    "created_at": "2024-01-01T00:00:00Z",
+    "trained_at": "2024-01-01T01:00:00Z",
+}
+
+_MOCK_PG = {
+    "id": "pgp_mock0000000000000001",
+    "name": "test-primer",
+    "is_global": False,
+    "canonical_query": "Is there a dog?",
+    "active_pipeline_config": None,
+    "priming_group_specific_shadow_pipeline_configs": None,
+    "disable_shadow_pipelines": False,
+    "created_at": "2024-01-01T00:00:00Z",
+}
+
+
+def _mock_response(status_code: int, json_body=None):
+    resp = MagicMock()
+    resp.status_code = status_code
+    resp.json.return_value = json_body or {}
+    resp.raise_for_status = MagicMock()
+    if status_code >= 400:
+        resp.raise_for_status.side_effect = Exception(f"HTTP {status_code}")
+    return resp
+
+
+def test_create_priming_group_mocked(gl_experimental: ExperimentalApi):
+    with patch("groundlight.experimental_api.requests.post") as mock_post:
+        mock_post.return_value = _mock_response(201, _MOCK_PG)
+
+        pg = gl_experimental.create_priming_group(
+            name="test-primer",
+            source_ml_pipeline_id="pipe_mock0000000000000001",
+            canonical_query="Is there a dog?",
+        )
+
+    assert isinstance(pg, PrimingGroup)
+    assert pg.id == "pgp_mock0000000000000001"
+    assert pg.name == "test-primer"
+    assert pg.canonical_query == "Is there a dog?"
+    assert pg.is_global is False
+    mock_post.assert_called_once()
+    _, kwargs = mock_post.call_args
+    assert kwargs["json"]["name"] == "test-primer"
+    assert kwargs["json"]["source_ml_pipeline_id"] == "pipe_mock0000000000000001"
+
+
+def test_create_priming_group_disable_shadow_pipelines_mocked(gl_experimental: ExperimentalApi):
+    pg_data = {**_MOCK_PG, "name": "test-primer-noshadow", "disable_shadow_pipelines": True}
+    with patch("groundlight.experimental_api.requests.post") as mock_post:
+        mock_post.return_value = _mock_response(201, pg_data)
+
+        pg = gl_experimental.create_priming_group(
+            name="test-primer-noshadow",
+            source_ml_pipeline_id="pipe_mock0000000000000001",
+            disable_shadow_pipelines=True,
+        )
+
+    assert pg.disable_shadow_pipelines is True
+    _, kwargs = mock_post.call_args
+    assert kwargs["json"]["disable_shadow_pipelines"] is True
+
+
+def test_get_priming_group_mocked(gl_experimental: ExperimentalApi):
+    with patch("groundlight.experimental_api.requests.get") as mock_get:
+        mock_get.return_value = _mock_response(200, _MOCK_PG)
+
+        pg = gl_experimental.get_priming_group("pgp_mock0000000000000001")
+
+    assert isinstance(pg, PrimingGroup)
+    assert pg.id == "pgp_mock0000000000000001"
+    assert pg.name == "test-primer"
+
+
+def test_get_priming_group_unknown_raises_mocked(gl_experimental: ExperimentalApi):
+    with patch("groundlight.experimental_api.requests.get") as mock_get:
+        mock_get.return_value = _mock_response(404)
+
+        with pytest.raises(NotFoundError):
+            gl_experimental.get_priming_group("pgp_doesnotexist000000000000")
+
+
+def test_delete_priming_group_mocked(gl_experimental: ExperimentalApi):
+    with patch("groundlight.experimental_api.requests.get") as mock_get, patch(
+        "groundlight.experimental_api.requests.delete"
+    ) as mock_delete:
+        mock_get.return_value = _mock_response(404)
+        mock_delete.return_value = _mock_response(204)
+
+        gl_experimental.delete_priming_group("pgp_mock0000000000000001")
+
+        # After deletion, getting it should raise NotFoundError
+        with pytest.raises(NotFoundError):
+            gl_experimental.get_priming_group("pgp_mock0000000000000001")
+
+    mock_delete.assert_called_once()
+
+
+def test_created_priming_group_appears_in_list_mocked(gl_experimental: ExperimentalApi):
+    list_response = {"results": [_MOCK_PG]}
+    with patch("groundlight.experimental_api.requests.get") as mock_get:
+        mock_get.return_value = _mock_response(200, list_response)
+
+        groups = gl_experimental.list_priming_groups()
+
+    assert any(g.id == "pgp_mock0000000000000001" for g in groups)
+
+
 # ---------------------------------------------------------------------------
 # create / get / delete  (expensive — require a trained pipeline)
 # ---------------------------------------------------------------------------