def test_custom_session_store(self): sessions.session_store = TestSessionStore session_store = sessions.get_session_store()(self.connection) session_store.set('key', 'val') self.assertEqual(session_store.get('key'), 'val')
from swampdragon import route_handler from swampdragon.pubsub_providers.subscriber_factory import get_subscription_provider from swampdragon.sessions.sessions import get_session_store import json import uuid pubsub = get_subscription_provider() session_store = get_session_store() class TestSession(object): def __init__(self): self.session_id = str(uuid.uuid4().hex) class TestConnection(object): uid = None channels = [] def __init__(self, user=None): self.uid = str(uuid.uuid4().hex) self.session = TestSession() self.user = user self.sent_data = [] self.published_data = [] self.pub_sub = pubsub self.session_store = session_store(self) def send(self, message):
from swampdragon import route_handler from swampdragon.pubsub_providers.subscriber_factory import get_subscription_provider from swampdragon.sessions.sessions import get_session_store import json import uuid pubsub = get_subscription_provider() session_store = get_session_store() class TestSession(object): def __init__(self): self.session_id = str(uuid.uuid4().hex) class TestConnection(object): uid = None channels = [] def __init__(self, user=None): self.uid = str(uuid.uuid4().hex) self.session = TestSession() self.user = user self.sent_data = [] self.published_data = [] self.pub_sub = pubsub self.session_store = session_store(self) def send(self, message): self.sent_data.append(json.dumps(message))