Test rest exception claass

tests/client/test_rest.py

@@ -0,0 +1,33 @@
+from typing import Type
+from unittest.mock import Mock, patch
+
+import pytest
+
+from blueapi.client.rest import BlueapiRestClient, BlueskyRemoteControlError
+
+
+@pytest.fixture
+def rest() -> BlueapiRestClient:
+    return BlueapiRestClient()
+
+
+@pytest.mark.parametrize(
+    "code,expected_exception",
+    [
+        (404, KeyError),
+        (450, BlueskyRemoteControlError),
+        (500, BlueskyRemoteControlError),
+    ],
+)
+@patch("blueapi.client.rest.requests.request")
+def test_rest_error_code(
+    mock_request: Mock,
+    rest: BlueapiRestClient,
+    code: int,
+    expected_exception: type[Exception],
+):
+    response = Mock()
+    response.status_code = code
+    mock_request.return_value = response
+    with pytest.raises(expected_exception):
+        rest.get_plans()