Skip to content

Commit 97a31a9

Browse files
committed
fix(test): correct wire test implementations
- Fix test_listen_v1_media to pass required request parameter to transcribe_file - Fix test_speak_v1_audio to consume iterator returned by generate()
1 parent e07cbf1 commit 97a31a9

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

tests/wire/test_listen_v1_media.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,5 +92,6 @@ def test_listen_v1_media_transcribe_file() -> None:
9292
"""Test transcribeFile endpoint with WireMock"""
9393
test_id = "listen.v1.media.transcribe_file.0"
9494
client = get_client(test_id)
95-
client.listen.v1.media.transcribe_url()
95+
# transcribe_file requires file content as bytes
96+
client.listen.v1.media.transcribe_file(request=b"test audio content")
9697
verify_request_count(test_id, "POST", "/v1/listen", None, 1)

tests/wire/test_speak_v1_audio.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,5 +5,6 @@ def test_speak_v1_audio_generate() -> None:
55
"""Test generate endpoint with WireMock"""
66
test_id = "speak.v1.audio.generate.0"
77
client = get_client(test_id)
8-
client.speak.v1.audio.generate(text="text")
8+
# generate() returns an iterator, need to consume it for the request to complete
9+
list(client.speak.v1.audio.generate(text="text"))
910
verify_request_count(test_id, "POST", "/v1/speak", None, 1)

0 commit comments

Comments
 (0)