from pprint import pprint

import order_flower_intent
from pylexo import LexInputEvent, LexOutputResponse, CloseLexOutputResponse, ElicitIntentOutputResponse, \
    ConfirmIntentOutputResponse, ElicitSlotOutputResponse, runtime_response
import json


orderFlowerJson = """
        {
            "messageVersion": "1.0",
            "invocationSource": "DialogCodeHook",
            "userId": "user_123",
            "sessionAttributes": {
                "RequestorCity": "Portland"
            },
            "bot": {
                "name": "OrderFlowers",
                "alias": null,
                "version": "$LATEST"
            },
            "outputDialogMode": "Text",
            "currentIntent": {
                "name": "OrderFlowers",
                "slots": {
                    "PickupTime": null,
                    "FlowerType": "Roses",
                    "PickupDate": null
                },
                "confirmationStatus": "None"
            }
        }
    """
GENERIC_INTENT_LEX_INPUT = LexInputEvent(json.loads(orderFlowerJson))
ORDER_FLOWER_INTENT_LEX_INPUT = order_flower_intent.LexInputEvent(json.loads(orderFlowerJson))



def test_generic():
    event = GENERIC_INTENT_LEX_INPUT
    d = event.to_primitive()
    assert event.messageVersion == '1.0'
    assert event.invocationSource == 'DialogCodeHook'
    assert event.userId == 'user_123'
    # assert event.sessionAttributes == {}
    assert event.bot.name == 'OrderFlowers'
    assert event.bot.alias is None
    assert event.bot.version == '$LATEST'
    assert event.outputDialogMode == 'Text'
    assert event.currentIntent.name == 'OrderFlowers'
    assert event.currentIntent.confirmationStatus == 'None'
    assert event.currentIntent.slots['PickupTime'] is None
    assert event.currentIntent.slots['FlowerType'] == 'Roses'
    assert event.currentIntent.slots['PickupDate'] is None

    event_dict = event.to_primitive()
    assert event_dict['messageVersion'] == '1.0'
    assert event_dict['invocationSource'] == 'DialogCodeHook'
    assert event_dict['userId'] == 'user_123'
    assert event_dict['sessionAttributes'] == {
        "RequestorCity": "Portland"
    }
    assert event_dict['bot']['name'] == 'OrderFlowers'
    assert event_dict['bot']['alias'] is None
    assert event_dict['bot']['version'] == '$LATEST'
    assert event_dict['outputDialogMode'] == 'Text'
    assert event_dict['currentIntent']['name'] == 'OrderFlowers'
    assert event_dict['currentIntent']['confirmationStatus'] == 'None'
    assert event_dict['currentIntent']['slots']['PickupTime'] is None
    assert event_dict['currentIntent']['slots']['FlowerType'] == 'Roses'
    assert event_dict['currentIntent']['slots']['PickupDate'] is None

def test_OrderFlower():
    event = ORDER_FLOWER_INTENT_LEX_INPUT
    assert event.messageVersion == '1.0'
    assert event.invocationSource == 'DialogCodeHook'
    assert event.userId == 'user_123'
    #assert event.sessionAttributes == {}
    assert event.bot.name == 'OrderFlowers'
    assert event.bot.alias is None
    assert event.bot.version == '$LATEST'
    assert event.outputDialogMode == 'Text'
    assert event.currentIntent.name == 'OrderFlowers'
    assert event.currentIntent.confirmationStatus == 'None'
    assert event.currentIntent.slots.PickupTime is None
    assert event.currentIntent.slots.FlowerType == 'Roses'
    assert event.currentIntent.slots.PickupDate is None

    event_dict = event.to_primitive()
    assert event_dict['messageVersion'] == '1.0'
    assert event_dict['invocationSource'] == 'DialogCodeHook'
    assert event_dict['userId'] == 'user_123'
    assert event_dict['sessionAttributes'] == {
        "RequestorCity": "Portland"
    }
    assert event_dict['bot']['name'] == 'OrderFlowers'
    assert event_dict['bot']['alias'] is None
    assert event_dict['bot']['version'] == '$LATEST'
    assert event_dict['outputDialogMode'] == 'Text'
    assert event_dict['currentIntent']['name'] == 'OrderFlowers'
    assert event_dict['currentIntent']['confirmationStatus'] == 'None'
    assert event_dict['currentIntent']['slots']['PickupTime'] is None
    assert event_dict['currentIntent']['slots']['FlowerType'] == 'Roses'
    assert event_dict['currentIntent']['slots']['PickupDate'] is None


