pythonasyncioeventletgeventlong-pollinglow-latencysocket-iosocketiosocketio-serverweb-serverwebsocket
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
42 lines
1.4 KiB
42 lines
1.4 KiB
import unittest
|
|
|
|
import six
|
|
if six.PY3:
|
|
from unittest import mock
|
|
else:
|
|
import mock
|
|
|
|
from socketio import middleware
|
|
|
|
|
|
class TestMiddleware(unittest.TestCase):
|
|
def test_wsgi_routing(self):
|
|
mock_wsgi_app = mock.MagicMock()
|
|
mock_sio_app = 'foo'
|
|
m = middleware.Middleware(mock_sio_app, mock_wsgi_app)
|
|
environ = {'PATH_INFO': '/foo'}
|
|
start_response = "foo"
|
|
m(environ, start_response)
|
|
mock_wsgi_app.assert_called_once_with(environ, start_response)
|
|
|
|
def test_sio_routing(self):
|
|
mock_wsgi_app = 'foo'
|
|
mock_sio_app = mock.Mock()
|
|
mock_sio_app.handle_request = mock.MagicMock()
|
|
m = middleware.Middleware(mock_sio_app, mock_wsgi_app)
|
|
environ = {'PATH_INFO': '/socket.io/'}
|
|
start_response = "foo"
|
|
m(environ, start_response)
|
|
mock_sio_app.handle_request.assert_called_once_with(environ,
|
|
start_response)
|
|
|
|
def test_404(self):
|
|
mock_wsgi_app = None
|
|
mock_sio_app = mock.Mock()
|
|
m = middleware.Middleware(mock_sio_app, mock_wsgi_app)
|
|
environ = {'PATH_INFO': '/foo/bar'}
|
|
start_response = mock.MagicMock()
|
|
r = m(environ, start_response)
|
|
self.assertEqual(r, ['Not Found'])
|
|
start_response.assert_called_once_with(
|
|
"404 Not Found", [('Content-type', 'text/plain')])
|
|
|