class TCPSocketConnectionTest(TestCase):
def setUp(self):
socket = Mock()
socket.recv = Mock()
socket.send = Mock()
with patch("socket.create_connection") as create_connection:
create_connection.return_value = socket
self.connection = TCPSocketConnection(("localhost", 1234))
def test_flush(self):
self.assertRaises(TypeError, self.connection.flush)
def test_receive(self):
self.assertRaises(TypeError, self.connection.receive, "")
def test_remaining(self):
self.assertRaises(TypeError, self.connection.remaining)
def test_read(self):
self.connection.socket.recv.return_value = bytearray.fromhex("7FAA")
self.assertEqual(self.connection.read(2), bytearray.fromhex("7FAA"))
def test_read_empty(self):
self.connection.socket.recv.return_value = bytearray.fromhex("")
with self.assertRaises(IOError):
self.connection.read(2)
def test_write(self):
self.connection.write(bytearray.fromhex("7FAA"))
self.connection.socket.send.assert_called_once_with(bytearray.fromhex("7FAA"))
class TCPSocketConnectionTest(TestCase):
def setUp(self):
socket = Mock()
socket.recv = Mock()
socket.send = Mock()
with patch("socket.create_connection") as create_connection:
create_connection.return_value = socket
self.connection = TCPSocketConnection(("localhost", 1234))
def test_flush(self):
self.assertRaises(TypeError, self.connection.flush)
def test_receive(self):
self.assertRaises(TypeError, self.connection.receive, "")
def test_remaining(self):
self.assertRaises(TypeError, self.connection.remaining)
def test_read(self):
self.connection.socket.recv.return_value = bytearray.fromhex("7FAA")
self.assertEqual(self.connection.read(2), bytearray.fromhex("7FAA"))
def test_read_empty(self):
self.connection.socket.recv.return_value = bytearray.fromhex("")
with self.assertRaises(IOError):
self.connection.read(2)
def test_write(self):
self.connection.write(bytearray.fromhex("7FAA"))
self.connection.socket.send.assert_called_once_with(
bytearray.fromhex("7FAA"))
class TCPSocketConnectionTest:
def setup_method(self):
socket = Mock()
socket.recv = Mock()
socket.send = Mock()
with patch("socket.create_connection") as create_connection:
create_connection.return_value = socket
self.connection = TCPSocketConnection(("localhost", 1234))
def test_flush(self):
with pytest.raises(TypeError):
self.connection.flush()
def test_receive(self):
with pytest.raises(TypeError):
self.connection.receive("")
def test_remaining(self):
with pytest.raises(TypeError):
self.connection.remaining()
def test_read(self):
self.connection.socket.recv.return_value = bytearray.fromhex("7FAA")
assert self.connection.read(2) == bytearray.fromhex("7FAA")
def test_read_empty(self):
self.connection.socket.recv.return_value = bytearray.fromhex("")
with pytest.raises(IOError):
self.connection.read(2)
def test_write(self):
self.connection.write(bytearray.fromhex("7FAA"))
# pytype: disable=attribute-error
self.connection.socket.send.assert_called_once_with(
bytearray.fromhex("7FAA"))
