Skip to content

Commit 3fa2760

Browse files
committed
Add test_generate_image_value_error
1 parent bc01070 commit 3fa2760

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

tests/test_firefly_client.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
FireflyAPIError,
1212
FireflyAuthError,
1313
)
14+
from unittest import mock
1415

1516
TOKEN_URL = "https://ims-na1.adobelogin.com/ims/token/v3"
1617
IMAGE_URL = "https://firefly-api.adobe.io/v3/images/generate"
@@ -149,3 +150,11 @@ def test_image_generation_unauthorized(client, mock_valid_ims_access_token_respo
149150
)
150151
with pytest.raises(FireflyAuthError):
151152
client.generate_image(prompt="unauthorized access")
153+
154+
155+
@responses.activate
156+
def test_generate_image_value_error(client, mock_valid_ims_access_token_response):
157+
# Patch requests.request to raise ValueError
158+
with mock.patch("requests.request", side_effect=ValueError("bad value")):
159+
with pytest.raises(FireflyAPIError):
160+
client.generate_image(prompt="trigger value error")

0 commit comments

Comments
 (0)