def test_close_socket(self):
close_socket(None)
sock = mock.Mock()
sock.configure_mock(**{"shutdown.side_effect": TypeError, "close.side_effect": TypeError})
close_socket(sock)
sock.shutdown.assert_called_with(socket.SHUT_RDWR)
sock.close.assert_called_with()
def test_socketpair(self):
server, client = socketpair()
self.assertEqual(client.send(b"ciao"), 4)
self.assertEqual(server.recv(io.DEFAULT_BUFFER_SIZE), b"ciao")
self.assertEqual(server.send(b"ciao a te"), 9)
self.assertEqual(client.recv(io.DEFAULT_BUFFER_SIZE), b"ciao a te")
close_socket(server)
self.assertTrue(is_socket_closed(server))
self.assertTrue(is_socket_closed(None))
def test_socketpair(self):
server, client = socketpair()
self.assertEqual(client.send(b'ciao'), 4)
self.assertEqual(server.recv(io.DEFAULT_BUFFER_SIZE), b'ciao')
self.assertEqual(server.send(b'ciao a te'), 9)
self.assertEqual(client.recv(io.DEFAULT_BUFFER_SIZE), b'ciao a te')
close_socket(server)
self.assertTrue(is_socket_closed(server))
self.assertTrue(is_socket_closed(None))
def test_close_socket(self):
close_socket(None)
sock = mock.Mock()
sock.configure_mock(**{
'shutdown.side_effect': TypeError,
'close.side_effect': TypeError
})
close_socket(sock)
sock.shutdown.assert_called_with(socket.SHUT_RDWR)
sock.close.assert_called_with()
