from flask import url_for, session


def test_should_show_choose_services_page(app_,
                                          mock_login,
                                          mock_get_user,
                                          api_user_active,
                                          mock_get_services):
    with app_.test_request_context():
        with app_.test_client() as client:
            client.login(api_user_active)
            response = client.get(url_for('main.choose_service'))

        assert response.status_code == 200
        resp_data = response.get_data(as_text=True)
        assert 'Choose service' in resp_data
        services = mock_get_services.side_effect()
        assert mock_get_services.called
        assert services['data'][0]['name'] in resp_data
        assert services['data'][1]['name'] in resp_data
        assert 'List all services' not in resp_data


def test_redirect_if_only_one_service(
    app_,
    mock_login,
    mock_get_user,
    api_user_active,
    mock_get_services_with_one_service
):
    with app_.test_request_context():
        with app_.test_client() as client:
            client.login(api_user_active)
            response = client.get(url_for('main.show_all_services_or_dashboard'))

        service = mock_get_services_with_one_service.side_effect()['data'][0]
        assert response.status_code == 302
        assert response.location == url_for('main.service_dashboard', service_id=service['id'], _external=True)


def test_redirect_if_multiple_services(
    app_,
    mock_login,
    mock_get_user,
    api_user_active,
    mock_get_services
):
    with app_.test_request_context():
        with app_.test_client() as client:
            client.login(api_user_active)
            response = client.get(url_for('main.show_all_services_or_dashboard'))

        assert response.status_code == 302
        assert response.location == url_for('main.choose_service', _external=True)


def test_redirect_if_service_in_session(
    app_,
    mock_login,
    mock_get_user,
    api_user_active,
    mock_get_services,
    mock_get_service
):
    with app_.test_request_context():
        with app_.test_client() as client:
            client.login(api_user_active)
            with client.session_transaction() as session:
                session['service_id'] = '147ad62a-2951-4fa1-9ca0-093cd1a52c52'
            response = client.get(url_for('main.show_all_services_or_dashboard'))

        assert response.status_code == 302
        assert response.location == url_for(
            'main.service_dashboard',
            service_id='147ad62a-2951-4fa1-9ca0-093cd1a52c52',
            _external=True
        )


def test_should_redirect_if_not_logged_in(app_):
    with app_.test_request_context():
        with app_.test_client() as client:
            response = client.get(url_for('main.show_all_services_or_dashboard'))
            assert response.status_code == 302
            assert url_for('main.index', _external=True) in response.location


def test_should_show_all_services_for_platform_admin_user(app_,
                                                          platform_admin_user,
                                                          mock_get_services,
                                                          mocker):
    with app_.test_request_context():
        with app_.test_client() as client:
            mocker.patch('app.user_api_client.get_user', return_value=platform_admin_user)
            client.login(platform_admin_user)
        response = client.get(url_for('main.choose_service'))

        assert response.status_code == 200
        resp_data = response.get_data(as_text=True)
        assert 'Choose service' in resp_data
        services = mock_get_services.side_effect()
        assert mock_get_services.called
        assert services['data'][0]['name'] in resp_data
        assert services['data'][1]['name'] in resp_data
        assert 'List all services' in resp_data