def test_adding_days_default(original_message, processed_message):
response = create_text_response(original_message)
assert response.query_result.fulfillment_messages[0].text.text[
0] == original_message
MessageHandler.add_weekday(response)
assert response.query_result.fulfillment_messages[0].text.text[
0] == processed_message
def test_strip_seconds(original_message, processed_message):
response = create_text_response(original_message)
assert response.query_result.fulfillment_messages[0].text.text[
0] == original_message
MessageHandler.strip_seconds(response)
assert response.query_result.fulfillment_messages[0].text.text[
0] == processed_message
def test_reformat_date(original_message, processed_message):
response = create_text_response(original_message)
assert response.query_result.fulfillment_messages[0].text.text[
0] == original_message
MessageHandler.reformat_date(response)
assert response.query_result.fulfillment_messages[0].text.text[
0] == processed_message
def reformat_text_in_intent(
response: session_pb2.DetectIntentResponse,
nlu_client: Client) -> session_pb2.DetectIntentResponse:
return MessageHandler.substitute_pattern(
pattern="<REPLACE:REPLACE_THIS_TEXT>",
replace="new text",
response=response)
def process_messages(
self,
response: session_pb2.DetectIntentResponse,
) -> session_pb2.DetectIntentResponse:
for j, message in enumerate(
response.query_result.fulfillment_messages):
found_triggers = MessageHandler.get_triggers(
message, get_session_from_response(response))
for found_trigger in found_triggers:
new_response: Optional[session_pb2.DetectIntentResponse] = \
self.trigger_handlers[found_trigger](response, message, found_trigger, found_triggers)
if new_response:
if not new_response.response_id == response.response_id:
return new_response
for found_trigger in found_triggers:
SingleMessageHandler.substitute_pattern_in_message(
message, found_trigger, "")
self.quicksend_to_api(response, message, j)
if not len(response.query_result.fulfillment_messages):
self.quicksend_to_api(response, None, 0)
return response
def test_replace_text_in_response(original_message, search, replace,
processed_message) -> None:
response = create_text_response(original_message)
response = MessageHandler.substitute_pattern(search, replace, response)
assert response.query_result.fulfillment_messages[0].text.text[
0] == processed_message
def test_remove_triggers_from_reponse(original_message,
processed_message) -> None:
response = create_text_response(original_message)
response = MessageHandler.remove_triggers_from_reponse(response)
assert response.query_result.fulfillment_messages[0].text.text[
0] == processed_message
def test_check_response_for_triggers(original_message,
expected_trigger) -> None:
response = create_text_response(original_message)
trigger = MessageHandler.check_for_triggers(response)
assert trigger[0] == expected_trigger[0]
assert trigger[1] == expected_trigger[1]