1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
import logging
import unittest
import six
if six.PY3:
from unittest import mock
else:
import mock
from engineio import async_eventlet
class TestAsyncEventlet(unittest.TestCase):
def setUp(self):
logging.getLogger('engineio').setLevel(logging.NOTSET)
def test_bad_environ(self):
wsgi = async_eventlet.WebSocketWSGI(None)
environ = {'foo': 'bar'}
start_response = 'bar'
self.assertRaises(RuntimeError, wsgi, environ, start_response)
@mock.patch('engineio.async_eventlet._WebSocketWSGI.__call__',
return_value='data')
def test_wsgi_call(self, _WebSocketWSGI):
_WebSocketWSGI.__call__ = lambda e, s: 'data'
environ = {'eventlet.input': mock.MagicMock()}
start_response = 'bar'
wsgi = async_eventlet.WebSocketWSGI(None)
self.assertEqual(wsgi(environ, start_response), 'data')
|
