def test_fullReceive(self):
msg = toSendBuffer(self.sampleBuffer)
msglen = len(msg)
rcv = ReceiveBuffer()
assert not rcv.isDone(), 'initial ReceiveBuffer isDone test'
assert rcv.is_empty()
rcv.addMore(msg)
assert not rcv.is_empty()
self.finalTests(rcv, self.sampleBuffer, 'completion')
def test_fullReceive(self):
msg = toSendBuffer(self.sampleBuffer)
msglen = len(msg)
rcv = ReceiveBuffer()
assert not rcv.isDone(), 'initial ReceiveBuffer isDone test'
assert rcv.is_empty()
rcv.addMore(msg)
assert not rcv.is_empty()
self.finalTests(rcv, self.sampleBuffer, 'completion')
def test_all_points_incomplete_each_byte(self):
message = 'hello'
extra = b'world'
msg = toSendBuffer(message)
origmsglen = len(msg)
msg += extra
for point in range(1, origmsglen - 1):
rcv = ReceiveBuffer()
assert not rcv.isDone()
assert rcv.is_empty()
for bpos in range(point):
rcv.addMore(msg[bpos:bpos+1])
assert not rcv.isDone()
assert not rcv.is_empty()
assert rcv.completed() is None
def test_all_points_incomplete_each_byte(self):
message = 'hello'
extra = b'world'
msg = toSendBuffer(message)
origmsglen = len(msg)
msg += extra
for point in range(1, origmsglen - 1):
rcv = ReceiveBuffer()
assert not rcv.isDone()
assert rcv.is_empty()
for bpos in range(point):
rcv.addMore(msg[bpos:bpos + 1])
assert not rcv.isDone()
assert not rcv.is_empty()
assert rcv.completed() is None
def test_all_points_extra(self):
message = 'hello'
extra = b'world'
msg = toSendBuffer(message)
origmsglen = len(msg)
msg += extra
for point in range(origmsglen+1, len(msg)):
rcv = ReceiveBuffer()
assert not rcv.isDone()
assert rcv.is_empty()
rcv.addMore(msg[:point])
assert rcv.isDone()
assert not rcv.is_empty()
print(rcv.completed())
rmsg, rextra = rcv.completed()
assert rmsg == message
assert rextra == extra[:point-origmsglen]
def test_exact(self):
message = 'hello'
extra = b'world'
msg = toSendBuffer(message)
origmsglen = len(msg)
msg += extra
rcv = ReceiveBuffer()
assert not rcv.isDone()
assert rcv.is_empty()
rcv.addMore(msg[:origmsglen])
assert rcv.isDone()
assert not rcv.is_empty()
# py.test bug, cannot just: assert rcv.completed() == message, b''
rmsg, rextra = rcv.completed()
print(rmsg, rextra)
assert rmsg == message
assert b'' == rextra
def test_all_points_extra(self):
message = 'hello'
extra = b'world'
msg = toSendBuffer(message)
origmsglen = len(msg)
msg += extra
for point in range(origmsglen + 1, len(msg)):
rcv = ReceiveBuffer()
assert not rcv.isDone()
assert rcv.is_empty()
rcv.addMore(msg[:point])
assert rcv.isDone()
assert not rcv.is_empty()
print(rcv.completed())
rmsg, rextra = rcv.completed()
assert rmsg == message
assert rextra == extra[:point - origmsglen]
def test_exact(self):
message = 'hello'
extra = b'world'
msg = toSendBuffer(message)
origmsglen = len(msg)
msg += extra
rcv = ReceiveBuffer()
assert not rcv.isDone()
assert rcv.is_empty()
rcv.addMore(msg[:origmsglen])
assert rcv.isDone()
assert not rcv.is_empty()
# py.test bug, cannot just: assert rcv.completed() == message, b''
rmsg, rextra = rcv.completed()
print(rmsg, rextra)
assert rmsg == message
assert b'' == rextra
