Skip to content

Commit 954bb30

Browse files
committed
enabling the invalid buffer test
Signed-off-by: Jesse Jaggars <jjaggars@redhat.com>
1 parent 4bdfa3f commit 954bb30

2 files changed

Lines changed: 6 additions & 5 deletions

File tree

receptor/messages/envelope.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -121,6 +121,9 @@ async def finish(self):
121121
async def get(self):
122122
return await self.q.get()
123123

124+
def get_nowait(self):
125+
return self.q.get_nowait()
126+
124127

125128
class Frame:
126129
"""

receptor/tests/test_framedbuffer.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import asyncio
12
import json
23
import uuid
34

@@ -119,10 +120,6 @@ async def test_malformed_frame(framed_buffer, msg_id):
119120
)
120121

121122

122-
@pytest.mark.skip(
123-
reason="""
124-
This test illustrates that sending an incomplete stream corrupts the transport"""
125-
)
126123
@pytest.mark.asyncio
127124
async def test_too_short(framed_buffer, msg_id):
128125
f1 = Frame(Frame.Types.HEADER, 1, 100, 1, 1)
@@ -133,4 +130,5 @@ async def test_too_short(framed_buffer, msg_id):
133130
await framed_buffer.put(f1.serialize() + too_short_header)
134131
await framed_buffer.put(f2.serialize() + too_short_payload)
135132

136-
await framed_buffer.get()
133+
with pytest.raises(asyncio.QueueEmpty):
134+
framed_buffer.get_nowait()

0 commit comments

Comments
 (0)