def test_flower_confirm_intent():
    confirmIntent = order_flower_intent.ConfirmIntentOutputResponse()
    confirmIntent.update_from_input(ORDER_FLOWER_INTENT_LEX_INPUT)
    assert confirmIntent.dialogAction.intentName == 'OrderFlowers'
    assert ORDER_FLOWER_INTENT_LEX_INPUT.sessionAttributes.RequestorCity == confirmIntent.sessionAttributes.RequestorCity
    assert dict(confirmIntent.sessionAttributes) == dict(ORDER_FLOWER_INTENT_LEX_INPUT.sessionAttributes)
    assert confirmIntent.dialogAction.slots.to_primitive() == ORDER_FLOWER_INTENT_LEX_INPUT.currentIntent.slots.to_primitive()


def test_confirm_intent():
    confirmIntent = ConfirmIntentOutputResponse()
    confirmIntent.update_slots(GENERIC_INTENT_LEX_INPUT)
    assert confirmIntent.dialogAction.slots.to_primitive() == GENERIC_INTENT_LEX_INPUT.currentIntent.slots


def test_lex_close_response():
    response = CloseLexOutputResponse()

    d = response.to_primitive()
    pprint(d)
    keys = d['dialogAction'].keys()
    assert 'message' not in keys
    assert 'responseCard' not in keys
    response.dialogAction.message.content = "Hello"
    pprint(response.to_primitive())

def test_lex_elicit_response():
    response = ElicitIntentOutputResponse()
    d = response.to_primitive()
    pprint(d)
    keys = d['dialogAction'].keys()
    assert 'message' not in keys
    assert 'responseCard' not in keys


def test_dynamic_output_slots_class():
    event = order_flower_intent.LexInputEvent(json.loads(orderFlowerJson))
    EllicitSlotOrderFlowerOutputResponse = ElicitSlotOutputResponse.create_class(order_flower_intent.SlotsProperty)
    response = EllicitSlotOrderFlowerOutputResponse()
    response.update_slots(event)
    assert response.dialogAction.slots.FlowerType == 'Roses'
    response.dialogAction.slots.FlowerType = 'Pink Rose'
    response_dict = response.to_primitive()
    pprint(response_dict)
    assert response_dict['dialogAction']['slots']['FlowerType'] == 'Pink Rose'

def test_output_classes():
    event = order_flower_intent.LexInputEvent(json.loads(orderFlowerJson))
    response = order_flower_intent.ConfirmIntentOutputResponse()
    response.update_slots(event)
    assert response.dialogAction.slots.FlowerType == 'Roses'


def test_slots_filled():
    assert ORDER_FLOWER_INTENT_LEX_INPUT.is_all_slots_filled() == False


def test_post_text():
    response = {
        'intentName': 'string',
        'slots': {
            'string': 'string'
        },
        'sessionAttributes': {
            'string': 'string'
        },
        'message': 'string',
        'messageFormat': 'PlainText',
        'dialogState': 'Fulfilled',
        'slotToElicit': 'string',
        'responseCard': {
            'version': 'string',
            'contentType': 'application/vnd.amazonaws.card.generic',
            'genericAttachments': [
                {
                    'title': 'string',
                    'subTitle': 'string',
                    'attachmentLinkUrl': 'string',
                    'imageUrl': 'string',
                    'buttons': [
                        {
                            'text': 'string',
                            'value': 'string'
                        },
                    ]
                },
            ]
        }
    }
    response_o = runtime_response.LexPostTextResponse(response)
    assert response_o.is_fulfilled()
    assert response_o.is_plain_text()
    assert not response_o.is_custom_payload()
    assert not response_o.is_composite()
    assert not response_o.is_confirm_intent()
    assert not response_o.is_elicit_intent()
    assert not response_o.is_elicit_slot()
    assert not response_o.is_failed()

from pprint import pprint

import order_flower_intent
from pylexo import LexInputEvent, LexOutputResponse, CloseLexOutputResponse, ElicitIntentOutputResponse, \
    ConfirmIntentOutputResponse, ElicitSlotOutputResponse, runtime_response
import json


