pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

/home no longer redirects and has add portfolio button

Browse Source
This commit is contained in:
George Drummond
2019-05-29 16:51:08 -04:00
parent c683a120e4
commit 83adf1295c
4 changed files with 14 additions and 85 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
tests/routes/test_home.py
View File

@@ -1,62 +1,16 @@
import pytest
from tests.factories import UserFactory, PortfolioFactory
from atst.domain.portfolios import Portfolios
from atst.models.portfolio_role import Status as PortfolioRoleStatus
from flask import url_for
from tests.factories import UserFactory
from atst.utils.localization import translate
def test_portfolio_owner_with_one_portfolio_redirected_to_reports(client, user_session):
portfolio = PortfolioFactory.create()
user_session(portfolio.owner)
response = client.get("/home", follow_redirects=False)
assert "/portfolios/{}/reports".format(portfolio.id) in response.location
def test_portfolio_owner_with_more_than_one_portfolio_redirected_to_portfolios(
client, user_session
):
owner = UserFactory.create()
PortfolioFactory.create(owner=owner)
PortfolioFactory.create(owner=owner)
user_session(owner)
response = client.get("/home", follow_redirects=False)
assert "/portfolios" in response.location
def test_non_owner_user_with_one_portfolio_redirected_to_portfolio_applications(
client, user_session
):
def test_home_route(client, user_session):
user = UserFactory.create()
portfolio = PortfolioFactory.create()
Portfolios._create_portfolio_role(
user, portfolio, status=PortfolioRoleStatus.ACTIVE
)
user_session(user)
response = client.get("/home", follow_redirects=False)
assert "/portfolios/{}/applications".format(portfolio.id) in response.location
response = client.get(url_for("atst.home"))
def test_non_owner_user_with_mulitple_portfolios_redirected_to_portfolios(
client, user_session
):
user = UserFactory.create()
portfolios = []
for _ in range(3):
portfolio = PortfolioFactory.create()
portfolios.append(portfolio)
role = Portfolios._create_portfolio_role(
user, portfolio, status=PortfolioRoleStatus.ACTIVE
)
user_session(user)
response = client.get("/home", follow_redirects=False)
alphabetically_first_portfolio = sorted(portfolios, key=lambda p: p.name)[0]
assert "/portfolios" in response.location
assert str(alphabetically_first_portfolio.id) in response.location
assert response.status_code == 200
assert translate("home.add_portfolio_button_text").encode("utf8") in response.data