# -*- coding: utf-8 -*-

"""

 (c) 2015-2018 - Copyright Red Hat Inc

 Authors:

   Pierre-Yves Chibon <pingou@pingoured.fr></pingou@pingoured.fr>

"""

from __future__ import unicode_literals

__requires__ = ['SQLAlchemy >= 0.8']

import pkg_resources

import datetime

import unittest

import shutil

import sys

import tempfile

import os

import six

import json

import pygit2

from mock import patch, MagicMock

sys.path.insert(0, os.path.join(os.path.dirname(

    os.path.abspath(__file__)), '..'))

import pagure.lib.query

import tests

class PagureFlaskApptests(tests.Modeltests):

    """ Tests for flask app controller of pagure """

    def test_watch_list(self):

        ''' Test for watch list of a user '''

        user = tests.FakeUser(username='pingou')

        with tests.user_set(self.app.application, user):

            output = self.app.get('/', follow_redirects=True)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                '
You have no Projects
',

                output_text)

            tests.create_projects(self.session)

            output = self.app.get('/', follow_redirects=True)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                '
My Projects
',

                output_text)

    def test_view_users(self):

        """ Test the view_users endpoint. """

        output = self.app.get('/users/?page=abc')

        self.assertEqual(output.status_code, 200)

        output_text = output.get_data(as_text=True)

        self.assertIn(

            '
\n      Users '

            '2', output_text)

        self.assertIn(

            '\n                  '

            '
pingou',

            output_text)

        self.assertIn(

            '\n                  '

            '
foo',

            output_text)

    @patch.dict('pagure.config.config', {'ITEM_PER_PAGE': 2})

    def test_view_user_repo_cnt(self):

        """ Test the repo counts on the view_user endpoint. """

        tests.create_projects(self.session)

        self.gitrepos = tests.create_projects_git(

            pagure.config.config['GIT_FOLDER'])

        output = self.app.get('/user/pingou')

        self.assertEqual(output.status_code, 200)

        output_text = output.get_data(as_text=True)

        self.assertIn(

                """

                Projects 

                  3

              """, output_text)

        self.assertIn(

                """

                Forks 

                  0

              """, output_text)

    def test_view_user(self):

        """ Test the view_user endpoint. """

        output = self.app.get('/user/pingou?repopage=abc&forkpage=def')

        self.assertEqual(output.status_code, 200)

        output_text = output.get_data(as_text=True)

        self.assertIn(

                """

                Projects 

                  0

              """, output_text)

        self.assertIn(

                """

                Forks 

                  0

              """, output_text)

        tests.create_projects(self.session)

        self.gitrepos = tests.create_projects_git(

            pagure.config.config['GIT_FOLDER'])

        output = self.app.get('/user/pingou?repopage=abc&forkpage=def')

        self.assertEqual(output.status_code, 200)

        output_text = output.get_data(as_text=True)

        self.assertIn(

                """

                Projects 

                  3

              """, output_text)

        self.assertIn(

                """

                Forks 

                  0

              """, output_text)

        self.assertNotIn(

            'page 1 of 2',

            output_text)

    @patch.dict('pagure.config.config', {'ENABLE_UI_NEW_PROJECTS': False})

    def test_new_project_when_turned_off_in_the_ui(self):

        """ Test the new_project endpoint when new project creation is

        not allowed in the UI of this pagure instance. """

        user = tests.FakeUser(username='foo')

        with tests.user_set(self.app.application, user):

            output = self.app.get('/new/')

            self.assertEqual(output.status_code, 404)

            data = {

                'description': 'Project #1',

                'name': 'project-1',

            }

            output = self.app.post('/new/', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 404)

    @patch.dict('pagure.config.config', {'ENABLE_UI_NEW_PROJECTS': False})

    def test_new_project_button_when_turned_off_in_the_ui_no_project(self):

        """ Test the index endpoint when new project creation is

        not allowed in the UI of this pagure instance. """

        user = tests.FakeUser(username='foo')

        with tests.user_set(self.app.application, user):

            output = self.app.get('/', follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                '
My Projects
',

                output_text)

            # master template

            self.assertNotIn(

                '

                output_text)

            # index_auth template

            self.assertNotIn(

                'title="Create New Project" aria-hidden="true">',

                output_text)

    @patch.dict('pagure.config.config', {'ENABLE_UI_NEW_PROJECTS': False})

    def test_new_project_button_when_turned_off_in_the_ui_w_project(self):

        """ Test the index endpoint when new project creation is

        not allowed in the UI of this pagure instance. """

        tests.create_projects(self.session)

        user = tests.FakeUser(username='pingou')

        with tests.user_set(self.app.application, user):

            output = self.app.get('/', follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                '
My Projects
',

                output_text)

            # master template

            self.assertNotIn(

                '

                output_text)

            # index_auth template

            self.assertNotIn(

                'title="Create New Project" aria-hidden="true">',

                output_text)

    def test_new_project_when_turned_off(self):

        """ Test the new_project endpoint when new project creation is

        not allowed in the pagure instance. """

        #turn the project creation off

        pagure.config.config['ENABLE_NEW_PROJECTS'] = False

        # Before

        projects = pagure.lib.query.search_projects(self.session)

        self.assertEqual(len(projects), 0)

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'project-1.git')))

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'tickets', 'project-1.git')))

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'docs', 'project-1.git')))

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'requests', 'project-1.git')))

        user = tests.FakeUser()

        with tests.user_set(self.app.application, user):

            output = self.app.get('/new/')

            self.assertEqual(output.status_code, 404)

            #just get the csrf token

            pagure.config.config['ENABLE_NEW_PROJECTS'] = True

            output = self.app.get('/new/')

            pagure.config.config['ENABLE_NEW_PROJECTS'] = False

            csrf_token = output.get_data(as_text=True).split(

                'name="csrf_token" type="hidden" value="')[1].split('">')[0]

            data = {

                'description': 'Project #1',

                'name': 'project-1',

            }

        user.username = 'foo'

        with tests.user_set(self.app.application, user):

            data['csrf_token'] =  csrf_token

            output = self.app.post('/new/', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 404)

        #After

        projects = pagure.lib.query.search_projects(self.session)

        self.assertEqual(len(projects), 0)

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'project-1.git')))

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'tickets', 'project-1.git')))

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'docs', 'project-1.git')))

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'requests', 'project-1.git')))

        pagure.config.config['ENABLE_NEW_PROJECTS'] = True

    def test_new_project(self):

        """ Test the new_project endpoint. """

        # Before

        projects = pagure.lib.query.search_projects(self.session)

        self.assertEqual(len(projects), 0)

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'project#1.git')))

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'tickets', 'project#1.git')))

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'docs', 'project#1.git')))

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'requests', 'project#1.git')))

        user = tests.FakeUser()

        with tests.user_set(self.app.application, user):

            output = self.app.get('/new/')

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                'Create new Project', output_text)

            csrf_token = output_text.split(

                'name="csrf_token" type="hidden" value="')[1].split('">')[0]

            data = {

                'description': 'Project #1',

            }

            output = self.app.post('/new/', data=data)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                'Create new Project', output_text)

            self.assertIn(

                '<small>\n            This field is required. \n'</small>

                '          ', output_text)

            data['name'] = 'project-1'

            output = self.app.post('/new/', data=data)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn('Create new Project', output_text)

            self.assertNotIn(

                '<small>\n            This field is required. \n'</small>

                '          ', output_text)

            data['csrf_token'] =  csrf_token

            output = self.app.post('/new/', data=data)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn('Create new Project', output_text)

            self.assertIn(

                'No user '

                '"username" found',

                output_text)

        user.username = 'foo'

        with tests.user_set(self.app.application, user):

            data['csrf_token'] = csrf_token

            output = self.app.post('/new/', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                '
\nProject #1',

                output_text)

            self.assertIn('
This repo is brand new!
',

                          output_text)

            self.assertIn(

                '<title>Overview - project-1 - Pagure</title>', output_text)

        # After

        projects = pagure.lib.query.search_projects(self.session)

        self.assertEqual(len(projects), 1)

        self.assertTrue(os.path.exists(

            os.path.join(self.path, 'repos', 'project-1.git')))

        self.assertTrue(os.path.exists(

            os.path.join(self.path, 'repos', 'tickets', 'project-1.git')))

        self.assertTrue(os.path.exists(

            os.path.join(self.path, 'repos', 'docs', 'project-1.git')))

        self.assertTrue(os.path.exists(

            os.path.join(self.path, 'repos', 'requests', 'project-1.git')))

    @patch.dict('pagure.config.config', {'PAGURE_ADMIN_USERS': ['pingou'],

                                         'ALLOW_ADMIN_IGNORE_EXISTING_REPOS': True})

    def test_adopt_repos(self):

        """ Test the new_project endpoint with existing git repo. """

        # Before

        projects = pagure.lib.query.search_projects(self.session)

        self.assertEqual(len(projects), 0)

        tests.create_projects_git(os.path.join(self.path, 'repos'), bare=True)

        tests.add_content_git_repo(os.path.join(self.path, 'repos', 'test.git'))

        user = tests.FakeUser(username='pingou')

        with tests.user_set(self.app.application, user):

            data = {

                'csrf_token': self.get_csrf(),

                'name': 'test',

                'description': 'Project #1',

            }

            output = self.app.post('/new/', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn('The main repo test.git already exists', output_text)

            data['ignore_existing_repos'] = 'y'

            output = self.app.post('/new/', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn("Alice Author", output_text)

    @patch.dict('pagure.config.config', {'PAGURE_ADMIN_USERS': [],

                                         'USERS_IGNORE_EXISTING_REPOS': ['pingou']})

    def test_adopt_repos_non_admin(self):

        """ Test the new_project endpoint with existing git repo for non-admins. """

        # Before

        projects = pagure.lib.query.search_projects(self.session)

        self.assertEqual(len(projects), 0)

        tests.create_projects_git(os.path.join(self.path, 'repos'), bare=True)

        tests.add_content_git_repo(os.path.join(self.path, 'repos', 'test.git'))

        user = tests.FakeUser(username='pingou')

        with tests.user_set(self.app.application, user):

            data = {

                'csrf_token': self.get_csrf(),

                'name': 'test',

                'description': 'Project #1',

            }

            output = self.app.post('/new/', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn('The main repo test.git already exists', output_text)

            data['ignore_existing_repos'] = 'y'

            output = self.app.post('/new/', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn("Alice Author", output_text)

    @patch.dict('pagure.config.config', {'PAGURE_ADMIN_USERS': [],

                                         'USERS_IGNORE_EXISTING_REPOS': []})

    def test_adopt_repos_not_allowed(self):

        """ Test the new_project endpoint with existing git repo for no access. """

        # Before

        projects = pagure.lib.query.search_projects(self.session)

        self.assertEqual(len(projects), 0)

        tests.create_projects_git(os.path.join(self.path, 'repos'), bare=True)

        tests.add_content_git_repo(os.path.join(self.path, 'repos', 'test.git'))

        user = tests.FakeUser(username='pingou')

        with tests.user_set(self.app.application, user):

            data = {

                'csrf_token': self.get_csrf(),

                'name': 'test',

                'description': 'Project #1',

            }

            output = self.app.post('/new/', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn('The main repo test.git already exists', output_text)

            data['ignore_existing_repos'] = 'y'

            output = self.app.post('/new/', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn('The main repo test.git already exists', output_text)

    @patch.dict('pagure.config.config', {'PROJECT_NAME_REGEX': '^1[a-z]*$'})

    def test_new_project_diff_regex(self):

        """ Test the new_project endpoint with a different regex. """

        # Before

        projects = pagure.lib.query.search_projects(self.session)

        self.assertEqual(len(projects), 0)

        user = tests.FakeUser(username='foo')

        with tests.user_set(self.app.application, user):

            output = self.app.get('/new/')

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                'Create new Project', output_text)

            csrf_token = self.get_csrf(output=output)

            data = {

                'description': 'Project #1',

                'name': 'project-1',

                'csrf_token':  csrf_token,

            }

            output = self.app.post('/new/', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                '<title>New project - Pagure</title>', output_text)

            self.assertIn(

                'Create new Project', output_text)

            self.assertIn(

                '<small>\n            Invalid input. \n'</small>

                '          ', output_text)

    @patch.dict('pagure.config.config', {'PRIVATE_PROJECTS': True})

    def test_new_project_private(self):

        """ Test the new_project endpoint for a private project. """

        # Before

        projects = pagure.lib.query.search_projects(self.session)

        self.assertEqual(len(projects), 0)

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'foo', 'project#1.git')))

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'tickets', 'foo', 'project#1.git')))

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'docs', 'foo', 'project#1.git')))

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'requests', 'foo', 'project#1.git')))

        user = tests.FakeUser()

        with tests.user_set(self.app.application, user):

            output = self.app.get('/new/')

            self.assertEqual(output.status_code, 200)

            self.assertIn(

                'Create new Project', output.get_data(as_text=True))

            csrf_token = self.get_csrf(output=output)

            data = {

                'description': 'Project #1',

                'private': True,

            }

            output = self.app.post('/new/', data=data)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                'Create new Project', output_text)

            self.assertIn(

                '<small>\n            This field is required. \n'</small>

                '          ', output_text)

            data['name'] = 'project-1'

            output = self.app.post('/new/', data=data)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn('Create new Project', output_text)

            self.assertNotIn(

                '<small>\n            This field is required. \n'</small>

                '          ', output_text)

            data['csrf_token'] =  csrf_token

            output = self.app.post('/new/', data=data)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn('Create new Project', output_text)

            self.assertIn(

                'No user '

                '"username" found',

                output_text)

        user.username = 'foo'

        with tests.user_set(self.app.application, user):

            data['csrf_token'] =  csrf_token

            output = self.app.post('/new/', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                '
\nProject #1',

                output_text)

            self.assertIn('
This repo is brand new!
',

                          output_text)

            self.assertIn(

                '<title>Overview - foo/project-1 - Pagure</title>', output_text)

        # After

        projects = pagure.lib.query.search_projects(self.session)

        self.assertEqual(len(projects), 0)

        projects = pagure.lib.query.search_projects(self.session, private=True)

        self.assertEqual(len(projects), 1)

        self.assertTrue(os.path.exists(

            os.path.join(self.path, 'repos', 'foo', 'project-1.git')))

        self.assertTrue(os.path.exists(

            os.path.join(self.path, 'repos', 'tickets', 'foo', 'project-1.git')))

        self.assertTrue(os.path.exists(

            os.path.join(self.path, 'repos', 'docs', 'foo', 'project-1.git')))

        self.assertTrue(os.path.exists(

            os.path.join(self.path, 'repos', 'requests', 'foo', 'project-1.git')))

    def test_non_ascii_new_project(self):

        """ Test the new_project endpoint with a non-ascii project. """

        # Before

        projects = pagure.lib.query.search_projects(self.session)

        self.assertEqual(len(projects), 0)

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'project-1.git')))

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'tickets', 'project-1.git')))

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'docs', 'project-1.git')))

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'requests', 'project-1.git')))

        user = tests.FakeUser()

        user.username = 'foo'

        with tests.user_set(self.app.application, user):

            output = self.app.get('/new/')

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                'Create new Project', output_text)

            csrf_token = output_text.split(

                'name="csrf_token" type="hidden" value="')[1].split('">')[0]

            data = {

                'description': 'Prõjéctö #1',

                'name': 'project-1',

                'csrf_token':  csrf_token,

                'create_readme': True,

            }

            output = self.app.post('/new/', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                '
\nPrõjéctö #1',

                output_text)

            self.assertIn(

                '''<section class="readme"></section>

              
project-1

Prõjéctö #1

            ''',

            output_text)

            data = {

                'description': 'Мой первый суперский репозиторий',

                'name': 'project-2',

                'csrf_token':  csrf_token,

                'create_readme': True,

            }

            output = self.app.post('/new/', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                '
\nМой первый суперский репозиторий',

                output_text)

            self.assertIn(

                '''<section class="readme"></section>

              
project-2

Мой первый суперский репозиторий

            ''',

            output_text)

        # After

        projects = pagure.lib.query.search_projects(self.session)

        self.assertEqual(len(projects), 2)

        for project in ['project-1', 'project-2']:

            self.assertTrue(os.path.exists(

                os.path.join(self.path, 'repos', '%s.git' % project)))

            self.assertTrue(os.path.exists(

                os.path.join(self.path, 'repos', 'tickets', '%s.git' % project)))

            self.assertTrue(os.path.exists(

                os.path.join(self.path, 'repos', 'docs', '%s.git' % project)))

            self.assertTrue(os.path.exists(

                os.path.join(self.path, 'repos', 'requests', '%s.git' % project)))

    @patch('pygit2.init_repository', wraps=pygit2.init_repository)

    def test_new_project_with_template(self, pygit2init):

        """ Test the new_project endpoint for a new project with a template set.

        """

        # Before

        projects = pagure.lib.query.search_projects(self.session)

        self.assertEqual(len(projects), 0)

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'project-1.git')))

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'tickets', 'project-1.git')))

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'docs', 'project-1.git')))

        self.assertFalse(os.path.exists(

            os.path.join(self.path, 'repos', 'requests', 'project-1.git')))

        user = tests.FakeUser()

        user.username = 'foo'

        with tests.user_set(self.app.application, user):

            output = self.app.get('/new/')

            self.assertEqual(output.status_code, 200)

            self.assertIn(

                'Create new Project', output.get_data(as_text=True))

            csrf_token = self.get_csrf(output=output)

            data = {

                'description': 'test',

                'name': 'project-1',

                'csrf_token':  csrf_token,

                'create_readme': True,

            }

            output = self.app.post('/new/', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            self.assertIn(

                '
\ntest',

                output.get_data(as_text=True))

            self.assertEqual(pygit2init.call_count, 4)

            pygit2init.assert_any_call(

                '%s/repos/project-1.git' % self.path,

                bare=True, template_path=None)

            path = os.path.join(self.path, 'repos', 'project-1.git')

            with patch.dict(

                    'pagure.config.config',

                    {'PROJECT_TEMPLATE_PATH': path}):

                data = {

                    'description': 'test2',

                    'name': 'project-2',

                    'csrf_token':  csrf_token,

                    'create_readme': True,

                }

                output = self.app.post('/new/', data=data, follow_redirects=True)

                self.assertEqual(output.status_code, 200)

                self.assertIn(

                    '
\ntest2',

                    output.get_data(as_text=True))

            self.assertEqual(pygit2init.call_count, 8)

            pygit2init.assert_any_call(

                '%s/repos/project-2.git' % self.path,

                bare=True,

                template_path='%s/repos/project-1.git' % self.path)

        # After

        projects = pagure.lib.query.search_projects(self.session)

        self.assertEqual(len(projects), 2)

        for project in ['project-1', 'project-2']:

            self.assertTrue(os.path.exists(

                os.path.join(self.path, 'repos', '%s.git' % project)))

            self.assertTrue(os.path.exists(

                os.path.join(self.path, 'repos', 'tickets', '%s.git' % project)))

            self.assertTrue(os.path.exists(

                os.path.join(self.path, 'repos', 'docs', '%s.git' % project)))

            self.assertTrue(os.path.exists(

                os.path.join(self.path, 'repos', 'requests', '%s.git' % project)))

    @patch.dict('pagure.config.config', {'CASE_SENSITIVE': True})

    def test_new_project_case_sensitive(self):

        tests.create_projects(self.session)

        tests.create_projects_git(os.path.join(self.path, 'repos'), bare=True)

        output = self.app.get('/test')

        self.assertEqual(output.status_code, 200)

        output = self.app.get('/TEST')

        self.assertEqual(output.status_code, 404)

        user = tests.FakeUser()

        user.username = 'foo'

        with tests.user_set(self.app.application, user):

            output = self.app.get('/new/')

            self.assertEqual(output.status_code, 200)

            csrf_token = self.get_csrf(output=output)

            data = {

                'description': 'TEST',

                'name': 'TEST',

                'csrf_token':  csrf_token,

                'create_readme': True,

            }

            self.app.post('/new/', data=data, follow_redirects=True)

            output = self.app.get('/TEST')

            self.assertEqual(output.status_code, 200)

    @patch('pagure.ui.app.admin_session_timedout')

    def test_user_settings(self, ast):

        """ Test the user_settings endpoint. """

        ast.return_value = False

        self.test_new_project()

        user = tests.FakeUser()

        with tests.user_set(self.app.application, user):

            output = self.app.get('/settings/')

            self.assertEqual(output.status_code, 404)

            self.assertIn('
Page not found (404)
', output.get_data(as_text=True))

        user.username = 'foo'

        with tests.user_set(self.app.application, user):

            output = self.app.get('/settings/')

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                '<title>foo\'s settings - Pagure</title>', output_text)

            ast.return_value = True

            output = self.app.get('/settings/')

            self.assertEqual(output.status_code, 302)

    @patch('pagure.decorators.admin_session_timedout')

    def test_add_user_sshkey(self, ast):

        """ Test the add_user_sshkey endpoint. """

        ast.return_value = False

        # User not logged in

        output = self.app.get('/settings/')

        self.assertEqual(output.status_code, 302)

        ast.return_value = False

        user = tests.FakeUser(username='pingou')

        with tests.user_set(self.app.application, user):

            output = self.app.get('/settings', follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn('Add SSH key', output_text)

            csrf_token = self.get_csrf(output=output)

            data = {

                'ssh_key': 'asdf',

            }

            # No CSRF token

            output = self.app.post(

                '/settings/usersettings/addkey', data=data , follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn('Add SSH key', output_text)

            data['csrf_token'] = csrf_token

            # First, invalid SSH key

            output = self.app.post(

                '/settings/usersettings/addkey', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn('Add SSH key', output_text)

            self.assertIn('SSH key invalid', output_text)

            # Next up, multiple SSH keys

            data['ssh_key'] = 'ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAgQDAzBMSIlvPRaEiLOTVInErkRIw9CzQQcnslDekAn1jFnGf+SNa1acvbTiATbCX71AA03giKrPxPH79dxcC7aDXerc6zRcKjJs6MAL9PrCjnbyxCKXRNNZU5U9X/DLaaL1b3caB+WD6OoorhS3LTEtKPX8xyjOzhf3OQSzNjhJp5Q==\nssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAgQDAzBMSIlvPRaEiLOTVInErkRIw9CzQQcnslDekAn1jFnGf+SNa1acvbTiATbCX71AA03giKrPxPH79dxcC7aDXerc6zRcKjJs6MAL9PrCjnbyxCKXRNNZU5U9X/DLaaL1b3caB+WD6OoorhS3LTEtKPX8xyjOzhf3OQSzNjhJp5Q=='

            output = self.app.post(

                '/settings/usersettings/addkey', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn('Please add single SSH keys.', output_text)

            # Now, a valid SSH key

            data['ssh_key'] = 'ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAgQDAzBMSIlvPRaEiLOTVInErkRIw9CzQQcnslDekAn1jFnGf+SNa1acvbTiATbCX71AA03giKrPxPH79dxcC7aDXerc6zRcKjJs6MAL9PrCjnbyxCKXRNNZU5U9X/DLaaL1b3caB+WD6OoorhS3LTEtKPX8xyjOzhf3OQSzNjhJp5Q=='

            output = self.app.post(

                '/settings/usersettings/addkey', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                "<title>pingou's settings - Pagure</title>", output_text)

            self.assertIn('SSH key added', output_text)

            self.assertNotIn('Push Access', output_text)

            # And now, adding the same key

            output = self.app.post(

                '/settings/usersettings/addkey', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn('SSH key already exists', output_text)

            # And next, a key with push access

            data['ssh_key'] = 'ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAgQC9Xwc2RDzPBhlEDARfHldGjudIVoa04tqT1JVKGQmyllTFz7Rb8CngQL3e7zyNzotnhwYKHdoiLlPkVEiDee4dWMUe48ilqId+FJZQGhyv8fu4BoFdE1AJUVylzmltbLg14VqG5gjTpXgtlrEva9arKwBMHJjRYc8ScaSn3OgyQw=='

            output = self.app.post(

                '/settings/usersettings/addkey', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                "<title>pingou's settings - Pagure</title>", output_text)

            self.assertIn('SSH key added', output_text)

    @patch('pagure.decorators.admin_session_timedout')

    def test_remove_user_sshkey(self, ast):

        """ Test the remove_sshkey endpoint. """

        ast.return_value = False

        user = tests.FakeUser()

        # User not logged in

        output = self.app.post('/settings/usersettings/removekey/1')

        self.assertEqual(output.status_code, 302)

        user.username = 'pingou'

        with tests.user_set(self.app.application, user):

            data = {'csrf_token': self.get_csrf()}

            output = self.app.post(