Move portfolio state machine helpers directly to model file to prevent import issues.

Having `get_stage_csp_class` in the csp module meant that any file that interacted with that import path would throw an error in a REPL. This will allow importing of the Azure and Mock providers for interactive dev.
This commit is contained in:
tomdds
2020-01-26 15:17:53 -05:00
parent b28281d04e
commit 466a575229
4 changed files with 25 additions and 27 deletions

View File

@@ -10,7 +10,7 @@ from tests.factories import (
from atst.models import FSMStates, PortfolioStateMachine, TaskOrder
from atst.models.mixins.state_machines import AzureStages, StageStates, compose_state
from atst.models.portfolio import Portfolio
from atst.domain.csp import get_stage_csp_class
from atst.models.portfolio_state_machine import get_stage_csp_class
# TODO: Write failure case tests