forked from pool/python-ws4py
4058aa1d21
- update to version 0.5.1 * fixed runtime error: Set changed size during iteration * on secure, only pass the requested number of bytes to the parsers * Change threaded client test to test ssl socket * exclude certain headers when requested * Disable build for Python 3.4 - launch tests using multibuild - added patch ws4py-no-cherrypy-test.patch to enable tests also on Python2.7 (since CherryPy does not exist for Python2) OBS-URL: https://build.opensuse.org/request/show/693231 OBS-URL: https://build.opensuse.org/package/show/devel:languages:python/python-ws4py?expand=0&rev=5
112 lines
3.3 KiB
Diff
112 lines
3.3 KiB
Diff
Index: ws4py-0.5.1/test/test_cherrypy.py
|
|
===================================================================
|
|
--- ws4py-0.5.1.orig/test/test_cherrypy.py
|
|
+++ /dev/null
|
|
@@ -1,106 +0,0 @@
|
|
-# -*- coding: utf-8 -*-
|
|
-import os
|
|
-import socket
|
|
-import time
|
|
-import unittest
|
|
-
|
|
-from mock import MagicMock, call
|
|
-
|
|
-import cherrypy
|
|
-from ws4py.server.cherrypyserver import WebSocketPlugin, WebSocketTool
|
|
-from ws4py.websocket import EchoWebSocket
|
|
-from ws4py.framing import Frame, OPCODE_TEXT, OPCODE_CLOSE
|
|
-
|
|
-class FakePoller(object):
|
|
- def __init__(self, timeout=0.1):
|
|
- self._fds = []
|
|
-
|
|
- def release(self):
|
|
- self._fds = []
|
|
-
|
|
- def register(self, fd):
|
|
- if fd not in self._fds:
|
|
- self._fds.append(fd)
|
|
-
|
|
- def unregister(self, fd):
|
|
- if fd in self._fds:
|
|
- self._fds.remove(fd)
|
|
-
|
|
- def poll(self):
|
|
- return self._fds
|
|
-
|
|
-class App(object):
|
|
- @cherrypy.expose
|
|
- def ws(self):
|
|
- assert cherrypy.request.ws_handler != None
|
|
-
|
|
-def setup_engine():
|
|
- # we don't need a HTTP server for this test
|
|
- cherrypy.server.unsubscribe()
|
|
-
|
|
- cherrypy.config.update({'log.screen': False})
|
|
-
|
|
- cherrypy.engine.websocket = WebSocketPlugin(cherrypy.engine)
|
|
- cherrypy.engine.websocket.subscribe()
|
|
-
|
|
- cherrypy.engine.websocket.manager.poller = FakePoller()
|
|
-
|
|
- cherrypy.tools.websocket = WebSocketTool()
|
|
-
|
|
- config={'/ws': {'tools.websocket.on': True,
|
|
- 'tools.websocket.handler_cls': EchoWebSocket}}
|
|
- cherrypy.tree.mount(App(), '/', config)
|
|
- cherrypy.engine.start()
|
|
-
|
|
-def teardown_engine():
|
|
- cherrypy.engine.exit()
|
|
-
|
|
-class CherryPyTest(unittest.TestCase):
|
|
- def setUp(self):
|
|
- setup_engine()
|
|
-
|
|
- def tearDown(self):
|
|
- teardown_engine()
|
|
-
|
|
- def test_plugin(self):
|
|
- manager = cherrypy.engine.websocket.manager
|
|
- self.assertEqual(len(manager), 0)
|
|
-
|
|
- s = MagicMock(spec=socket.socket)
|
|
- s.recv.return_value = Frame(opcode=OPCODE_TEXT, body=b'hello',
|
|
- fin=1, masking_key=os.urandom(4)).build()
|
|
- h = EchoWebSocket(s, [], [])
|
|
- cherrypy.engine.publish('handle-websocket', h, ('127.0.0.1', 0))
|
|
- self.assertEqual(len(manager), 1)
|
|
- self.assertTrue(h in manager)
|
|
-
|
|
- # the following call to .close() on the
|
|
- # websocket object will initiate
|
|
- # the closing handshake
|
|
- # This next line mocks the response
|
|
- # from the client to actually
|
|
- # complete the handshake.
|
|
- # The manager will then remove the websocket
|
|
- # from its pool
|
|
- s.recv.return_value = Frame(opcode=OPCODE_CLOSE, body=b"ok we're done",
|
|
- fin=1, masking_key=os.urandom(4)).build()
|
|
- h.close()
|
|
-
|
|
- # the poller runs a thread, give it time to get there
|
|
- # just wait up to 5 seconds.
|
|
- left_iteration = 50
|
|
- while left_iteration:
|
|
- left_iteration -= 1
|
|
- time.sleep(.1)
|
|
- if len(manager) == 0:
|
|
- break
|
|
-
|
|
- self.assertEqual(len(manager), 0)
|
|
-
|
|
-if __name__ == '__main__':
|
|
- suite = unittest.TestSuite()
|
|
- loader = unittest.TestLoader()
|
|
- for testcase in [CherryPyTest]:
|
|
- tests = loader.loadTestsFromTestCase(testcase)
|
|
- suite.addTests(tests)
|
|
- unittest.TextTestRunner(verbosity=2).run(suite)
|