Merge pull request #677 from dod-ccpo/move-component-to-portfolio

Move defense_component to portfolios model
2019-02-27 10:05:30 -05:00
parent 7dc7b15f09 8621f2054b
commit 3849a69e99
10 changed files with 101 additions and 6 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
        )