@@ -1656,22 +1656,25 @@ def test_sendfile(self):
1656
1656
1657
1657
async def serve_callback (stream ):
1658
1658
data = await stream .readline ()
1659
- self . assertEqual ( data , b'begin \n ' )
1659
+ await stream . write ( b'ack-' + data )
1660
1660
data = await stream .readline ()
1661
- self . assertEqual ( data , b'data \n ' )
1661
+ await stream . write ( b'ack-' + data )
1662
1662
data = await stream .readline ()
1663
- self .assertEqual (data , b'end\n ' )
1664
- await stream .write (b'done\n ' )
1663
+ await stream .write (b'ack-' + data )
1665
1664
await stream .close ()
1666
1665
1667
1666
async def do_connect (host , port ):
1668
1667
stream = await asyncio .connect (host , port )
1669
1668
await stream .write (b'begin\n ' )
1669
+ data = await stream .readline ()
1670
+ self .assertEqual (b'ack-begin\n ' , data )
1670
1671
with open (support .TESTFN , 'rb' ) as fp :
1671
1672
await stream .sendfile (fp )
1673
+ data = await stream .readline ()
1674
+ self .assertEqual (b'ack-data\n ' , data )
1672
1675
await stream .write (b'end\n ' )
1673
1676
data = await stream .readline ()
1674
- self .assertEqual (data , b'done \n ' )
1677
+ self .assertEqual (data , b'ack-end \n ' )
1675
1678
await stream .close ()
1676
1679
1677
1680
async def test ():
0 commit comments