def test_context_id_not_None(self): product_content = Mock() client_handlers = {} user_info_cache = Mock() favorites_cache = Mock() target = Target(product_content=product_content, client_handlers=client_handlers, user_info_cache=user_info_cache, favorites_cache=favorites_cache) handler = Mock(name="new_client_handler") handler.id = "new_id" handler.context_id = "context_id_value" handler.context_rev = "context_rev_value" target.open(handler) target.post_context = Mock() self.assertEqual(0, target.post_context.call_count) self.assertEqual("context_rev_value", handler.context_rev) self.assertEqual("context_id_value", handler.context_id) self.assertEqual(1, handler.write_message.call_count) self.assertDictEqual( {'context_id': 'context_id_value', 'type': 'connection_opened'}, handler.write_message.call_args_list[0][0][0] )
def test_existing_id(self): client_handlers = { "existing_id": "existing_handler" } product_content = Mock() user_info_cache = Mock() favorites_cache = Mock() target = Target(product_content=product_content, client_handlers=client_handlers, user_info_cache=user_info_cache, favorites_cache=favorites_cache) target.get_context = Mock( return_value="get_context_value" ) target.post_context = Mock() handler = Mock(name="new_client_handler") handler.id = "existing_id" handler.context_id = "context_id" handler._context = None target.open(handler) self.assertTrue("existing_id" in client_handlers) self.assertEqual("existing_handler", client_handlers["existing_id"]) self.assertEqual("context_id", handler.context_id) self.assertIsNone(handler._context) self.assertEqual(0, target.post_context.call_count) self.assertEqual(1, handler.write_message.call_count) self.assertDictEqual( {'context_id': 'context_id', 'type': 'connection_opened'}, handler.write_message.call_args_list[0][0][0] )
def test_existing_id(self): client_handlers = {"existing_id": "existing_handler"} product_content = Mock() user_info_cache = Mock() favorites_cache = Mock() target = Target(product_content=product_content, client_handlers=client_handlers, user_info_cache=user_info_cache, favorites_cache=favorites_cache) target.get_context = Mock(return_value="get_context_value") target.post_context = Mock() handler = Mock(name="new_client_handler") handler.id = "existing_id" handler.context_id = "context_id" handler._context = None target.open(handler) self.assertTrue("existing_id" in client_handlers) self.assertEqual("existing_handler", client_handlers["existing_id"]) self.assertEqual("context_id", handler.context_id) self.assertIsNone(handler._context) self.assertEqual(0, target.post_context.call_count) self.assertEqual(1, handler.write_message.call_count) self.assertDictEqual( { 'context_id': 'context_id', 'type': 'connection_opened' }, handler.write_message.call_args_list[0][0][0])
def test_context_id_not_None(self): product_content = Mock() client_handlers = {} user_info_cache = Mock() favorites_cache = Mock() target = Target(product_content=product_content, client_handlers=client_handlers, user_info_cache=user_info_cache, favorites_cache=favorites_cache) handler = Mock(name="new_client_handler") handler.id = "new_id" handler.context_id = "context_id_value" handler.context_rev = "context_rev_value" target.open(handler) target.post_context = Mock() self.assertEqual(0, target.post_context.call_count) self.assertEqual("context_rev_value", handler.context_rev) self.assertEqual("context_id_value", handler.context_id) self.assertEqual(1, handler.write_message.call_count) self.assertDictEqual( { 'context_id': 'context_id_value', 'type': 'connection_opened' }, handler.write_message.call_args_list[0][0][0])
class WebSocket(WebSocketHandler): _param_extractor = None _cookie_extractor = None _logic = None user_id = None id = None application_id = None session_id = None context_id = None context_rev = None context = None suggest_id = None offset = None page_size = None def initialize(self, product_content, client_handlers, user_info_cache, favorites_cache): from api.logic.websocket import WebSocket as WebSocketLogic self._logic = WebSocketLogic( product_content=product_content, client_handlers=client_handlers, user_info_cache=user_info_cache, favorites_cache=favorites_cache, ) self._param_extractor = ParamExtractor(self) self._cookie_extractor = WebSocketCookieExtractor(self) def check_origin(self, origin): return True def open(self): self.id = uuid4() self.user_id = self._cookie_extractor.user_id() self.application_id = self._cookie_extractor.application_id() self.session_id = self._cookie_extractor.session_id() self.context_id = self._param_extractor.context_id() self.suggest_id = self._param_extractor.suggest_id() self.locale = self._param_extractor.locale() self.page_size = 20 self._logic.open(self) def on_message(self, message): self._logic.on_message(self, json_decode(message)) if self.user_id is None: # maybe they logged in self.user_id = self._cookie_extractor.user_id() def on_close(self): self._logic.on_close(self)
class WebSocket(WebSocketHandler): _param_extractor = None _cookie_extractor = None _logic = None user_id = None id = None application_id = None session_id = None context_id = None context_rev = None context = None suggest_id = None offset = None page_size = None def initialize(self, product_content, client_handlers, user_info_cache, favorites_cache): from api.logic.websocket import WebSocket as WebSocketLogic self._logic = WebSocketLogic(product_content=product_content, client_handlers=client_handlers, user_info_cache=user_info_cache, favorites_cache=favorites_cache) self._param_extractor = ParamExtractor(self) self._cookie_extractor = WebSocketCookieExtractor(self) def check_origin(self, origin): return True def open(self): self.id = uuid4() self.user_id = self._cookie_extractor.user_id() self.application_id = self._cookie_extractor.application_id() self.session_id = self._cookie_extractor.session_id() self.context_id = self._param_extractor.context_id() self.suggest_id = self._param_extractor.suggest_id() self.locale = self._param_extractor.locale() self.page_size = 20 self._logic.open(self) def on_message(self, message): self._logic.on_message(self, json_decode(message)) if self.user_id is None: # maybe they logged in self.user_id = self._cookie_extractor.user_id() def on_close(self): self._logic.on_close(self)
def test_context_id_None(self): product_content = Mock() client_handlers = {} user_info_cache = Mock() favorites_cache = Mock() target = Target(product_content=product_content, client_handlers=client_handlers, user_info_cache=user_info_cache, favorites_cache=favorites_cache) target.get_context = Mock() target.post_context = Mock(return_value=("context_id_value", "context_rev_value")) handler = Mock() handler.context_id = None handler.user_id = "user_id_value" handler.application_id = "application_id_value" handler.session_id = "session_id_value" handler.locale = "locale_value" target.open(handler) self.assertEqual(0, target.get_context.call_count) self.assertEqual(1, target.post_context.call_count) self.assertEqual("user_id_value", target.post_context.call_args_list[0][0][0]) self.assertEqual("application_id_value", target.post_context.call_args_list[0][0][1]) self.assertEqual("session_id_value", target.post_context.call_args_list[0][0][2]) self.assertEqual("locale_value", target.post_context.call_args_list[0][0][3]) self.assertEqual("context_id_value", handler.context_id) self.assertEqual("context_rev_value", handler.context_rev) self.assertEqual(1, handler.write_message.call_count) self.assertDictEqual( { 'context_id': 'context_id_value', 'type': 'connection_opened' }, handler.write_message.call_args_list[0][0][0])
def test_context_id_None(self): product_content = Mock() client_handlers = {} user_info_cache = Mock() favorites_cache = Mock() target = Target(product_content=product_content, client_handlers=client_handlers, user_info_cache=user_info_cache, favorites_cache=favorites_cache) target.get_context = Mock() target.post_context = Mock( return_value=("context_id_value", "context_rev_value") ) handler = Mock() handler.context_id = None handler.user_id = "user_id_value" handler.application_id = "application_id_value" handler.session_id = "session_id_value" handler.locale = "locale_value" target.open(handler) self.assertEqual(0, target.get_context.call_count) self.assertEqual(1, target.post_context.call_count) self.assertEqual("user_id_value", target.post_context.call_args_list[0][0][0]) self.assertEqual("application_id_value", target.post_context.call_args_list[0][0][1]) self.assertEqual("session_id_value", target.post_context.call_args_list[0][0][2]) self.assertEqual("locale_value", target.post_context.call_args_list[0][0][3]) self.assertEqual("context_id_value", handler.context_id) self.assertEqual("context_rev_value", handler.context_rev) self.assertEqual(1, handler.write_message.call_count) self.assertDictEqual( {'context_id': 'context_id_value', 'type': 'connection_opened'}, handler.write_message.call_args_list[0][0][0] )