throw error for missing permission_sets in PermissionSets.get_many

2019-03-19 06:11:35 -04:00
parent 366ada5a90
commit 7c5e931c67
3 changed files with 13 additions and 4 deletions
--- a/atst/domain/permission_sets.py
+++ b/atst/domain/permission_sets.py
@@ -32,12 +32,17 @@ class PermissionSets(object):

    @classmethod
    def get_many(cls, perms_set_names):
-        return (
+        permission_sets = (
            db.session.query(PermissionSet)
            .filter(PermissionSet.name.in_(perms_set_names))
            .all()
        )

+        if len(permission_sets) != len(perms_set_names):
+            raise NotFoundError("permission_set")
+
+        return permission_sets
+

 ATAT_ROLES = [
    {