lex-bot-tester sample: AWS Lex Bot tester demonstration code

lex-bot-tester-sample.py

#! /usr/bin/env python
# -*- coding: utf-8 -*-
"""
    Lex Bot Tester
    Copyright (C) 2017  Diego Torres Milano

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
"""
import re
import unittest

from com.dtmilano.aws.lex.conversation import Conversation, ConversationItem
from com.dtmilano.aws.lex.lexbottest import LexBotTest
from com.dtmilano.aws.lex.lexmodelsclient import LexModelsClient
from com.dtmilano.aws.lex.lexruntimeclient import DialogState

RE_DATE = re.compile('\d+-\d+-\d+')
RE_WEEK = re.compile('\d+-w\d+')


class LexBotTestTests(LexBotTest):
    def test_conversations_text_order_flowers(self):
        bot_name = 'OrderFlowers'
        bot_alias = 'OrderFlowersLatest'
        user_id = 'ClientId'
        lmc = LexModelsClient(bot_name, bot_alias)
        conversations = []
        for i in lmc.get_intents_for_bot():
            r = lmc.get_result_class_for_intent(i)
            if i == 'OrderFlowers':
                conversations.append(Conversation(
                    ConversationItem('I would like to order some roses',
                                     r(DialogState.ELICIT_SLOT, flower_type='roses')),
                    ConversationItem('white', r(DialogState.ELICIT_SLOT, flower_type='roses', flower_color='white')),
                    ConversationItem('next Sunday',
                                     r(DialogState.ELICIT_SLOT, flower_type='roses', flower_color='white',
                                       pickup_date=RE_DATE)),
                    ConversationItem('noon', r(DialogState.CONFIRM_INTENT, flower_type='roses', flower_color='white',
                                               pickup_date=RE_DATE, pickup_time='12:00')),
                    ConversationItem('yes', r(DialogState.FULFILLED, flower_type='roses', flower_color='white',
                                              pickup_date=RE_DATE, pickup_time='12:00')),
                ))
            elif i == 'Cancel':
                conversations.append(Conversation(
                    ConversationItem('Cancel', r(DialogState.READY_FOR_FULFILLMENT))
                ))
        self.conversations_text(bot_name, bot_alias, user_id, conversations)

    def test_conversations_text_book_car(self):
        bot_name = 'BookTrip'
        bot_alias = 'BookTripLatest'
        user_id = 'ClientId'
        lmc = LexModelsClient(bot_name, bot_alias)
        conversations = []
        for i in lmc.get_intents_for_bot():
            r = lmc.get_result_class_for_intent(i)
            if i == 'BookCar':
                conversations.append(Conversation(
                    ConversationItem('book a car',
                                     r(DialogState.ELICIT_SLOT)),
                    ConversationItem('LA', r(DialogState.ELICIT_SLOT, pick_up_city='LA')),
                    ConversationItem('next week',
                                     r(DialogState.ELICIT_SLOT, pick_up_city='LA', pick_up_date=RE_WEEK)),
                    ConversationItem('a month from now',
                                     r(DialogState.ELICIT_SLOT, pick_up_city='LA', pick_up_date=RE_WEEK,
                                       return_date=RE_DATE)),
                    ConversationItem('25', r(DialogState.ELICIT_SLOT, pick_up_city='LA', pick_up_date=RE_WEEK,
                                             return_date=RE_DATE, driver_age='25')),
                    ConversationItem('economy', r(DialogState.CONFIRM_INTENT, pick_up_city='LA', pick_up_date=RE_WEEK,
                                                  return_date=RE_DATE, driver_age='25', car_type='economy')),
                    ConversationItem('yes',
                                     r(DialogState.READY_FOR_FULFILLMENT, pick_up_city='LA', pick_up_date=RE_WEEK,
                                       return_date=RE_DATE, driver_age='25', car_type='economy')),
                ))
            elif i == 'Cancel':
                conversations.append(Conversation(
                    ConversationItem('Cancel', r(DialogState.READY_FOR_FULFILLMENT))
                ))
            self.conversations_text(bot_name, bot_alias, user_id, conversations)


if __name__ == '__main__':
    unittest.main()