Skip to content

Commit ef60b9b

Browse files
committed
Fix the api test
1 parent 50dc7ff commit ef60b9b

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

ros/api/common/utils.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
is_valid_uuid,
1010
system_ids_by_org_id
1111
)
12-
12+
from ros.extensions import db
1313

1414
LOG = logging.getLogger(__name__)
1515
prefix = "VALIDATE REQUEST"
@@ -54,9 +54,8 @@ def check_for_user():
5454
def check_for_system(ident, inventory_id):
5555
if not is_valid_uuid(inventory_id):
5656
abort(404, message='Invalid inventory_id, Id should be in form of UUID4')
57-
58-
system_id = system_ids_by_org_id(ident['org_id']).filter(
59-
System.inventory_id == inventory_id).first()
57+
system_id = db.session.scalars(system_ids_by_org_id(ident['org_id']).filter(
58+
System.inventory_id == inventory_id)).first()
6059

6160
if system_id is None:
6261
abort(404, message=f"System {inventory_id} doesn't exist.")

0 commit comments

Comments
 (0)