def test_regular(self): sender = MagicMock() context = MagicMock() suggest = MagicMock() target = Target(sender, context, suggest) target.json_decode = MagicMock(return_value={"_rev": "context_revision_value"}) response = Mock() response.headers = {"_id": "suggest_id_value"} handler = MagicMock() target.post_suggest_callback(response, handler, "message_value") suggest.write_new_suggestion.assert_called_once_with(handler) self.assertEqual("suggest_id_value", handler.suggest_id)
def test_regular(self): sender = Mock() context = Mock() suggest = Mock() target = Target(sender, context, suggest) handler = Mock() handler.user_id = "user_id_value" handler.application_id = "application_id_value" handler.session_id = "session_id_value" handler.locale = "locale_value" target.on_new_message_empty(handler, "message_value") self.assertEqual(1, context.get_context.call_count) self.assertEqual(handler, context.get_context.call_args_list[0][0][0])
def test_regular_has_entities(self): sender = MagicMock() context = MagicMock() suggest = Mock() target = Target(sender, context, suggest) target.json_decode = MagicMock( return_value={ "entities": [ { "source": "non_detection" } ], "_rev": "_context_rev_value" } ) handler = MagicMock() handler.user_id = "user_id_value" handler.application_id = "application_id_value" handler.session_id = "session_id_value" handler.locale = "locale_value" response = MagicMock() response.body = "response_value" target.get_context_callback(response, handler, "message_value") self.assertEqual(1, suggest.post_suggest.call_count) self.assertEqual("user_id_value", suggest.post_suggest.call_args_list[0][0][0]) self.assertEqual("application_id_value", suggest.post_suggest.call_args_list[0][0][1]) self.assertEqual("session_id_value", suggest.post_suggest.call_args_list[0][0][2]) self.assertEqual("locale_value", suggest.post_suggest.call_args_list[0][0][3]) self.assertDictEqual({'_rev': '_context_rev_value', 'entities': [{'source': 'non_detection'}]}, suggest.post_suggest.call_args_list[0][0][4]) target.json_decode.assert_called_once_with('response_value') self.assertDictEqual({'_rev': '_context_rev_value', 'entities': [{'source': 'non_detection'}]}, handler.context) self.assertEqual("_context_rev_value", handler.context_rev)