Be more tolerant with not needed roles

Signed-off-by: Marc Schöchlin <[email protected]>

Author: Marc Schöchlin

src/openstack_workload_generator/entities/project.py

@@ -236,7 +236,7 @@ def create_and_get_project(self) -> Project:
         self.adapt_quota()
 
         self.assign_role_to_user_for_project("manager")
-        self.assign_role_to_user_for_project("load-balancer_member")
+        self.assign_role_to_user_for_project("load-balancer_member", required=False)
         self.assign_role_to_user_for_project("member")
 
         self.workload_network = WorkloadGeneratorNetwork(

src/openstack_workload_generator/entities/user.py

@@ -20,9 +20,16 @@ def __init__(self, conn: Connection, user_name: str, domain: Domain):
             user_name, query={"domain_id": self.domain.id}
         )
 
-    def assign_role_to_user(self, role_name: str):
+    def assign_role_to_user(self, role_name: str, mandatory: bool = True):
+
+        role_id = self.get_role_id_by_name(role_name, mandatory)
+
+        if role_id is None:
+            LOGGER.warning(f"Role '{role_name}' not found, not assigning it")
+            return
+
         self.conn.identity.assign_project_role_to_user(
-            self.obj.id, self.domain.id, self.get_role_id_by_name(role_name)
+            self.obj.id, self.domain.id, role_id
         )
         LOGGER.info(
             f"Assigned role '{role_name}' to user '{self.obj.name}' in {DomainCache.ident_by_id(self.domain.id)}"
@@ -55,8 +62,11 @@ def delete_user(self):
         LOGGER.warning(f"Deleted user: {self.obj.name} / {self.obj.id}")
         self.obj = None
 
-    def get_role_id_by_name(self, role_name) -> str:
+    def get_role_id_by_name(self, role_name, mandatory: bool = True) -> str | None:
         for role in self.conn.identity.roles():
             if role.name == role_name:
                 return role.id
-        raise RuntimeError(f"No such role {role_name}")
+        if mandatory:
+            raise RuntimeError(f"No such role {role_name}")
+        else:
+            return None