def test_recieve_zero_length(self):
nstr = NetString(self.socket)
# close on empty socket
try:
nstr.read_string()
assert False
except NetStringClosed as e:
assert str(e) == 'connection closed'
def test_not_numbers_in_string_length(self):
# close on not numbers in string length
self.socket.output.sendall(b'abc')
nstr = NetString(self.socket)
try:
nstr.read_string()
assert False
except NetStringException as e:
assert str(e) == 'can\'t receive, wrong net string format'
def test_too_long_string(self):
data = str(NetString.MAX_STRING_LEN).encode()
self.socket.output.sendall(data)
nstr = NetString(self.socket)
try:
data = nstr.read_string()
assert False
except NetStringException as e:
assert str(e) == 'can\'t receive, wrong net string format'
def test_too_short_string(self):
data = "some string"
data = '%d:%s' % (len(data), data)
self.socket.output.sendall(data.encode())
nstr = NetString(self.socket)
try:
data = nstr.read_string()
assert False
except NetStringClosed as e:
assert str(e) == 'can\'t receive, net string closed'
def test_comma_at_end_of_string(self):
data = "some string"
data = '%d:%sx' % (len(data), data)
self.socket.output.sendall(data.encode())
nstr = NetString(self.socket)
try:
data = nstr.read_string()
assert False
except NetStringException as e:
assert str(e) == 'wrong net string format'
def test_slen_greater_max(self):
data = (str(NetString.MAX_STRING_LEN - 1) + ":abc").encode()
NetString.MAX_STRING_LEN -= 2
self.socket.output.sendall(data)
nstr = NetString(self.socket)
try:
data = nstr.read_string()
assert False
except NetStringException as e:
assert str(e) == 'net string too long'
def test_to_wire_and_read_back(self):
some_string = b"some string"
nstr = NetString(self.socket)
nstr.write_string(some_string)
data = nstr.read_string()
assert data == some_string