Merge pull request #949 from dod-ccpo/no-test-code-in-prod

Do not use test functions in production code
This commit is contained in:
dandds 2019-06-19 15:18:39 -04:00 committed by GitHub
commit 9c2b9c6da6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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