Python单元在单套接字的tcp服务器上测试多个测试

import unittest
import socket
import threading

# 单套接字的TCP服务器类
class TCPServer:
    def __init__(self, host, port):
        self.host = host
        self.port = port
        self.server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.server_socket.bind((self.host, self.port))
        self.server_socket.listen(1)

    def start(self):
        while True:
            client_socket, client_address = self.server_socket.accept()
            # 处理客户端请求的逻辑

# 测试用例类
class TCPServerTestCase(unittest.TestCase):
    def setUp(self):
        self.server = TCPServer('localhost', 8000)
        self.server_thread = threading.Thread(target=self.server.start)
        self.server_thread.start()

    def tearDown(self):
        self.server.server_socket.close()
        self.server_thread.join()

    def test_server_connection(self):
        client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        client_socket.connect(('localhost', 8000))
        # 发送测试数据并验证服务器的响应

    def test_server_behavior(self):
        # 编写其他测试用例来验证服务器在不同情况下的行为

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