Revert "Update api/test/unit/views/test_image_views.py"

This reverts commit 11518b6.
WordPress · Oct 25, 2023 · 16c920e · 16c920e
1 parent 11518b6
commit 16c920e
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/api/test/unit/views/test_image_views.py b/api/test/unit/views/test_image_views.py
@@ -85,11 +85,15 @@ def test_thumbnail_uses_upstream_thumb_for_smk(
 @pytest.mark.django_db
 def test_watermark_raises_424_for_invalid_image(api_client):
     image = ImageFactory.create()
-    with patch(
-        "PIL.Image.open", side_effect=UnidentifiedImageError("bad image")
-    ), pytest.raises(UpstreamWatermarkException):
+    expected_error_message = (
+        "cannot identify image file <_io.BytesIO object at 0xffff86d8fec0>"
+    )
+
+    with patch("PIL.Image.open") as mock_open:
+        mock_open.side_effect = UnidentifiedImageError(expected_error_message)
         res = api_client.get(f"/v1/images/{image.identifier}/watermark/")
     assert res.status_code == 424
+    assert res.data["detail"] == expected_error_message
 
 
 @pytest.mark.django_db