Skip to content

Commit 697654f

Browse files
committed
Update unit test
1 parent 8a357d1 commit 697654f

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

packages/smithy-aws-event-stream/tests/unit/_private/test_serializers.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -80,17 +80,18 @@ async def test_send_to_closed_writer():
8080
assert publisher.closed
8181

8282

83-
async def test_close_with_signing_sends_end_frame():
83+
async def test_close_sends_empty_end_frame_when_signing():
8484
writer = AsyncMock()
8585
end_frame = EventMessage()
8686
signing_config = AsyncMock()
87-
signing_config.signer.sign.return_value = end_frame
87+
signing_config.signer.sign_empty.return_value = end_frame
8888

8989
publisher: AWSEventPublisher[Any] = AWSEventPublisher(
9090
payload_codec=JSONCodec(), async_writer=writer, signing_config=signing_config
9191
)
9292

9393
await publisher.close()
9494

95+
signing_config.signer.sign_empty.assert_awaited_once()
9596
writer.write.assert_awaited_once_with(end_frame.encode())
9697
assert publisher.closed

0 commit comments

Comments
 (0)