How to write unit test for Python socketserver request handler?

Question

How can I write unit tests for a socketserver request handler? I have tried to use the following code but have encountered a problem:

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()

If I try to use multiple test functions, like this

    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)

I get the error

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

on the line self.server = ..., even though I instructed the test server to allow_reuse_address.

Edit: removed one of the two errors I had originally posted in order to simplify the question.

Might you have another server already running on that port?

Ken Kinder
– Ken Kinder

2011-10-12 21:08:16 +00:00
Commented Oct 12, 2011 at 21:08 — Ken Kinder
– Ken Kinder, Commented Oct 12, 2011 at 21:08

Mark Tolonen · Accepted Answer · 2011-10-26 04:51:59Z

4

It worked for me (after creating a RequestHandler) with:

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

It may be a bug in socketserver shutdown that it doesn't call server_close itself.

answered Oct 26, 2011 at 4:51

Mark Tolonen

181k26 gold badges182 silver badges279 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to write unit test for Python socketserver request handler?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related