def test_read_until(self):
recv_chunks = [b'1', b'234', b'56', b'789', b'0']
mock_socket = unittest.mock.Mock()
mock_socket.recv.side_effect = recv_chunks
self.assertEqual(NailgunProtocol._read_until(mock_socket, 10),
b'1234567890')
self.assertEqual(mock_socket.recv.call_count, len(recv_chunks))
def test_read_until(self):
recv_chunks = [b"1", b"234", b"56", b"789", b"0"]
mock_socket = unittest.mock.Mock()
mock_socket.recv.side_effect = recv_chunks
self.assertEqual(NailgunProtocol._read_until(mock_socket, 10),
b"1234567890")
self.assertEqual(mock_socket.recv.call_count, len(recv_chunks))
def test_read_until_truncated_recv(self):
self.server_sock.sendall(b'x')
self.server_sock.close()
with self.assertRaises(NailgunProtocol.TruncatedRead):
NailgunProtocol._read_until(self.client_sock, 3)
def test_read_until(self): recv_chunks = ['1', '234', '56', '789', '0'] mock_socket = mock.Mock() mock_socket.recv.side_effect = recv_chunks self.assertEqual(NailgunProtocol._read_until(mock_socket, 10), '1234567890') self.assertEqual(mock_socket.recv.call_count, len(recv_chunks))
