import pytest

from tests.factories import UserFactory, PortfolioFactory, RequestFactory
from atst.domain.portfolios import Portfolios
from atst.models.portfolio_role import Status as PortfolioRoleStatus


def test_request_owner_with_one_portfolio_redirected_to_reports(client, user_session):
    request = RequestFactory.create()
    portfolio = Portfolios.create_from_request(request)

    user_session(request.creator)
    response = client.get("/home", follow_redirects=False)

    assert "/portfolios/{}/reports".format(portfolio.id) in response.location


def test_request_owner_with_more_than_one_portfolio_redirected_to_portfolios(
    client, user_session
):
    request_creator = UserFactory.create()
    Portfolios.create_from_request(RequestFactory.create(creator=request_creator))
    Portfolios.create_from_request(RequestFactory.create(creator=request_creator))

    user_session(request_creator)
    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
):
    user = UserFactory.create()
    portfolio = PortfolioFactory.create()
    Portfolios._create_portfolio_role(
        user, portfolio, "developer", status=PortfolioRoleStatus.ACTIVE
    )

    user_session(user)
    response = client.get("/home", follow_redirects=False)

    assert "/portfolios/{}/applications".format(portfolio.id) in response.location


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, "developer", 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


@pytest.mark.skip(reason="this may no longer be accurate")
def test_ccpo_user_redirected_to_requests(client, user_session):
    user = UserFactory.from_atat_role("ccpo")
    for _ in range(3):
        portfolio = PortfolioFactory.create()
        Portfolios._create_portfolio_role(user, portfolio, "developer")

    user_session(user)
    response = client.get("/home", follow_redirects=False)

    assert "/requests" in response.location