Add unit tests for HTTP API blueprints

google-labs-jules[bot] · rnovatorov · google-labs-jules[bot] · commit 87c906c7c9e2 · 2026-04-01T13:03:40.000Z
- Implement tests for Blueprint data model and DTO conversions
- Implement tests for Blueprints Client methods: get, upload, download, validate
- Fix linting issues (black formatting) identified in CI
- Address PR feedback regarding test coverage and structure

Co-authored-by: rnovatorov &lt;20299819+rnovatorov@users.noreply.github.com&gt;
diff --git a/tests/unit/test_http/test_api/test_blueprints/test_blueprint.py b/tests/unit/test_http/test_api/test_blueprints/test_blueprint.py
@@ -1,14 +1,19 @@
 import datetime
+
 from enapter.http.api.blueprints.blueprint import Blueprint
 
+
 def test_blueprint_from_dto():
     dto = {
         "id": "bp_123",
         "created_at": "2023-01-01T12:00:00+00:00",
     }
     blueprint = Blueprint.from_dto(dto)
     assert blueprint.id == "bp_123"
-    assert blueprint.created_at == datetime.datetime(2023, 1, 1, 12, 0, 0, tzinfo=datetime.timezone.utc)
+    assert blueprint.created_at == datetime.datetime(
+        2023, 1, 1, 12, 0, 0, tzinfo=datetime.timezone.utc
+    )
+
 
 def test_blueprint_to_dto():
     created_at = datetime.datetime(2023, 1, 1, 12, 0, 0, tzinfo=datetime.timezone.utc)
@@ -19,6 +24,7 @@ def test_blueprint_to_dto():
         "created_at": "2023-01-01T12:00:00+00:00",
     }
 
+
 def test_blueprint_roundtrip():
     dto = {
         "id": "bp_123",
diff --git a/tests/unit/test_http/test_api/test_blueprints/test_client.py b/tests/unit/test_http/test_api/test_blueprints/test_client.py
@@ -121,7 +121,9 @@ async def test_download_blueprint(blueprints_client, mock_client):
     content = await blueprints_client.download(blueprint_id="bp_123")
 
     assert content == b"blueprint zip content"
-    mock_client.get.assert_called_once_with("v3/blueprints/bp_123/zip", params={"view": "ORIGINAL"})
+    mock_client.get.assert_called_once_with(
+        "v3/blueprints/bp_123/zip", params={"view": "ORIGINAL"}
+    )
 
 
 @pytest.mark.asyncio
@@ -132,10 +134,14 @@ async def test_download_blueprint_compiled(blueprints_client, mock_client):
     mock_response.content = b"compiled blueprint zip content"
     mock_client.get = AsyncMock(return_value=mock_response)
 
-    content = await blueprints_client.download(blueprint_id="bp_123", view=BlueprintView.COMPILED)
+    content = await blueprints_client.download(
+        blueprint_id="bp_123", view=BlueprintView.COMPILED
+    )
 
     assert content == b"compiled blueprint zip content"
-    mock_client.get.assert_called_once_with("v3/blueprints/bp_123/zip", params={"view": "COMPILED"})
+    mock_client.get.assert_called_once_with(
+        "v3/blueprints/bp_123/zip", params={"view": "COMPILED"}
+    )
 
 
 @pytest.mark.asyncio
@@ -157,9 +163,7 @@ async def test_validate_blueprint_with_errors(blueprints_client, mock_client):
     """Test validating a blueprint with errors."""
     mock_response = MagicMock(spec=httpx.Response)
     mock_response.status_code = 200
-    mock_response.json.return_value = {
-        "validation_errors": ["Error 1", "Error 2"]
-    }
+    mock_response.json.return_value = {"validation_errors": ["Error 1", "Error 2"]}
     mock_client.post = AsyncMock(return_value=mock_response)
 
     data = b"invalid blueprint data"
