def test_close(self): mock_sock = mock.MagicMock() connection = BaseSocketConnection(mock_sock) connection.close() assert mock_sock.close.call_count == 1
def test_send_message(self): mock_sock = mock.MagicMock() connection = BaseSocketConnection(mock_sock) connection.send_message('some message') assert mock_sock.sendall.call_count == 1 assert mock_sock.sendall.call_args[0] == (b'\x00\x00\x00\x0csome message',)
def test_send_message(self): mock_sock = mock.MagicMock() connection = BaseSocketConnection(mock_sock) connection.send_message('some message') assert mock_sock.sendall.call_count == 1 assert mock_sock.sendall.call_args[0] == ( b'\x00\x00\x00\x0csome 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'
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'