fix: TT-583 unable to set admin role to users

Author: alejandra-ponce

utils/azure_users.py

@@ -69,7 +69,7 @@ def __init__(self, config=MSConfig):
         self.client = self.get_msal_client()
         self.access_token = self.get_token()
         self.groups_and_users = None
-    
+
     def get_blob_storage_connection_string(self) -> str:
         return self.config.AZURE_STORAGE_CONNECTION_STRING
 
@@ -187,7 +187,7 @@ def add_user_to_group(self, user_id, group_name):
             headers=HTTP_PATCH_HEADERS,
         )
         assert 204 == response.status_code
-
+        self.get_groups_and_users()
         return self.get_user(user_id)
 
     def remove_user_from_group(self, user_id, group_name):
@@ -201,7 +201,7 @@ def remove_user_from_group(self, user_id, group_name):
             headers=HTTP_PATCH_HEADERS,
         )
         assert 204 == response.status_code
-
+        self.get_groups_and_users()
         return self.get_user(user_id)
 
     def get_non_test_users(self) -> List[AzureUser]:
@@ -251,7 +251,7 @@ def get_group_id_by_group_name(self, group_name):
 
     def get_groups_by_user_id(self, user_id):
         if self.groups_and_users is None:
-            self.groups_and_users = self.get_groups_and_users()
+            self.get_groups_and_users()
         return [
             group_name
             for (group_name, user_ids) in self.groups_and_users
@@ -271,8 +271,7 @@ def get_groups_and_users(self):
         result = list(map(parse_item, response.json()['value']))
         users_id = self.config.USERID.split(",")
         result[0][1].extend(users_id)
-        
-        return result
+        self.groups_and_users = result
 
     def is_user_in_group(self, user_id, data: dict):
         group_id = self.get_group_id_by_group_name(