rename and rearrange some portfolio route functions
This commit is contained in:
65
tests/routes/portfolios/test_index.py
Normal file
65
tests/routes/portfolios/test_index.py
Normal file
@@ -0,0 +1,65 @@
|
||||
from flask import url_for
|
||||
|
||||
from tests.factories import (
|
||||
random_future_date,
|
||||
random_past_date,
|
||||
PortfolioFactory,
|
||||
TaskOrderFactory,
|
||||
UserFactory,
|
||||
)
|
||||
from atst.utils.localization import translate
|
||||
|
||||
|
||||
def test_portfolio_index_with_existing_portfolios(client, user_session):
|
||||
portfolio = PortfolioFactory.create()
|
||||
user_session(portfolio.owner)
|
||||
|
||||
response = client.get(url_for("portfolios.portfolios"))
|
||||
|
||||
assert response.status_code == 200
|
||||
assert portfolio.name.encode("utf8") in response.data
|
||||
assert (
|
||||
translate("portfolios.index.empty.start_button").encode("utf8")
|
||||
not in response.data
|
||||
)
|
||||
|
||||
|
||||
def test_portfolio_index_without_existing_portfolios(client, user_session):
|
||||
user = UserFactory.create()
|
||||
user_session(user)
|
||||
|
||||
response = client.get(url_for("portfolios.portfolios"))
|
||||
|
||||
assert response.status_code == 200
|
||||
assert (
|
||||
translate("portfolios.index.empty.start_button").encode("utf8") in response.data
|
||||
)
|
||||
|
||||
|
||||
def test_portfolio_reports(client, user_session):
|
||||
portfolio = PortfolioFactory.create(
|
||||
applications=[
|
||||
{"name": "application1", "environments": [{"name": "application1 prod"}]}
|
||||
]
|
||||
)
|
||||
task_order = TaskOrderFactory.create(
|
||||
number="42",
|
||||
start_date=random_past_date(),
|
||||
end_date=random_future_date(),
|
||||
portfolio=portfolio,
|
||||
)
|
||||
user_session(portfolio.owner)
|
||||
response = client.get(url_for("portfolios.reports", portfolio_id=portfolio.id))
|
||||
assert response.status_code == 200
|
||||
assert portfolio.name in response.data.decode()
|
||||
expiration_date = task_order.end_date.strftime("%Y-%m-%d")
|
||||
assert expiration_date in response.data.decode()
|
||||
|
||||
|
||||
def test_portfolio_reports_with_mock_portfolio(client, user_session):
|
||||
portfolio = PortfolioFactory.create(name="A-Wing")
|
||||
user_session(portfolio.owner)
|
||||
response = client.get(url_for("portfolios.reports", portfolio_id=portfolio.id))
|
||||
assert response.status_code == 200
|
||||
assert portfolio.name in response.data.decode()
|
||||
assert "$251,626.00 Total spend to date" in response.data.decode()
|
Reference in New Issue
Block a user