def test_get_protocol_existing(self):
from ptah.sockjs.session import Session
s = Session('12345', request=self.request)
p = object()
s.protocols['test'] = p
self.assertIs(s.get_protocol('test'), p)
def test_get_protocol_unknown(self):
from ptah.sockjs.session import Session
s = Session('12345', request=self.request)
self.assertIsNone(s.get_protocol('unknown'))
