Как написать модульный тест для обработчика запросов сокет-сервера Python? - PullRequest
Новая
       28

Как написать модульный тест для обработчика запросов сокет-сервера Python?

5 голосов
/ 11 октября 2011

Как я могу написать модульные тесты для обработчика запросов сокет-сервера? Я пытался использовать следующий код, но столкнулся с проблемой: 

import socketserver, unittest, threading

class TestServer(socketserver.ThreadingMixIn, socketserver.TCPServer):
    allow_reuse_address = True

class MyRequestHandlerTest(unittest.TestCase):
    def setUp(self):
        self.server = TestServer((HOST, PORT), MyRequestHandler)
        self.server_thread = threading.Thread(target=self.server.serve_forever)
        self.client = socket.create_connection((HOST, PORT))
        #self.server_thread.setDaemon(True)
        self.server_thread.start()

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

Если я попытаюсь использовать несколько тестовых функций, как это 

    def test_foo(self):
        self.client.send(b'foo\n')
        result = self.client.recv(1024)
        self.assertEqual(b'FOO', result)

    def test_bar(self):
        self.client.send(b'bar\n')
        result = self.client.recv(1024)
        self.assertEqual(b'BAR', result)

Я получаю ошибку 

socket.error: [Errno 98] Address already in use

в строке self.server = ..., даже если я дал указание тестовому серверу allow_reuse_address.

Редактировать: удалил одну из двух ошибок, которые я первоначально отправил, чтобы упростить вопрос.

1 Ответ

2 голосов
/ 26 октября 2011

Это сработало для меня (после создания RequestHandler) с: 

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

Это может быть ошибка в socketserver shutdown, что он не вызывает server_close.

...