fix: Linter errors in tests after multitenancy

Author: KShivendu

supertokens_python/recipe/emailpassword/__init__.py

@@ -17,6 +17,7 @@
 
 from supertokens_python.ingredients.emaildelivery.types import EmailDeliveryConfig
 from supertokens_python.recipe.emailpassword.types import EmailTemplateVars
+from supertokens_python.ingredients.emaildelivery import types as emaildelivery_types
 
 from . import exceptions as ex
 from . import utils
@@ -28,6 +29,7 @@
 InputSignUpFeature = utils.InputSignUpFeature
 InputFormField = utils.InputFormField
 SMTPService = emaildelivery_services.SMTPService
+EmailDeliveryInterface = emaildelivery_types.EmailDeliveryInterface
 
 if TYPE_CHECKING:
     from supertokens_python.supertokens import AppInfo

supertokens_python/recipe/emailverification/__init__.py

@@ -19,6 +19,7 @@
 from . import utils
 from .emaildelivery import services as emaildelivery_services
 from . import recipe
+from . import types
 from .interfaces import TypeGetEmailForUserIdFunction
 from .recipe import EmailVerificationRecipe
 from .types import EmailTemplateVars
@@ -28,6 +29,7 @@
 exception = ex
 SMTPService = emaildelivery_services.SMTPService
 EmailVerificationClaim = recipe.EmailVerificationClaim
+EmailDeliveryInterface = types.EmailDeliveryInterface
 
 
 if TYPE_CHECKING:

supertokens_python/recipe/passwordless/__init__.py

@@ -47,6 +47,8 @@
 SMTPService = emaildelivery_services.SMTPService
 TwilioService = smsdelivery_services.TwilioService
 SuperTokensSMSService = smsdelivery_services.SuperTokensSMSService
+EmailDeliveryInterface = types.EmailDeliveryInterface
+SMSDeliveryInterface = types.SMSDeliveryInterface
 
 
 def init(

supertokens_python/recipe/thirdpartyemailpassword/__init__.py

@@ -18,6 +18,8 @@
 from supertokens_python.ingredients.emaildelivery.types import EmailDeliveryConfig
 from supertokens_python.recipe.thirdparty import provider
 from supertokens_python.recipe.thirdpartyemailpassword.types import EmailTemplateVars
+from supertokens_python.ingredients.emaildelivery import types as emaildelivery_types
+from . import types
 
 from .. import emailpassword
 from . import exceptions as ex
@@ -34,6 +36,9 @@
 ProviderConfigForClientType = provider.ProviderConfigForClientType
 SMTPService = emaildelivery_services.SMTPService
 
+EmailDeliveryInterface = emaildelivery_types.EmailDeliveryInterface
+PasswordResetEmailTemplateVars = types.PasswordResetEmailTemplateVars
+
 if TYPE_CHECKING:
     from supertokens_python.supertokens import AppInfo
 

supertokens_python/recipe/thirdpartypasswordless/__init__.py

@@ -27,6 +27,8 @@
 from .recipe import ThirdPartyPasswordlessRecipe
 from .smsdelivery import services as smsdelivery_services
 from .types import EmailTemplateVars, SMSTemplateVars
+from supertokens_python.ingredients.emaildelivery import types as emaildelivery_types
+from supertokens_python.ingredients.smsdelivery import types as smsdelivery_types
 
 InputOverrideConfig = utils.InputOverrideConfig
 exceptions = ex
@@ -42,6 +44,8 @@
 ProviderConfig = provider.ProviderConfig
 ProviderClientConfig = provider.ProviderClientConfig
 ProviderConfigForClientType = provider.ProviderConfigForClientType
+EmailDeliveryInterface = emaildelivery_types.EmailDeliveryInterface
+SMSDeliveryInterface = smsdelivery_types.SMSDeliveryInterface
 
 if TYPE_CHECKING:
     from supertokens_python.supertokens import AppInfo

tests/Django/test_django.py

@@ -343,12 +343,15 @@ def override_email_password_apis(original_implementation: APIInterface):
             original_func = original_implementation.email_exists_get
 
             async def email_exists_get(
-                email: str, api_options: APIOptions, user_context: Dict[str, Any]
+                email: str,
+                tenant_id: str,
+                api_options: APIOptions,
+                user_context: Dict[str, Any],
             ):
                 response_dict = {"custom": True}
                 api_options.response.set_status_code(203)
                 api_options.response.set_json_content(response_dict)
-                return await original_func(email, api_options, user_context)
+                return await original_func(email, tenant_id, api_options, user_context)
 
             original_implementation.email_exists_get = email_exists_get
             return original_implementation

tests/Fastapi/test_fastapi.py

@@ -454,12 +454,15 @@ def override_email_password_apis(original_implementation: EPAPIInterface):
         original_func = original_implementation.email_exists_get
 
         async def email_exists_get(
-            email: str, api_options: APIOptions, user_context: Dict[str, Any]
+            email: str,
+            tenant_id: str,
+            api_options: APIOptions,
+            user_context: Dict[str, Any],
         ):
             response_dict = {"custom": True}
             api_options.response.set_status_code(203)
             api_options.response.set_json_content(response_dict)
-            return await original_func(email, api_options, user_context)
+            return await original_func(email, tenant_id, api_options, user_context)
 
         original_implementation.email_exists_get = email_exists_get
         return original_implementation

tests/Flask/test_flask.py

@@ -82,12 +82,15 @@ def override_email_password_apis(original_implementation: APIInterface):
         original_func = original_implementation.email_exists_get
 
         async def email_exists_get(
-            email: str, api_options: APIOptions, user_context: Dict[str, Any]
+            email: str,
+            tenant_id: str,
+            api_options: APIOptions,
+            user_context: Dict[str, Any],
         ):
             response_dict = {"custom": True}
             api_options.response.set_status_code(203)
             api_options.response.set_json_content(response_dict)
-            return await original_func(email, api_options, user_context)
+            return await original_func(email, tenant_id, api_options, user_context)
 
         original_implementation.email_exists_get = email_exists_get
         return original_implementation