adding types

Virtual-Josh · Virtual-Josh · commit e3f208f23108 · 2019-12-13T16:05:21.000-08:00
diff --git a/libraries/botbuilder-core/botbuilder/core/teams/teams_activity_handler.py b/libraries/botbuilder-core/botbuilder/core/teams/teams_activity_handler.py
@@ -248,12 +248,12 @@ async def on_teams_messaging_extension_submit_action_dispatch(
 
         if action.bot_message_preview_action == "edit":
             return await self.on_teams_messaging_extension_bot_message_preview_edit(
-                turn_context, action
+                turn_context, deserializer_helper(MessagingExtensionAction, action)
             )
 
         if action.bot_message_preview_action == "send":
             return await self.on_teams_messaging_extension_bot_message_preview_send(
-                turn_context, action
+                turn_context, deserializer_helper(MessagingExtensionAction, action)
             )
 
         raise _InvokeResponseException(
@@ -262,12 +262,12 @@ async def on_teams_messaging_extension_submit_action_dispatch(
         )
 
     async def on_teams_messaging_extension_bot_message_preview_edit(  # pylint: disable=unused-argument
-        self, turn_context: TurnContext, action
+        self, turn_context: TurnContext, action: MessagingExtensionAction
     ) -> MessagingExtensionActionResponse:
         raise _InvokeResponseException(status_code=HTTPStatus.NOT_IMPLEMENTED)
 
     async def on_teams_messaging_extension_bot_message_preview_send(  # pylint: disable=unused-argument
-        self, turn_context: TurnContext, action
+        self, turn_context: TurnContext, action: MessagingExtensionAction
     ) -> MessagingExtensionActionResponse:
         raise _InvokeResponseException(status_code=HTTPStatus.NOT_IMPLEMENTED)
 
diff --git a/libraries/botbuilder-testing/botbuilder/testing/storage_base_tests.py b/libraries/botbuilder-testing/botbuilder/testing/storage_base_tests.py
@@ -1,3 +1,6 @@
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License.
+
 """
 Base tests that all storage providers should implement in their own tests.
 They handle the storage-based assertions, internally.
