File tree Expand file tree Collapse file tree
packages/smithy-aws-event-stream/tests/unit/_private Expand file tree Collapse file tree Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments