socket python的单元测试

import unittest
import socket

class SocketTest(unittest.TestCase):
    def test_socket_connection(self):
        # 创建Socket对象
        server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        
        # 绑定地址和端口
        server_socket.bind(('localhost', 8888))
        
        # 监听连接
        server_socket.listen(1)
        
        # 创建客户端Socket对象
        client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        
        # 连接服务器
        client_socket.connect(('localhost', 8888))
        
        # 接受连接
        connection, address = server_socket.accept()
        
        # 发送数据
        client_socket.send(b'Hello, server!')
        
        # 接收数据
        data = connection.recv(1024)
        
        # 断言接收到的数据是否正确
        self.assertEqual(data, b'Hello, server!')
        
        # 关闭连接
        client_socket.close()
        connection.close()
        server_socket.close()

if __name__ == '__main__':
    unittest.main()