def test_tcp_client_timeout(self):
"""
Given: A SocketConnection 'tcp' object and a TCP server, set not to respond.
When: Calling SocketConnection.open(), .send(), .recv(), and .close()
Then: send() returns length of payload.
and: Sent works as expected, and recv() returns bytes('') after timing out.
"""
data_to_send = six.binary_type(b'uuddlrlrba')
# Given
server = MiniTestServer(stay_silent=True)
server.bind()
t = threading.Thread(target=server.serve_once)
t.daemon = True
t.start()
uut = SocketConnection(host=socket.gethostname(),
port=server.active_port,
proto='tcp')
uut.logger = logging.getLogger("SulleyUTLogger")
# When
uut.open()
send_result = uut.send(data=data_to_send)
received = uut.recv(10000)
uut.close()
# Wait for the other thread to terminate
t.join(THREAD_WAIT_TIMEOUT)
self.assertFalse(t.isAlive())
# Then
self.assertEqual(send_result, len(data_to_send))
self.assertEqual(data_to_send, server.received)
self.assertEqual(received, six.binary_type(b''))
def test_udp_client(self):
"""
Given: A SocketConnection 'udp' object and a UDP server.
When: Calling SocketConnection.open(), .send(), .recv(), and .close()
Then: send() returns length of payload.
and: Sent and received data is as expected.
"""
data_to_send = bytes('"Rum idea this is, that tidiness is a timid, quiet sort of thing;'
' why, tidiness is a toil for giants."')
# Given
server = MiniTestServer(proto='udp')
server.bind()
t = threading.Thread(target=server.serve_once)
t.daemon = True
t.start()
uut = SocketConnection(host=socket.gethostname(), port=server.active_port, proto='udp',
bind=(socket.gethostname(), 0))
uut.logger = logging.getLogger("SulleyUTLogger")
# When
uut.open()
send_result = uut.send(data=data_to_send)
received = uut.recv(10000)
uut.close()
# Wait for the other thread to terminate
t.join(THREAD_WAIT_TIMEOUT)
self.assertFalse(t.isAlive())
# Then
self.assertEqual(send_result, len(data_to_send))
self.assertEqual(data_to_send, server.received)
self.assertEqual(received, server.data_to_send)
