pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fields for the new task order

Browse Source
This commit is contained in:
dandds
2018-12-14 13:19:22 -05:00
parent 6d92755a7f
commit c5580733ba
7 changed files with 308 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
tests/factories.py
View File

@@ -5,7 +5,7 @@ from uuid import uuid4
import datetime
from faker import Faker as _Faker
from atst.forms.data import SERVICE_BRANCHES
from atst.forms import data
from atst.models.environment import Environment
from atst.models.request import Request
from atst.models.request_revision import RequestRevision
@@ -25,8 +25,12 @@ from atst.models.invitation import Invitation, Status as InvitationStatus
from atst.domain.invitations import Invitations
def random_choice(choices):
return random.choice([k for k, v in choices if k])
def random_service_branch():
return random.choice([k for k, v in SERVICE_BRANCHES if k])
return random_choice(data.SERVICE_BRANCHES)
class Base(factory.alchemy.SQLAlchemyModelFactory):
@@ -352,9 +356,12 @@ class TaskOrderFactory(Base):
class Meta:
model = TaskOrder
clin_0001 = random.randrange(100, 100_000)
clin_0003 = random.randrange(100, 100_000)
clin_1001 = random.randrange(100, 100_000)
clin_1003 = random.randrange(100, 100_000)
clin_2001 = random.randrange(100, 100_000)
clin_2003 = random.randrange(100, 100_000)
clin_01 = random.randrange(100, 100_000)
clin_03 = random.randrange(100, 100_000)
defense_component = random_service_branch()
app_migration = random_choice(data.APP_MIGRATION)
native_apps = "no"
complexity = random_choice(data.PROJECT_COMPLEXITY)
dev_team = random_choice(data.DEV_TEAM)
team_experience = random_choice(data.TEAM_EXPERIENCE)

11
tests/routes/task_orders/test_edit_task_order.py
View File

@@ -26,16 +26,9 @@ def test_create_new_workspace(client, user_session):
response = client.post(
url_for("task_orders.update", task_order_id=task_order.id),
data={
"clin_0001": 12345,
"clin_0003": 12345,
"clin_1001": 12345,
"clin_1003": 12345,
"clin_2001": 12345,
"clin_2003": 12345,
},
data={**TaskOrderFactory.dictionary(), "clin_01": 12345, "clin_03": 12345},
follow_redirects=False,
)
assert response.status_code == 200
assert task_order.clin_0001 == 12345
assert task_order.clin_01 == 12345