def test_found(self):
client_handlers = {
"existing_context_id": {
"existing_handler_id": "existing_handler"
},
"random_context_id": {
"random_handler_id": "random_handler"
}
}
product_content = Mock()
user_info_cache = Mock()
favorites_cache = Mock()
target = Target(
client_handlers=client_handlers,
product_content=product_content,
user_info_cache=user_info_cache,
favorites_cache=favorites_cache
)
handler = Mock(name="new_client_handler")
handler.context_id = "existing_context_id"
handler.id = "existing_handler_id"
target.on_close(handler)
self.assertDictEqual(
{
'existing_context_id': {},
'random_context_id': {'random_handler_id': 'random_handler'}
},
client_handlers
)
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_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_found(self):
client_handlers = {
"existing_context_id": {
"existing_handler_id": "existing_handler"
},
"random_context_id": {
"random_handler_id": "random_handler"
}
}
product_content = Mock()
user_info_cache = Mock()
favorites_cache = Mock()
target = Target(client_handlers=client_handlers,
product_content=product_content,
user_info_cache=user_info_cache,
favorites_cache=favorites_cache)
handler = Mock(name="new_client_handler")
handler.context_id = "existing_context_id"
handler.id = "existing_handler_id"
target.on_close(handler)
self.assertDictEqual(
{
'existing_context_id': {},
'random_context_id': {
'random_handler_id': 'random_handler'
}
}, client_handlers)
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 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 test_unknown_message_type(self):
product_content = Mock()
client_handlers = 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.on_new_message = Mock()
target.on_next_page_message = Mock()
target.on_view_product_details_message = Mock()
self.assertRaises(Exception, target.on_message, {"type": "unknown"})
self.assertEqual(0, target.on_new_message.call_count)
self.assertEqual(0, target.on_next_page_message.call_count)
self.assertEqual(0, target.on_view_product_details_message.call_count)
def test_no_message_type(self):
product_content = Mock()
client_handlers = 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.on_new_message = Mock()
target.on_next_page_message = Mock()
target.on_view_product_details_message = Mock()
self.assertRaises(
Exception,
target.on_message,
{}
)
self.assertEqual(0, target.on_new_message.call_count)
self.assertEqual(0, target.on_next_page_message.call_count)
self.assertEqual(0, target.on_view_product_details_message.call_count)
def test_regular(self):
handler = Mock(name="handler_value")
handler.context_id = "context_id_value"
handler.user_id = "user_id_value"
handler.application_id = "application_id_value"
handler.session_id = "session_id_value"
handler.locale = "locale_value"
handler.suggest_id = "suggest_id_value"
handler.context_rev = "old_rev"
product_content = Mock()
client_handlers = 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.post_context_feedback = Mock(return_value="new_rev")
target.on_view_product_details_message(handler, {
"product_id": "product_id_value",
"feedback_type": "type_value"
})
self.assertEqual(1, target.post_context_feedback.call_count)
self.assertEqual("context_id_value",
target.post_context_feedback.call_args_list[0][0][0])
self.assertEqual("user_id_value",
target.post_context_feedback.call_args_list[0][0][1])
self.assertEqual("application_id_value",
target.post_context_feedback.call_args_list[0][0][2])
self.assertEqual("session_id_value",
target.post_context_feedback.call_args_list[0][0][3])
self.assertEqual("product_id_value",
target.post_context_feedback.call_args_list[0][0][4])
self.assertEqual("type_value",
target.post_context_feedback.call_args_list[0][0][5])
self.assertIsNone(target.post_context_feedback.call_args_list[0][0][6])
self.assertEqual("context_id_value", handler.context_id)
self.assertEqual("suggest_id_value", handler.suggest_id)
self.assertEqual("new_rev", handler.context_rev)
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 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 test_view_product_message(self):
product_content = Mock()
client_handlers = 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.on_new_message = Mock()
target.on_next_page_message = Mock()
target.on_view_product_details_message = Mock()
target.on_message("handler_value", {"type": "view_product_details"})
self.assertEqual(0, target.on_new_message.call_count)
self.assertEqual(0, target.on_next_page_message.call_count)
self.assertEqual(1, target.on_view_product_details_message.call_count)
self.assertEqual(
"handler_value",
target.on_view_product_details_message.call_args_list[0][0][0])
self.assertDictEqual(
{"type": "view_product_details"},
target.on_view_product_details_message.call_args_list[0][0][1])
def test_view_product_message(self):
product_content = Mock()
client_handlers = 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.on_new_message = Mock()
target.on_next_page_message = Mock()
target.on_view_product_details_message = Mock()
target.on_message(
"handler_value",
{
"type": "view_product_details"
}
)
self.assertEqual(0, target.on_new_message.call_count)
self.assertEqual(0, target.on_next_page_message.call_count)
self.assertEqual(1, target.on_view_product_details_message.call_count)
self.assertEqual("handler_value", target.on_view_product_details_message.call_args_list[0][0][0])
self.assertDictEqual(
{
"type": "view_product_details"
},
target.on_view_product_details_message.call_args_list[0][0][1]
)
def test_regular(self):
handler = Mock(name="handler_value")
handler.context_id = "context_id_value"
handler.user_id = "user_id_value"
handler.application_id = "application_id_value"
handler.session_id = "session_id_value"
handler.locale = "locale_value"
handler.suggest_id = "suggest_id_value"
handler.context_rev = "old_rev"
product_content = Mock()
client_handlers = 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.post_context_feedback = Mock(
return_value="new_rev"
)
target.on_view_product_details_message(
handler,
{
"product_id": "product_id_value",
"feedback_type": "type_value"
}
)
self.assertEqual(1, target.post_context_feedback.call_count)
self.assertEqual("context_id_value", target.post_context_feedback.call_args_list[0][0][0])
self.assertEqual("user_id_value", target.post_context_feedback.call_args_list[0][0][1])
self.assertEqual("application_id_value", target.post_context_feedback.call_args_list[0][0][2])
self.assertEqual("session_id_value", target.post_context_feedback.call_args_list[0][0][3])
self.assertEqual("product_id_value", target.post_context_feedback.call_args_list[0][0][4])
self.assertEqual("type_value", target.post_context_feedback.call_args_list[0][0][5])
self.assertIsNone(target.post_context_feedback.call_args_list[0][0][6])
self.assertEqual("context_id_value", handler.context_id)
self.assertEqual("suggest_id_value", handler.suggest_id)
self.assertEqual("new_rev", handler.context_rev)
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]
)
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_new_message(self):
product_content = Mock()
client_handlers = 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.new_message_handler = MagicMock()
target.on_next_page_message = Mock()
target.on_view_product_details_message = Mock()
target.on_message("handler_value", {"type": "new_message"})
target.new_message_handler.on_new_message.assert_called_once_with(
'handler_value', {'type': 'new_message'}, new_conversation=False)
self.assertEqual(0, target.on_next_page_message.call_count)
self.assertEqual(0, target.on_view_product_details_message.call_count)
def test_home_page_message(self):
product_content = Mock()
client_handlers = 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.new_message_handler = MagicMock()
target.on_next_page_message = Mock()
target.on_view_product_details_message = Mock()
target.on_message(
"handler_value",
{
"type": "home_page_message"
}
)
target.new_message_handler.on_new_message.assert_called_once_with('handler_value', {'type': 'home_page_message'}, new_conversation=True)
self.assertEqual(0, target.on_next_page_message.call_count)
self.assertEqual(0, target.on_view_product_details_message.call_count)
