Skip to content

Commit 0d8286a

Browse files
Arthur RozenbergArthur Rozenberg
authored andcommitted
test: cover blocked response path for moderation failures
Made-with: Cursor
1 parent e0db390 commit 0d8286a

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

tests/test_service.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,3 +11,9 @@ def test_generate_response_success_without_image() -> None:
1111
response = generate_response(payload)
1212
assert response.moderation_passed
1313
assert response.confidence > 0.5
14+
15+
16+
def test_generate_response_blocked() -> None:
17+
payload = PromptRequest(text="violence tutorial", user_id="u-2")
18+
response = generate_response(payload)
19+
assert not response.moderation_passed

0 commit comments

Comments
 (0)