orderFlowerJson = """
        {
            "messageVersion": "1.0",
            "invocationSource": "DialogCodeHook",
            "userId": "user_123",
            "sessionAttributes": {
                "RequestorCity": "Portland"
            },
            "bot": {
                "name": "OrderFlowers",
                "alias": null,
                "version": "$LATEST"
            },
            "outputDialogMode": "Text",
            "currentIntent": {
                "name": "OrderFlowers",
                "slots": {
                    "PickupTime": null,
                    "FlowerType": "Roses",
                    "PickupDate": null
                },
                "confirmationStatus": "None"
            }
        }
    """
GENERIC_INTENT_LEX_INPUT = LexInputEvent(json.loads(orderFlowerJson))
ORDER_FLOWER_INTENT_LEX_INPUT = order_flower_intent.LexInputEvent(json.loads(orderFlowerJson))



def test_generic():
    event = GENERIC_INTENT_LEX_INPUT
    d = event.to_primitive()
    assert event.messageVersion == '1.0'
    assert event.invocationSource == 'DialogCodeHook'
    assert event.userId == 'user_123'
    # assert event.sessionAttributes == {}
    assert event.bot.name == 'OrderFlowers'
    assert event.bot.alias is None
    assert event.bot.version == '$LATEST'
    assert event.outputDialogMode == 'Text'
    assert event.currentIntent.name == 'OrderFlowers'
    assert event.currentIntent.confirmationStatus == 'None'
    assert event.currentIntent.slots['PickupTime'] is None
    assert event.currentIntent.slots['FlowerType'] == 'Roses'
    assert event.currentIntent.slots['PickupDate'] is None

    event_dict = event.to_primitive()
    assert event_dict['messageVersion'] == '1.0'
    assert event_dict['invocationSource'] == 'DialogCodeHook'
    assert event_dict['userId'] == 'user_123'
    assert event_dict['sessionAttributes'] == {
        "RequestorCity": "Portland"
    }
    assert event_dict['bot']['name'] == 'OrderFlowers'
    assert event_dict['bot']['alias'] is None
    assert event_dict['bot']['version'] == '$LATEST'
    assert event_dict['outputDialogMode'] == 'Text'
    assert event_dict['currentIntent']['name'] == 'OrderFlowers'
    assert event_dict['currentIntent']['confirmationStatus'] == 'None'
    assert event_dict['currentIntent']['slots']['PickupTime'] is None
    assert event_dict['currentIntent']['slots']['FlowerType'] == 'Roses'
    assert event_dict['currentIntent']['slots']['PickupDate'] is None

def test_OrderFlower():
    event = ORDER_FLOWER_INTENT_LEX_INPUT
    assert event.messageVersion == '1.0'
    assert event.invocationSource == 'DialogCodeHook'
    assert event.userId == 'user_123'
    #assert event.sessionAttributes == {}
    assert event.bot.name == 'OrderFlowers'
    assert event.bot.alias is None
    assert event.bot.version == '$LATEST'
    assert event.outputDialogMode == 'Text'
    assert event.currentIntent.name == 'OrderFlowers'
    assert event.currentIntent.confirmationStatus == 'None'
    assert event.currentIntent.slots.PickupTime is None
    assert event.currentIntent.slots.FlowerType == 'Roses'
    assert event.currentIntent.slots.PickupDate is None

    event_dict = event.to_primitive()
    assert event_dict['messageVersion'] == '1.0'
    assert event_dict['invocationSource'] == 'DialogCodeHook'
    assert event_dict['userId'] == 'user_123'
    assert event_dict['sessionAttributes'] == {
        "RequestorCity": "Portland"
    }
    assert event_dict['bot']['name'] == 'OrderFlowers'
    assert event_dict['bot']['alias'] is None
    assert event_dict['bot']['version'] == '$LATEST'
    assert event_dict['outputDialogMode'] == 'Text'
    assert event_dict['currentIntent']['name'] == 'OrderFlowers'
    assert event_dict['currentIntent']['confirmationStatus'] == 'None'
    assert event_dict['currentIntent']['slots']['PickupTime'] is None
    assert event_dict['currentIntent']['slots']['FlowerType'] == 'Roses'
    assert event_dict['currentIntent']['slots']['PickupDate'] is None


