Add defense_component to Portfolio when new task_order is created without an existing portfolio

2019-02-26 10:26:23 -05:00
parent f8a6d04d64
commit 7ee8858cc7
7 changed files with 26 additions and 8 deletions
--- a/atst/domain/portfolios/portfolios.py
+++ b/atst/domain/portfolios/portfolios.py
@@ -16,8 +16,10 @@ class PortfolioError(Exception):

 class Portfolios(object):
    @classmethod
-    def create(cls, user, name):
-        portfolio = PortfoliosQuery.create(name=name)
+    def create(cls, user, name, defense_component):
+        portfolio = PortfoliosQuery.create(
+            name=name, defense_component=defense_component
+        )
        Portfolios._create_portfolio_role(
            user, portfolio, "owner", status=PortfolioRoleStatus.ACTIVE
        )