为什么带有线程的Python unittest测试用例挂起?

时间:2014-03-24 15:11:27

标签: python windows python-3.x python-unittest

我有以下测试用例。请注意,以下测试用例不是 试图测试任何东西,但只是试图证明悬挂问题 我遇到了。

import http.server
import urllib.request
import threading
import unittest

class FooTest(unittest.TestCase):
    def setUp(self):
        print('---- setup start')
        self.httpd = http.server.HTTPServer(('', 8080), http.server.SimpleHTTPRequestHandler)
        thread = threading.Thread(target=self.httpd.serve_forever)
        thread.start()
        print('---- setup complete')

    def tearDown(self):
        print('---- teardown start')
        self.httpd.shutdown()
        print('---- teardown complete')

    def test1(self):
        print('---- test1 start')
        print(threading.current_thread())
        urllib.request.urlopen('http://127.0.0.1:8080/foo')
        print('---- test1 complete')

    def test2(self):
        print('---- test2 start')
        print(threading.current_thread())
        urllib.request.urlopen('http://127.0.0.1:8080/foo')
        print('---- test2 complete')

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

当我尝试执行此测试用例时,我预计会出现2个错误。相反, 测试用例在以下输出后挂起。

C:\lab>python foo.py -v
test1 (__main__.FooTest) ... ---- setup start
---- setup complete
---- test1 start
<_MainThread(MainThread, started 12980)>
127.0.0.1 - - [24/Mar/2014 21:53:57] code 404, message File not found
127.0.0.1 - - [24/Mar/2014 21:53:57] "GET /foo HTTP/1.1" 404 -
---- teardown start
---- teardown complete
ERROR
test2 (__main__.FooTest) ... ---- setup start
---- setup complete
---- test2 start
<_MainThread(MainThread, started 12980)>

如果我从代码中删除test2,那么我预计只会出现1个错误 我看得够了。

C:\lab>python foo.py -v
test1 (__main__.FooTest) ... ---- setup start
---- setup complete
---- test1 start
<_MainThread(MainThread, started 15720)>
127.0.0.1 - - [24/Mar/2014 21:55:12] code 404, message File not found
127.0.0.1 - - [24/Mar/2014 21:55:12] "GET /foo HTTP/1.1" 404 -
---- teardown start
---- teardown complete
ERROR

======================================================================
ERROR: test1 (__main__.FooTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "foo.py", line 22, in test1
    urllib.request.urlopen('http://127.0.0.1:8080/foo')
  File "C:\Python34\lib\urllib\request.py", line 153, in urlopen
    return opener.open(url, data, timeout)
  File "C:\Python34\lib\urllib\request.py", line 461, in open
    response = meth(req, response)
  File "C:\Python34\lib\urllib\request.py", line 574, in http_response
    'http', request, response, code, msg, hdrs)
  File "C:\Python34\lib\urllib\request.py", line 499, in error
    return self._call_chain(*args)
  File "C:\Python34\lib\urllib\request.py", line 433, in _call_chain
    result = func(*args)
  File "C:\Python34\lib\urllib\request.py", line 582, in http_error_default
    raise HTTPError(req.full_url, code, msg, hdrs, fp)
urllib.error.HTTPError: HTTP Error 404: File not found

----------------------------------------------------------------------
Ran 1 test in 0.032s

FAILED (errors=1)

如果有2个测试有错误,为什么测试用例会挂起?

2 个答案:

答案 0 :(得分:4)

添加对server_close的调用以关闭套接字:

def setUp(self):
    print('---- setup start')
    handler = http.server.SimpleHTTPRequestHandler
    self.httpd = http.server.HTTPServer(('', 8080), handler)
    threading.Thread(target=self.serve).start()
    print('---- setup complete')

def serve(self):
    try:
        self.httpd.serve_forever()
    finally:
        self.httpd.server_close()

答案 1 :(得分:2)

我找到了另一种在TestCase中运行httpd的方法

import unittest
import threading
from http.server import HTTPServer, BaseHTTPRequestHandler

class TestHttpRequests(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.httpd = HTTPServer(handler=BaseHTTPRequestHandler)
        threading.Thread(target=cls.httpd.serve_forever).start()

    @classmethod
    def tearDownClass(cls):
        cls.httpd.shutdown()