From 2a79b6ea6df2ede5b399173badddce397318fa6c Mon Sep 17 00:00:00 2001 From: Philip Kalinsky Date: Mon, 9 Mar 2020 11:18:23 -0400 Subject: [PATCH] test jobs update the last before COMPLETED state --- tests/test_jobs.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_jobs.py b/tests/test_jobs.py index 4e909046..d5dfbcd5 100644 --- a/tests/test_jobs.py +++ b/tests/test_jobs.py @@ -9,6 +9,7 @@ from atat.domain.csp.cloud import MockCloudProvider from atat.domain.csp.cloud.models import BillingInstructionCSPPayload, UserRoleCSPResult from atat.domain.portfolios import Portfolios from atat.models import ApplicationRoleStatus, Portfolio, FSMStates +from atat.models.mixins.state_machines import AzureStages from atat.jobs import ( RecordFailure, @@ -337,7 +338,8 @@ class TestDoProvisionPortfolio: portfolio=portfolio ) # The stage before "COMPLETED" - sm.state = FSMStates.BILLING_OWNER_CREATED + last_step = [e.name for e in AzureStages][-1] + sm.state = getattr(FSMStates, f"{last_step}_CREATED") do_provision_portfolio(csp=csp, portfolio_id=portfolio.id) # send_PPOC_email was called