From 3f61ddd584ea3313ed7e8ff55749af59fb078f46 Mon Sep 17 00:00:00 2001
From: Mike Salvatore <mike.s.salvatore@gmail.com>
Date: Thu, 7 Jul 2022 07:48:54 -0400
Subject: [PATCH] Common: Fix type hints in credentials.py

---
 monkey/common/credentials/credentials.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/monkey/common/credentials/credentials.py b/monkey/common/credentials/credentials.py
index a300405be..75775f22d 100644
--- a/monkey/common/credentials/credentials.py
+++ b/monkey/common/credentials/credentials.py
@@ -37,7 +37,9 @@ class CredentialsSchema(Schema):
     secrets = fields.List(fields.Mapping())
 
     @post_load
-    def _make_credentials(self, data, **kwargs) -> Mapping[str, Sequence[Mapping[str, Any]]]:
+    def _make_credentials(
+        self, data: MutableMapping, **kwargs: Mapping[str, Any]
+    ) -> Mapping[str, Sequence[Mapping[str, Any]]]:
         data["identities"] = tuple(
             [
                 CredentialsSchema._build_credential_component(component)
@@ -54,7 +56,7 @@ class CredentialsSchema(Schema):
         return data
 
     @staticmethod
-    def _build_credential_component(data: MutableMapping[str, Any]):
+    def _build_credential_component(data: Mapping[str, Any]) -> ICredentialComponent:
         credential_component_type = CredentialComponentType[data["credential_type"]]
         credential_component_class = CREDENTIAL_COMPONENT_TYPE_TO_CLASS[credential_component_type]
         credential_component_schema = CREDENTIAL_COMPONENT_TYPE_TO_CLASS_SCHEMA[
@@ -68,6 +70,7 @@ class CredentialsSchema(Schema):
         self, credentials: Credentials, **kwargs
     ) -> Mapping[str, Sequence[Mapping[str, Any]]]:
         data = {}
+
         data["identities"] = tuple(
             [
                 CredentialsSchema._serialize_credential_component(component)
@@ -90,6 +93,7 @@ class CredentialsSchema(Schema):
         credential_component_schema = CREDENTIAL_COMPONENT_TYPE_TO_CLASS_SCHEMA[
             credential_component.credential_type
         ]
+
         return credential_component_schema.dump(credential_component)