async def test_should_not_send_retry_if_not_specified(self):
async def exec_test(turn_context: TurnContext):
dc = await dialogs.create_context(turn_context)
results = await dc.continue_dialog()
if results.status == DialogTurnStatus.Empty:
await dc.begin_dialog('AttachmentPrompt', PromptOptions())
elif results.status == DialogTurnStatus.Complete:
attachment = results.result[0]
content = MessageFactory.text(attachment.content)
await turn_context.send_activity(content)
await convo_state.save_changes(turn_context)
# Initialize TestAdapter.
adapter = TestAdapter(exec_test)
# Create ConversationState with MemoryStorage and register the state as middleware.
convo_state = ConversationState(MemoryStorage())
# Create a DialogState property, DialogSet and AttachmentPrompt.
dialog_state = convo_state.create_property('dialog_state')
dialogs = DialogSet(dialog_state)
dialogs.add(AttachmentPrompt('AttachmentPrompt'))
# Create incoming activity with attachment.
attachment = Attachment(content='some content', content_type='text/plain')
attachment_activity = Activity(type=ActivityTypes.message, attachments=[attachment])
step1 = await adapter.send('hello')
step2 = await step1.send('what?')
step3 = await step2.send(attachment_activity)
await step3.assert_reply('some content')
def __init__(self, user_state: UserState):
super(UserProfileDialog, self).__init__(UserProfileDialog.__name__)
self.user_profile_accessor = user_state.create_property("UserProfile")
self.add_dialog(
WaterfallDialog(
WaterfallDialog.__name__,
[
self.transport_step,
self.name_step,
self.name_confirm_step,
self.age_step,
self.picture_step,
self.confirm_step,
self.summary_step,
],
))
self.add_dialog(TextPrompt(TextPrompt.__name__))
self.add_dialog(
NumberPrompt(NumberPrompt.__name__,
UserProfileDialog.age_prompt_validator))
self.add_dialog(ChoicePrompt(ChoicePrompt.__name__))
self.add_dialog(ConfirmPrompt(ConfirmPrompt.__name__))
self.add_dialog(
AttachmentPrompt(AttachmentPrompt.__name__,
UserProfileDialog.picture_prompt_validator))
self.initial_dialog_id = WaterfallDialog.__name__
def __init__(self, user_state: UserState):
super(UserProfileDialog, self).__init__(UserProfileDialog.__name__)
self.user_profile_accessor = user_state.create_property("UserProfile")
self.add_dialog(
WaterfallDialog(
WaterfallDialog.__name__,
[
self.first_step,
self.second_step,
self.third_step,
self.fourth_step,
self.fifth_step,
self.sixth_step,
self.seventh_step,
self.eighth_step,
],
))
self.add_dialog(TextPrompt(TextPrompt.__name__))
self.add_dialog(
NumberPrompt(NumberPrompt.__name__,
UserProfileDialog.age_prompt_validator))
self.add_dialog(ChoicePrompt(ChoicePrompt.__name__))
self.add_dialog(ConfirmPrompt(ConfirmPrompt.__name__))
self.add_dialog(
AttachmentPrompt(AttachmentPrompt.__name__,
UserProfileDialog.picture_prompt_validator))
self.initial_dialog_id = WaterfallDialog.__name__
async def test_attachment_prompt_with_input_hint(self):
prompt_activity = Activity(
type=ActivityTypes.message,
text="please add an attachment.",
input_hint=InputHints.accepting_input,
)
async def exec_test(turn_context: TurnContext):
dialog_context = await dialogs.create_context(turn_context)
results = await dialog_context.continue_dialog()
if results.status == DialogTurnStatus.Empty:
options = PromptOptions(prompt=copy.copy(prompt_activity))
await dialog_context.prompt("AttachmentPrompt", options)
elif results.status == DialogTurnStatus.Complete:
attachment = results.result[0]
content = MessageFactory.text(attachment.content)
await turn_context.send_activity(content)
await convo_state.save_changes(turn_context)
# Initialize TestAdapter.
adapter = TestAdapter(exec_test)
# Create ConversationState with MemoryStorage and register the state as middleware.
convo_state = ConversationState(MemoryStorage())
# Create a DialogState property, DialogSet and AttachmentPrompt.
dialog_state = convo_state.create_property("dialog_state")
dialogs = DialogSet(dialog_state)
dialogs.add(AttachmentPrompt("AttachmentPrompt"))
step1 = await adapter.send("hello")
await step1.assert_reply(prompt_activity)
async def test_should_send_ignore_retry_rompt_if_validator_replies(self):
async def exec_test(turn_context: TurnContext):
dialog_context = await dialogs.create_context(turn_context)
results = await dialog_context.continue_dialog()
if results.status == DialogTurnStatus.Empty:
options = PromptOptions(
prompt=Activity(
type=ActivityTypes.message, text="please add an attachment."
),
retry_prompt=Activity(
type=ActivityTypes.message, text="please try again."
),
)
await dialog_context.prompt("AttachmentPrompt", options)
elif results.status == DialogTurnStatus.Complete:
attachment = results.result[0]
content = MessageFactory.text(attachment.content)
await turn_context.send_activity(content)
await convo_state.save_changes(turn_context)
# Initialize TestAdapter.
adapter = TestAdapter(exec_test)
# Create ConversationState with MemoryStorage and register the state as middleware.
convo_state = ConversationState(MemoryStorage())
# Create a DialogState property, DialogSet and AttachmentPrompt.
dialog_state = convo_state.create_property("dialog_state")
dialogs = DialogSet(dialog_state)
async def aux_validator(prompt_context: PromptValidatorContext):
assert prompt_context, "Validator missing prompt_context"
if not prompt_context.recognized.succeeded:
await prompt_context.context.send_activity("Bad input.")
return prompt_context.recognized.succeeded
dialogs.add(AttachmentPrompt("AttachmentPrompt", aux_validator))
# Create incoming activity with attachment.
attachment = Attachment(content="some content", content_type="text/plain")
attachment_activity = Activity(
type=ActivityTypes.message, attachments=[attachment]
)
invalid_activty = Activity(type=ActivityTypes.message, text="invalid")
step1 = await adapter.send("hello")
step2 = await step1.assert_reply("please add an attachment.")
step3 = await step2.send(invalid_activty)
step4 = await step3.assert_reply("Bad input.")
step5 = await step4.send(attachment_activity)
await step5.assert_reply("some content")
def __init__(self):
super().__init__(FileUploadDialog.__name__)
self.add_dialog(AttachmentPrompt(AttachmentPrompt.__name__))
self.add_dialog(ConfirmPrompt(ConfirmPrompt.__name__))
self.add_dialog(
WaterfallDialog(
WaterfallDialog.__name__,
[self.prompt_upload_step, self.handle_attachment_step, self.final_step],
)
)
self.initial_dialog_id = WaterfallDialog.__name__
async def test_attachment_prompt_with_custom_retry(self):
async def exec_test(turn_context: TurnContext):
dc = await dialogs.create_context(turn_context)
results = await dc.continue_dialog()
if results.status == DialogTurnStatus.Empty:
options = PromptOptions(
prompt=Activity(type=ActivityTypes.message, text='please add an attachment.'),
retry_prompt=Activity(type=ActivityTypes.message, text='please try again.')
)
await dc.prompt('AttachmentPrompt', options)
elif results.status == DialogTurnStatus.Complete:
attachment = results.result[0]
content = MessageFactory.text(attachment.content)
await turn_context.send_activity(content)
await convo_state.save_changes(turn_context)
# Initialize TestAdapter.
adapter = TestAdapter(exec_test)
# Create ConversationState with MemoryStorage and register the state as middleware.
convo_state = ConversationState(MemoryStorage())
# Create a DialogState property, DialogSet and AttachmentPrompt.
dialog_state = convo_state.create_property('dialog_state')
dialogs = DialogSet(dialog_state)
async def aux_validator(prompt_context: PromptValidatorContext):
assert prompt_context, 'Validator missing prompt_context'
return prompt_context.recognized.succeeded
dialogs.add(AttachmentPrompt('AttachmentPrompt', aux_validator))
# Create incoming activity with attachment.
attachment = Attachment(content='some content', content_type='text/plain')
attachment_activity = Activity(type=ActivityTypes.message, attachments=[attachment])
invalid_activty = Activity(type=ActivityTypes.message, text='invalid')
step1 = await adapter.send('hello')
step2 = await step1.assert_reply('please add an attachment.')
step3 = await step2.send(invalid_activty)
step4 = await step3.assert_reply('please try again.')
step5 = await step4.send(attachment_activity)
await step5.assert_reply('some content')
async def test_retry_attachment_prompt(self):
async def exec_test(turn_context: TurnContext):
dialog_context = await dialogs.create_context(turn_context)
results = await dialog_context.continue_dialog()
if results.status == DialogTurnStatus.Empty:
options = PromptOptions(
prompt=Activity(
type=ActivityTypes.message, text="please add an attachment."
)
)
await dialog_context.prompt("AttachmentPrompt", options)
elif results.status == DialogTurnStatus.Complete:
attachment = results.result[0]
content = MessageFactory.text(attachment.content)
await turn_context.send_activity(content)
await convo_state.save_changes(turn_context)
# Initialize TestAdapter.
adapter = TestAdapter(exec_test)
# Create ConversationState with MemoryStorage and register the state as middleware.
convo_state = ConversationState(MemoryStorage())
# Create a DialogState property, DialogSet and AttachmentPrompt.
dialog_state = convo_state.create_property("dialog_state")
dialogs = DialogSet(dialog_state)
dialogs.add(AttachmentPrompt("AttachmentPrompt"))
# Create incoming activity with attachment.
attachment = Attachment(content="some content", content_type="text/plain")
attachment_activity = Activity(
type=ActivityTypes.message, attachments=[attachment]
)
step1 = await adapter.send("hello")
step2 = await step1.assert_reply("please add an attachment.")
step3 = await step2.send("hello again")
step4 = await step3.assert_reply("please add an attachment.")
step5 = await step4.send(attachment_activity)
await step5.assert_reply("some content")
def __init__(self, dialog_id: str = None):
super(WaterfallPhoto, self).__init__(dialog_id
or WaterfallPhoto.__name__)
#self.user_profile_accessor = user_state.create_property("UserProfile")
self.add_dialog(
WaterfallDialog(
WaterfallDialog.__name__,
[
self.picture_step,
self.confirm_step,
self.evaluate_data_step,
],
))
self.add_dialog(TextPrompt(TextPrompt.__name__))
#self.add_dialog(TextPrompt(TextPrompt.__name__, WaterfallQuery.insertCorrectdate))
self.add_dialog(NumberPrompt(NumberPrompt.__name__))
self.add_dialog(ChoicePrompt(ChoicePrompt.__name__))
self.add_dialog(ConfirmPrompt(ConfirmPrompt.__name__))
self.add_dialog(
AttachmentPrompt(AttachmentPrompt.__name__,
WaterfallPhoto.picture_prompt_validator))
self.initial_dialog_id = WaterfallDialog.__name__
def test_attachment_prompt_with_none_id_should_fail(self):
with self.assertRaises(TypeError):
AttachmentPrompt(None)
def test_attachment_prompt_with_empty_id_should_fail(self):
empty_id = ""
with self.assertRaises(TypeError):
AttachmentPrompt(empty_id)