# -*- coding: utf-8 -*- # Copyright (C) 2010-2018 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 base_url = { 'rss_feed_home': '/{repo_name}/feed/rss', 'atom_feed_home': '/{repo_name}/feed/atom', }[name].format(**kwargs) if params: base_url = '{}?{}'.format(base_url, urllib.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", ['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, '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, '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)