diff --git a/atst/routes/dev.py b/atst/routes/dev.py index 72a9b75f..aeb3cff0 100644 --- a/atst/routes/dev.py +++ b/atst/routes/dev.py @@ -1,3 +1,5 @@ +import random + from flask import ( Blueprint, request, @@ -11,8 +13,8 @@ import pendulum from . import redirect_after_login_url, current_user_setup from atst.domain.users import Users from atst.domain.permission_sets import PermissionSets +from atst.forms.data import SERVICE_BRANCHES from atst.queue import queue -from tests.factories import random_service_branch from atst.utils import pick bp = Blueprint("dev", __name__) @@ -31,6 +33,11 @@ _ALL_PERMS = [ PermissionSets.VIEW_AUDIT_LOG, ] + +def random_service_branch(): + return random.choice([k for k, v in SERVICE_BRANCHES if k]) # nosec + + _DEV_USERS = { "sam": { "dod_id": "6346349876",