Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

adding return types #499

Merged
merged 6 commits into from
Dec 12, 2019
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -16,8 +16,11 @@
TeamsChannelAccount,
MessagingExtensionAction,
MessagingExtensionQuery,
MessagingExtensionActionResponse,
MessagingExtensionResponse,
O365ConnectorCardActionQuery,
TaskModuleRequest,
TaskModuleResponse,
)
from botframework.connector import Channels

Expand Down Expand Up @@ -49,7 +52,7 @@ async def on_turn(self, turn_context: TurnContext):

await super().on_turn(turn_context)

async def on_invoke_activity(self, turn_context: TurnContext):
async def on_invoke_activity(self, turn_context: TurnContext) -> InvokeResponse:
try:
if (
not turn_context.activity.name
Expand Down Expand Up @@ -171,7 +174,9 @@ async def on_invoke_activity(self, turn_context: TurnContext):
except _InvokeResponseException as err:
return err.create_invoke_response()

async def on_teams_card_action_invoke_activity(self, turn_context: TurnContext):
async def on_teams_card_action_invoke_activity(
self, turn_context: TurnContext
) -> InvokeResponse:
raise _InvokeResponseException(status_code=HTTPStatus.NOT_IMPLEMENTED)

async def on_teams_signin_verify_state(self, turn_context: TurnContext):
Expand All @@ -181,7 +186,7 @@ async def on_teams_file_consent(
self,
turn_context: TurnContext,
file_consent_card_response: FileConsentCardResponse,
):
) -> InvokeResponse:
if file_consent_card_response.action == "accept":
await self.on_teams_file_consent_accept_activity(
turn_context, file_consent_card_response
Expand Down Expand Up @@ -220,22 +225,22 @@ async def on_teams_o365_connector_card_action( # pylint: disable=unused-argumen

async def on_teams_app_based_link_query( # pylint: disable=unused-argument
self, turn_context: TurnContext, query: AppBasedLinkQuery
):
) -> MessagingExtensionResponse:
raise _InvokeResponseException(status_code=HTTPStatus.NOT_IMPLEMENTED)

async def on_teams_messaging_extension_query( # pylint: disable=unused-argument
self, turn_context: TurnContext, query: MessagingExtensionQuery
):
) -> MessagingExtensionResponse:
raise _InvokeResponseException(status_code=HTTPStatus.NOT_IMPLEMENTED)

async def on_teams_messaging_extension_select_item( # pylint: disable=unused-argument
self, turn_context: TurnContext, query
):
) -> MessagingExtensionResponse:
raise _InvokeResponseException(status_code=HTTPStatus.NOT_IMPLEMENTED)

async def on_teams_messaging_extension_submit_action_dispatch(
self, turn_context: TurnContext, action: MessagingExtensionAction
):
) -> MessagingExtensionActionResponse:
if not action.bot_message_preview_action:
return await self.on_teams_messaging_extension_submit_action_activity(
turn_context, action
Expand All @@ -247,7 +252,7 @@ async def on_teams_messaging_extension_submit_action_dispatch(
)

if action.bot_message_preview_action == "send":
return await self.on_teams_messaging_extension_bot_message_send_activity(
return await self.on_teams_messaging_extension_bot_message_preview_send_activity(
turn_context, action
)

Expand All @@ -258,27 +263,27 @@ async def on_teams_messaging_extension_submit_action_dispatch(

async def on_teams_messaging_extension_bot_message_preview_edit_activity( # pylint: disable=unused-argument
self, turn_context: TurnContext, action
):
) -> MessagingExtensionActionResponse:
raise _InvokeResponseException(status_code=HTTPStatus.NOT_IMPLEMENTED)

async def on_teams_messaging_extension_bot_message_send_activity( # pylint: disable=unused-argument
async def on_teams_messaging_extension_bot_message_preview_send_activity( # pylint: disable=unused-argument
self, turn_context: TurnContext, action
):
) -> MessagingExtensionActionResponse:
raise _InvokeResponseException(status_code=HTTPStatus.NOT_IMPLEMENTED)

async def on_teams_messaging_extension_submit_action_activity( # pylint: disable=unused-argument
self, turn_context: TurnContext, action: MessagingExtensionAction
):
) -> MessagingExtensionActionResponse:
raise _InvokeResponseException(status_code=HTTPStatus.NOT_IMPLEMENTED)

async def on_teams_messaging_extension_fetch_task( # pylint: disable=unused-argument
self, turn_context: TurnContext, action: MessagingExtensionAction
):
) -> MessagingExtensionActionResponse:
raise _InvokeResponseException(status_code=HTTPStatus.NOT_IMPLEMENTED)

async def on_teams_messaging_extension_configuration_query_settings_url( # pylint: disable=unused-argument
self, turn_context: TurnContext, query: MessagingExtensionQuery
):
) -> MessagingExtensionResponse:
raise _InvokeResponseException(status_code=HTTPStatus.NOT_IMPLEMENTED)

async def on_teams_messaging_extension_configuration_setting( # pylint: disable=unused-argument
Expand All @@ -293,12 +298,12 @@ async def on_teams_messaging_extension_card_button_clicked( # pylint: disable=u

async def on_teams_task_module_fetch( # pylint: disable=unused-argument
self, turn_context: TurnContext, task_module_request: TaskModuleRequest
):
) -> TaskModuleResponse:
raise _InvokeResponseException(status_code=HTTPStatus.NOT_IMPLEMENTED)

async def on_teams_task_module_submit( # pylint: disable=unused-argument
self, turn_context: TurnContext, task_module_request: TaskModuleRequest
):
) -> TaskModuleResponse:
raise _InvokeResponseException(status_code=HTTPStatus.NOT_IMPLEMENTED)

async def on_conversation_update_activity(self, turn_context: TurnContext):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -171,11 +171,13 @@ async def on_teams_messaging_extension_bot_message_preview_edit_activity(
turn_context, action
)

async def on_teams_messaging_extension_bot_message_send_activity(
async def on_teams_messaging_extension_bot_message_preview_send_activity(
self, turn_context: TurnContext, action: MessagingExtensionAction
):
self.record.append("on_teams_messaging_extension_bot_message_send_activity")
return await super().on_teams_messaging_extension_bot_message_send_activity(
self.record.append(
"on_teams_messaging_extension_bot_message_preview_send_activity"
)
return await super().on_teams_messaging_extension_bot_message_preview_send_activity(
turn_context, action
)

Expand Down Expand Up @@ -531,7 +533,10 @@ async def test_on_teams_messaging_extension_bot_message_send_activity(self):
assert len(bot.record) == 3
assert bot.record[0] == "on_invoke_activity"
assert bot.record[1] == "on_teams_messaging_extension_submit_action_dispatch"
assert bot.record[2] == "on_teams_messaging_extension_bot_message_send_activity"
assert (
bot.record[2]
== "on_teams_messaging_extension_bot_message_preview_send_activity"
)

async def test_on_teams_messaging_extension_bot_message_send_activity_with_none(
self,
Expand Down