def setUp(self): super(SockJSSubscriberTestCase, self).setUp() self.publisher = self._new_client() self.subscriber = SockJSSubscriber(self.client)
from tornado.web import StaticFileHandler, FallbackHandler, RedirectHandler,\ Application from tornado.wsgi import WSGIContainer from sockjs.tornado import SockJSConnection, SockJSRouter from tornadoredis import Client as RedisClient from tornadoredis.pubsub import SockJSSubscriber from .. import config toolname = config.toolname from .oauth_handler import app as oauth_wsgi # noqa from ..celery import app as celery_app # noqa # Create the tornadoredis.Client instance # and use it for redis channel subscriptions subscriber = SockJSSubscriber( RedisClient(host=config.redis_server, port=config.redis_port, selected_db=config.redis_db)) class SockConnection(SockJSConnection): def on_open(self, info): self.channel = str(uuid4()) subscriber.subscribe(self.channel, self) def on_close(self): subscriber.unsubscribe(self.channel, self) def on_message(self, message): data = json.loads(message) data['redis_channel'] = self.channel if data['action'] == 'watchlist':