From 78e9cdd82c5617390a91c5b7dcda6512fbe17df8 Mon Sep 17 00:00:00 2001 From: dandds Date: Mon, 20 Aug 2018 10:56:29 -0400 Subject: [PATCH] use SERVICE_BRANCHES constant for requests factory --- tests/factories.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/factories.py b/tests/factories.py index a47b38ab..a6370060 100644 --- a/tests/factories.py +++ b/tests/factories.py @@ -3,6 +3,7 @@ import string import factory from uuid import uuid4 +from atst.forms.data import SERVICE_BRANCHES from atst.models.request import Request from atst.models.request_status_event import RequestStatusEvent, RequestStatus from atst.models.pe_number import PENumber @@ -66,7 +67,7 @@ class RequestFactory(factory.alchemy.SQLAlchemyModelFactory): "start_date": "2018-08-08", "cloud_native": "yes", "dollar_value": dollar_value, - "dod_component": "Army and Air Force Exchange Service", + "dod_component": SERVICE_BRANCHES[2][1], "data_transfers": "Less than 100GB", "expected_completion_date": "Less than 1 month", "jedi_migration": "yes", @@ -87,7 +88,7 @@ class RequestFactory(factory.alchemy.SQLAlchemyModelFactory): "email_request": user.email, "fname_request": user.first_name, "lname_request": user.last_name, - "service_branch": "Air Force, Department of the", + "service_branch": SERVICE_BRANCHES[1][1], "date_latest_training": "2018-08-06", }, }