root/entrouvert/wsgi/tests.py @ 03bce4e0
| 91345889 | Benjamin Dauvergne | from entrouvert.wsgi.middleware import VersionMiddleware
|
|
from unittest import TestCase
|
|||
class MiddlewareTestCase(TestCase):
|
|||
def setUp(self):
|
|||
VersionMiddleware.ENTROUVERT_PACKAGES.append('nose')
|
|||
def application(self, environ, start_response):
|
|||
start_response('coin', {'a':'b'})
|
|||
return ['xxx']
|
|||
def test_version_middleware(self):
|
|||
import json
|
|||
app = VersionMiddleware(self.application)
|
|||
class start_response_cls:
|
|||
status = None
|
|||
headers = None
|
|||
def __call__(self, status, headers):
|
|||
self.status = status
|
|||
self.headers = headers
|
|||
start_response = start_response_cls()
|
|||
result = app({ 'REQUEST_METHOD': 'GET', 'PATH_INFO': '/__version__'}, start_response)
|
|||
result = json.loads(''.join(result))
|
|||
| b16fd40c | Frédéric Péters | self.assertTrue(type(result) is dict)
|
|
self.assertTrue('nose' in result)
|
|||
| 91345889 | Benjamin Dauvergne | result = app({ 'REQUEST_METHOD': 'GET', 'PATH_INFO': '/'}, start_response)
|
|
self.assertEqual(result, ['xxx'])
|
|||
self.assertEqual(start_response.status, 'coin')
|
|||
self.assertEqual(start_response.headers, {'a': 'b'})
|
|||
def tearDown(self):
|
|||
VersionMiddleware.ENTROUVERT_PACKAGES.append('nose')
|