Portfolio archiving

tests/routes/portfolios/test_index.py

@@ -3,6 +3,7 @@ from flask import url_for
 from tests.factories import (
     random_future_date,
     random_past_date,
+    ApplicationFactory,
     PortfolioFactory,
     TaskOrderFactory,
     UserFactory,
@@ -109,3 +110,35 @@ def test_portfolio_reports_with_mock_portfolio(client, user_session):
     assert response.status_code == 200
     assert portfolio.name in response.data.decode()
     assert "$251,626.00 Total spend to date" in response.data.decode()
+
+
+def test_delete_portfolio_success(client, user_session):
+    portfolio = PortfolioFactory.create()
+    owner = portfolio.owner
+    user_session(owner)
+
+    assert len(Portfolios.for_user(user=owner)) == 1
+
+    response = client.post(
+        url_for("portfolios.delete_portfolio", portfolio_id=portfolio.id)
+    )
+
+    assert response.status_code == 302
+    assert url_for("atst.home") in response.location
+    assert len(Portfolios.for_user(user=owner)) == 0
+
+
+def test_delete_portfolio_failure(client, user_session):
+    portfolio = PortfolioFactory.create()
+    application = ApplicationFactory.create(portfolio=portfolio)
+    owner = portfolio.owner
+    user_session(owner)
+
+    assert len(Portfolios.for_user(user=owner)) == 1
+
+    response = client.post(
+        url_for("portfolios.delete_portfolio", portfolio_id=portfolio.id)
+    )
+
+    assert response.status_code == 500
+    assert len(Portfolios.for_user(user=owner)) == 1