test state machine update expected states

2020-01-28 12:06:34 -05:00
parent 5223fda8f8
commit d042282ca0
1 changed files with 2 additions and 0 deletions
--- a/tests/domain/test_portfolio_state_machine.py
+++ b/tests/domain/test_portfolio_state_machine.py
@@ -104,6 +104,8 @@ def test_fsm_transition_start(mock_cloud_provider, portfolio: Portfolio):
        FSMStates.TASK_ORDER_BILLING_CREATION_CREATED,
        FSMStates.TASK_ORDER_BILLING_VERIFICATION_CREATED,
        FSMStates.BILLING_INSTRUCTION_CREATED,
+        FSMStates.PRODUCT_PURCHASE_CREATED,
+        FSMStates.PRODUCT_PURCHASE_VERIFICATION_CREATED,
    ]

    # Should source all creds for portfolio? might be easier to manage than per-step specific ones