rhodecode/apps/repository/tests/test_repo_feed.py

# Copyright (C) 2010-2023 RhodeCode GmbH
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License, version 3
# (only), as published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
# This program is dual-licensed. If you wish to learn more about the
# RhodeCode Enterprise Edition, including its added features, Support services,
# and proprietary license terms, please see https://rhodecode.com/licenses/

import pytest
from rhodecode.model.auth_token import AuthTokenModel
from rhodecode.tests import TestController


def route_path(name, params=None, **kwargs):
    import urllib.request
    import urllib.parse
    import urllib.error

    base_url = {
        'rss_feed_home': '/{repo_name}/feed-rss',
        'atom_feed_home': '/{repo_name}/feed-atom',
        'rss_feed_home_old': '/{repo_name}/feed/rss',
        'atom_feed_home_old': '/{repo_name}/feed/atom',
    }[name].format(**kwargs)

    if params:
        base_url = '{}?{}'.format(base_url, urllib.parse.urlencode(params))
    return base_url


class TestFeedView(TestController):

    @pytest.mark.parametrize("feed_type,response_types,content_type",[
        ('rss', ['<rss version="2.0"'],
         "application/rss+xml"),
        ('atom', ['xmlns="http://www.w3.org/2005/Atom"', 'xml:lang="en-us"'],
         "application/atom+xml"),
    ])
    def test_feed(self, backend, feed_type, response_types, content_type):
        self.log_user()
        response = self.app.get(
            route_path('{}_feed_home'.format(feed_type),
                       repo_name=backend.repo_name))

        for content in response_types:
            response.mustcontain(content)

        assert response.content_type == content_type

    @pytest.mark.parametrize("feed_type, content_type", [
        ('rss', "application/rss+xml"),
        ('atom', "application/atom+xml")
    ])
    def test_feed_with_auth_token(
            self, backend, user_admin, feed_type, content_type):
        auth_token = user_admin.feed_token
        assert auth_token != ''

        response = self.app.get(
            route_path(
                '{}_feed_home'.format(feed_type),
                repo_name=backend.repo_name,
                params=dict(auth_token=auth_token)),
            status=200)

        assert response.content_type == content_type

    @pytest.mark.parametrize("feed_type, content_type", [
        ('rss', "application/rss+xml"),
        ('atom', "application/atom+xml")
    ])
    def test_feed_with_auth_token_by_uid(
            self, backend, user_admin, feed_type, content_type):
        auth_token = user_admin.feed_token
        assert auth_token != ''

        response = self.app.get(
            route_path(
                '{}_feed_home'.format(feed_type),
                repo_name='_{}'.format(backend.repo.repo_id),
                params=dict(auth_token=auth_token)),
            status=200)

        assert response.content_type == content_type

    @pytest.mark.parametrize("feed_type, content_type", [
        ('rss', "application/rss+xml"),
        ('atom', "application/atom+xml")
    ])
    def test_feed_old_urls_with_auth_token(
            self, backend, user_admin, feed_type, content_type):
        auth_token = user_admin.feed_token
        assert auth_token != ''

        response = self.app.get(
            route_path(
                '{}_feed_home_old'.format(feed_type),
                repo_name=backend.repo_name,
                params=dict(auth_token=auth_token)),
            status=200)

        assert response.content_type == content_type

    @pytest.mark.parametrize("feed_type", ['rss', 'atom'])
    def test_feed_with_auth_token_of_wrong_type(
            self, backend, user_util, feed_type):
        user = user_util.create_user()
        auth_token = AuthTokenModel().create(
            user.user_id, u'test-token', -1, AuthTokenModel.cls.ROLE_API)
        auth_token = auth_token.api_key

        self.app.get(
            route_path(
                '{}_feed_home'.format(feed_type),
                repo_name=backend.repo_name,
                params=dict(auth_token=auth_token)),
            status=302)

        auth_token = AuthTokenModel().create(
            user.user_id, u'test-token', -1, AuthTokenModel.cls.ROLE_FEED)
        auth_token = auth_token.api_key
        self.app.get(
            route_path(
                '{}_feed_home'.format(feed_type),
                repo_name=backend.repo_name,
                params=dict(auth_token=auth_token)),
            status=200)