CR fixes

Author: k-shlomi

src/scripts/create_teams.py

@@ -8,7 +8,7 @@
 from loguru import logger
 from pydantic import ValidationError
 
-from src.shared.clients.frontegg import get_jwt_token
+from src.shared.clients.frontegg import get_jit_jwt_token
 from src.shared.clients.jit import get_existing_teams, create_teams, list_assets, add_teams_to_asset, delete_teams
 from src.shared.diff_tools import get_different_items_in_lists
 from src.shared.models import Asset, TeamObject, Organization, TeamTemplate
@@ -151,7 +151,7 @@ def get_teams_for_assets(organization: Organization) -> Dict[str, List[str]]:
 
 
 def main():
-    jit_token = get_jwt_token()
+    jit_token = get_jit_jwt_token()
     if not jit_token:
         logger.error("Failed to retrieve JWT token. Exiting...")
         return

src/shared/clients/frontegg.py

@@ -15,7 +15,7 @@
 JIT_CLIENT_ID = os.getenv("JIT_CLIENT_ID")
 
 
-def get_jwt_token() -> Optional[str]:
+def get_jit_jwt_token() -> Optional[str]:
     payload = {
         "clientId": JIT_CLIENT_ID,
         "secret": JIT_CLIENT_SECRET

tests/shared/test_clients.py

@@ -1,6 +1,6 @@
 import pytest
 
-from src.shared.clients.frontegg import get_jwt_token, FRONTEGG_AUTH_URL
+from src.shared.clients.frontegg import get_jit_jwt_token, FRONTEGG_AUTH_URL
 from src.shared.clients.github import get_teams_from_github_topics
 from src.shared.clients.jit import list_assets, get_existing_teams, create_teams, add_teams_to_asset, delete_teams
 from src.shared.models import TeamObject, Asset, Organization, TeamTemplate, Resource
@@ -87,7 +87,7 @@ def test_get_jwt_token(status_code, expected_result, mocker):
 
     requests_post_mock = mocker.patch("requests.post", return_value=response_mock)
 
-    token = get_jwt_token()
+    token = get_jit_jwt_token()
 
     requests_post_mock.assert_called_once_with(
         FRONTEGG_AUTH_URL,