jingrow/gunicorn
1
0
Fork 0
mirror of https://github.com/frappe/gunicorn.git synced 2026-01-14 11:09:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix tests

Browse Source
This commit is contained in:
benoitc 2014-08-16 12:26:04 +02:00
parent 719e61bf18
commit f80ddf368c
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tests/test_009-gaiohttp.py
View File

@ -12,6 +12,7 @@ from aiohttp.wsgi import WSGIServerHttpProtocol
import asyncio import asyncio
from gunicorn.workers import gaiohttp from gunicorn.workers import gaiohttp
from gunicorn.workers._gaiohttp import _wrp
from gunicorn.config import Config from gunicorn.config import Config
from unittest import mock from unittest import mock
@ -32,7 +33,7 @@ class WorkerTests(unittest.TestCase):
def tearDown(self): def tearDown(self):
self.loop.close() self.loop.close()
@mock.patch('gunicorn.workers.gaiohttp.asyncio') @mock.patch('gunicorn.workers._gaiohttp.asyncio')
def test_init_process(self, m_asyncio): def test_init_process(self, m_asyncio):
try: try:
self.worker.init_process() self.worker.init_process()
@ -45,7 +46,7 @@ class WorkerTests(unittest.TestCase):
self.assertTrue(m_asyncio.new_event_loop.called) self.assertTrue(m_asyncio.new_event_loop.called)
self.assertTrue(m_asyncio.set_event_loop.called) self.assertTrue(m_asyncio.set_event_loop.called)
@mock.patch('gunicorn.workers.gaiohttp.asyncio') @mock.patch('gunicorn.workers._gaiohttp.asyncio')
def test_run(self, m_asyncio): def test_run(self, m_asyncio):
self.worker.loop = mock.Mock() self.worker.loop = mock.Mock()
self.worker.run() self.worker.run()
@ -64,7 +65,7 @@ class WorkerTests(unittest.TestCase):
self.worker.wsgi, ('localhost', 8080)) self.worker.wsgi, ('localhost', 8080))
self.assertIsInstance(f, WSGIServerHttpProtocol) self.assertIsInstance(f, WSGIServerHttpProtocol)
@mock.patch('gunicorn.workers.gaiohttp.asyncio') @mock.patch('gunicorn.workers._gaiohttp.asyncio')
def test__run(self, m_asyncio): def test__run(self, m_asyncio):
self.worker.ppid = 1 self.worker.ppid = 1
self.worker.alive = True self.worker.alive = True
@ -84,7 +85,7 @@ class WorkerTests(unittest.TestCase):
self.assertTrue(self.worker.log.info.called) self.assertTrue(self.worker.log.info.called)
self.assertTrue(self.worker.notify.called) self.assertTrue(self.worker.notify.called)
@mock.patch('gunicorn.workers.gaiohttp.asyncio') @mock.patch('gunicorn.workers._gaiohttp.asyncio')
def test__run_unix_socket(self, m_asyncio): def test__run_unix_socket(self, m_asyncio):
self.worker.ppid = 1 self.worker.ppid = 1
self.worker.alive = True self.worker.alive = True
@ -128,8 +129,8 @@ class WorkerTests(unittest.TestCase):
self.assertFalse(self.worker.servers) self.assertFalse(self.worker.servers)
self.assertTrue(conn.closing.called) self.assertTrue(conn.closing.called)
@mock.patch('gunicorn.workers.gaiohttp.os') @mock.patch('gunicorn.workers._gaiohttp.os')
@mock.patch('gunicorn.workers.gaiohttp.asyncio.sleep') @mock.patch('gunicorn.workers._gaiohttp.asyncio.sleep')
def test__run_exc(self, m_sleep, m_os): def test__run_exc(self, m_sleep, m_os):
m_os.getpid.return_value = 1 m_os.getpid.return_value = 1
m_os.getppid.return_value = 1 m_os.getppid.return_value = 1
@ -179,14 +180,14 @@ class WorkerTests(unittest.TestCase):
conn = object() conn = object()
tracking = {} tracking = {}
meth = mock.Mock() meth = mock.Mock()
wrp = gaiohttp._wrp(conn, meth, tracking) wrp = _wrp(conn, meth, tracking)
wrp() wrp()
self.assertIn(id(conn), tracking) self.assertIn(id(conn), tracking)
self.assertTrue(meth.called) self.assertTrue(meth.called)
meth = mock.Mock() meth = mock.Mock()
wrp = gaiohttp._wrp(conn, meth, tracking, False) wrp = _wrp(conn, meth, tracking, False)
wrp() wrp()
self.assertNotIn(1, tracking) self.assertNotIn(1, tracking)