def test_flower_confirm_intent():
    confirmIntent = order_flower_intent.ConfirmIntentOutputResponse()
    confirmIntent.update_from_input(ORDER_FLOWER_INTENT_LEX_INPUT)
    assert confirmIntent.dialogAction.intentName == 'OrderFlowers'
    assert ORDER_FLOWER_INTENT_LEX_INPUT.sessionAttributes.RequestorCity == confirmIntent.sessionAttributes.RequestorCity
    assert dict(confirmIntent.sessionAttributes) == dict(ORDER_FLOWER_INTENT_LEX_INPUT.sessionAttributes)
    assert confirmIntent.dialogAction.slots.to_primitive() == ORDER_FLOWER_INTENT_LEX_INPUT.currentIntent.slots.to_primitive()


def test_confirm_intent():
    confirmIntent = ConfirmIntentOutputResponse()
    confirmIntent.update_slots(GENERIC_INTENT_LEX_INPUT)
    assert confirmIntent.dialogAction.slots.to_primitive() == GENERIC_INTENT_LEX_INPUT.currentIntent.slots


def test_lex_close_response():
    response = CloseLexOutputResponse()

    d = response.to_primitive()
    pprint(d)
    keys = d['dialogAction'].keys()
    assert 'message' not in keys
    assert 'responseCard' not in keys
    response.dialogAction.message.content = "Hello"
    pprint(response.to_primitive())

def test_lex_elicit_response():
    response = ElicitIntentOutputResponse()
    d = response.to_primitive()
    pprint(d)
    keys = d['dialogAction'].keys()
    assert 'message' not in keys
    assert 'responseCard' not in keys


def test_dynamic_output_slots_class():
    event = order_flower_intent.LexInputEvent(json.loads(orderFlowerJson))
    EllicitSlotOrderFlowerOutputResponse = ElicitSlotOutputResponse.create_class(order_flower_intent.SlotsProperty)
    response = EllicitSlotOrderFlowerOutputResponse()
    response.update_slots(event)
    assert response.dialogAction.slots.FlowerType == 'Roses'
    response.dialogAction.slots.FlowerType = 'Pink Rose'
    response_dict = response.to_primitive()
    pprint(response_dict)
    assert response_dict['dialogAction']['slots']['FlowerType'] == 'Pink Rose'

def test_output_classes():
    event = order_flower_intent.LexInputEvent(json.loads(orderFlowerJson))
    response = order_flower_intent.ConfirmIntentOutputResponse()
    response.update_slots(event)
    assert response.dialogAction.slots.FlowerType == 'Roses'


def test_slots_filled():
    assert ORDER_FLOWER_INTENT_LEX_INPUT.is_all_slots_filled() == False


def test_post_text():
    response = {
        'intentName': 'string',
        'slots': {
            'string': 'string'
        },
        'sessionAttributes': {
            'string': 'string'
        },
        'message': 'string',
        'messageFormat': 'PlainText',
        'dialogState': 'Fulfilled',
        'slotToElicit': 'string',
        'responseCard': {
            'version': 'string',
            'contentType': 'application/vnd.amazonaws.card.generic',
            'genericAttachments': [
                {
                    'title': 'string',
                    'subTitle': 'string',
                    'attachmentLinkUrl': 'string',
                    'imageUrl': 'string',
                    'buttons': [
                        {
                            'text': 'string',
                            'value': 'string'
                        },
                    ]
                },
            ]
        }
    }
    response_o = runtime_response.LexPostTextResponse(response)
    assert response_o.is_fulfilled()
    assert response_o.is_plain_text()
    assert not response_o.is_custom_payload()
    assert not response_o.is_composite()
    assert not response_o.is_confirm_intent()
    assert not response_o.is_elicit_intent()
    assert not response_o.is_elicit_slot()
    assert not response_o.is_failed()

{             "messageVersion": "1.0",             "invocationSource": "DialogCodeHook",             "userId": "user_123",             "sessionAttributes": {                 "RequestorCity": "Portland"             },             "bot": {                 "name": "OrderFlowers",                 "alias": null,                 "version": "$LATEST"             },             "outputDialogMode": "Text",             "currentIntent": {                 "name": "OrderFlowers",                 "slots": {                     "PickupTime": null,                     "FlowerType": "Roses",                     "PickupDate": null                 },                 "confirmationStatus": "None"             }         } # assert event.sessionAttributes == {} #assert event.sessionAttributes == {}