如何为Python socketserver请求处理程序编写单元测试？

import socket
import threading
from unittest import TestCase, mock
from socketserver import ThreadingMixIn, TCPServer, StreamRequestHandler

class MyRequestHandler(StreamRequestHandler):
    def handle(self):
        data = self.rfile.readline().strip()
        self.wfile.write(f"Received: {data}\n".encode())

class ThreadedTCPServer(ThreadingMixIn, TCPServer):
    pass

class TestMyRequestHandler(TestCase):
    def setUp(self):
        self.server = ThreadedTCPServer(('localhost', 0), MyRequestHandler)
        self.server_thread = threading.Thread(target=self.server.serve_forever)
        self.server_thread.daemon = True
        self.server_thread.start()
        self.client = socket.create_connection(self.server.server_address)

    def tearDown(self):
        self.client.close()
        self.server.shutdown()
        self.server.server_close()

    @mock.patch('socket.socket')
    def test_handle(self, mock_socket):
        mock_socket.return_value.recv.side_effect = ['test data', b'']
        self.client.sendall(b'test data\n')
        response = self.client.recv(1024)
        self.assertEqual(response, b'Received: test data\n')