def test_receive_message_empty(self):
mock_sock = mock.MagicMock()
mock_sock.recv = mock.MagicMock(side_effect=[None, ''])
connection = BaseSocketConnection(mock_sock)
result = connection.receive_message()
assert result is None
def test_receive_message(self):
mock_sock = mock.MagicMock()
mock_sock.recv = mock.MagicMock(
side_effect=['\x00\x00\x00\x0c', 'some message'])
connection = BaseSocketConnection(mock_sock)
result = connection.receive_message()
assert result == 'some message'
def test_receive_message_empty(self):
mock_sock = mock.MagicMock()
mock_sock.recv = mock.MagicMock(side_effect=[
None,
''
])
connection = BaseSocketConnection(mock_sock)
result = connection.receive_message()
assert result is None
def test_receive_message(self):
mock_sock = mock.MagicMock()
mock_sock.recv = mock.MagicMock(side_effect=[
'\x00\x00\x00\x0c',
'some message'
])
connection = BaseSocketConnection(mock_sock)
result = connection.receive_message()
assert result == 'some message'
