123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928 |
- # -*- coding: utf-8 -*-
- """
- (c) 2015-2016 - Copyright Red Hat Inc
- Authors:
- Pierre-Yves Chibon <pingou@pingoured.fr>
- """
- __requires__ = ['SQLAlchemy >= 0.8']
- import pkg_resources
- import unittest
- import shutil
- import sys
- import os
- import six
- import json
- from mock import patch
- sys.path.insert(0, os.path.join(os.path.dirname(
- os.path.abspath(__file__)), '..'))
- import pagure.lib
- import tests
- class PagureFlaskApptests(tests.Modeltests):
- """ Tests for flask app controller of pagure """
- def setUp(self):
- """ Set up the environnment, ran before every tests. """
- super(PagureFlaskApptests, self).setUp()
- pagure.APP.config['TESTING'] = True
- pagure.SESSION = self.session
- pagure.ui.SESSION = self.session
- pagure.ui.app.SESSION = self.session
- pagure.ui.filters.SESSION = self.session
- pagure.ui.repo.SESSION = self.session
- pagure.APP.config['GIT_FOLDER'] = self.path
- pagure.APP.config['FORK_FOLDER'] = os.path.join(
- self.path, 'forks')
- pagure.APP.config['TICKETS_FOLDER'] = os.path.join(
- self.path, 'tickets')
- pagure.APP.config['DOCS_FOLDER'] = os.path.join(
- self.path, 'docs')
- pagure.APP.config['REQUESTS_FOLDER'] = os.path.join(
- self.path, 'requests')
- self.app = pagure.APP.test_client()
- def test_index(self):
- """ Test the index endpoint. """
- output = self.app.get('/')
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<h2 class="m-b-1">All Projects '
- '<span class="label label-default">0</span></h2>', output.data)
- tests.create_projects(self.session)
- output = self.app.get('/?page=abc')
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<h2 class="m-b-1">All Projects '
- '<span class="label label-default">2</span></h2>', output.data)
- # Add a 3rd project with a long description
- item = pagure.lib.model.Project(
- user_id=2, # foo
- name='test3',
- description='test project #3 with a very long description',
- hook_token='aaabbbeee',
- )
- self.session.add(item)
- self.session.commit()
- user = tests.FakeUser(username='foo')
- with tests.user_set(pagure.APP, user):
- output = self.app.get('/?repopage=abc&forkpage=def')
- self.assertIn(
- 'Projects <span class="label label-default">1</span>',
- output.data)
- self.assertIn(
- 'Forks <span class="label label-default">0</span>',
- output.data)
- self.assertEqual(
- output.data.count('<p>No group found</p>'), 1)
- self.assertEqual(
- output.data.count('<div class="card-header">'), 6)
- def test_watch_list(self):
- ''' Test for watch list of a user '''
- user = tests.FakeUser(username='pingou')
- with tests.user_set(pagure.APP, user):
- output = self.app.get('/')
- self.assertIn(
- '<div class="text-xs-center">You have no projects</div>',
- output.data)
- self.assertIn(
- '<p>You have no forks</p>',
- output.data)
- self.assertIn(
- '<p>No project in watch list</p>',
- output.data)
- tests.create_projects(self.session)
- output = self.app.get('/')
- self.assertIn(
- 'My Projects <span class="label label-default">2</span>',
- output.data)
- self.assertIn(
- 'My Forks <span class="label label-default">0</span>',
- output.data)
- self.assertIn(
- 'My Watch List <span class="label label-default">2</span>',
- output.data)
- def test_view_users(self):
- """ Test the view_users endpoint. """
- output = self.app.get('/users/?page=abc')
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<h2 class="m-b-1">\n Users '
- '<span class="label label-default">2</span></h2>', output.data)
- self.assertIn(
- '<a class="project_link logo_link" href="/user/pingou">',
- output.data)
- self.assertIn(
- '<a class="project_link logo_link" href="/user/foo">',
- output.data)
- 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)
- self.assertIn(
- 'Projects <span class="label label-default">0</span>',
- output.data)
- self.assertIn(
- 'Forks <span class="label label-default">0</span>',
- output.data)
- tests.create_projects(self.session)
- self.gitrepos = tests.create_projects_git(
- pagure.APP.config['GIT_FOLDER'])
- output = self.app.get('/user/pingou?repopage=abc&forkpage=def')
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- 'Projects <span class="label label-default">2</span>',
- output.data)
- self.assertIn(
- 'Forks <span class="label label-default">0</span>', output.data)
- 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.APP.config['ENABLE_NEW_PROJECTS'] = False
- # Before
- projects = pagure.lib.search_projects(self.session)
- self.assertEqual(len(projects), 0)
- self.assertFalse(os.path.exists(
- os.path.join(self.path, 'project-1.git')))
- self.assertFalse(os.path.exists(
- os.path.join(self.path, 'tickets', 'project-1.git')))
- self.assertFalse(os.path.exists(
- os.path.join(self.path, 'docs', 'project-1.git')))
- self.assertFalse(os.path.exists(
- os.path.join(self.path, 'requests', 'project-1.git')))
- user = tests.FakeUser()
- with tests.user_set(pagure.APP, user):
- output = self.app.get('/new/')
- self.assertEqual(output.status_code, 404)
- #just get the csrf token
- pagure.APP.config['ENABLE_NEW_PROJECTS'] = True
- output = self.app.get('/new/')
- pagure.APP.config['ENABLE_NEW_PROJECTS'] = False
- csrf_token = output.data.split(
- 'name="csrf_token" type="hidden" value="')[1].split('">')[0]
- data = {
- 'description': 'Project #1',
- 'name': 'project-1',
- }
- user.username = 'foo'
- with tests.user_set(pagure.APP, 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.search_projects(self.session)
- self.assertEqual(len(projects), 0)
- self.assertFalse(os.path.exists(
- os.path.join(self.path, 'project-1.git')))
- self.assertFalse(os.path.exists(
- os.path.join(self.path, 'tickets', 'project-1.git')))
- self.assertFalse(os.path.exists(
- os.path.join(self.path, 'docs', 'project-1.git')))
- self.assertFalse(os.path.exists(
- os.path.join(self.path, 'requests', 'project-1.git')))
- pagure.APP.config['ENABLE_NEW_PROJECTS'] = True
- def test_new_project(self):
- """ Test the new_project endpoint. """
- # Before
- projects = pagure.lib.search_projects(self.session)
- self.assertEqual(len(projects), 0)
- self.assertFalse(os.path.exists(
- os.path.join(self.path, 'project#1.git')))
- self.assertFalse(os.path.exists(
- os.path.join(self.path, 'tickets', 'project#1.git')))
- self.assertFalse(os.path.exists(
- os.path.join(self.path, 'docs', 'project#1.git')))
- self.assertFalse(os.path.exists(
- os.path.join(self.path, 'requests', 'project#1.git')))
- user = tests.FakeUser()
- with tests.user_set(pagure.APP, user):
- output = self.app.get('/new/')
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- u'<strong>Create new Project</strong>', output.data)
- csrf_token = output.data.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)
- self.assertIn(
- u'<strong>Create new Project</strong>', output.data)
- self.assertIn(
- u'<small>\n This field is required. \n'
- ' </small>', output.data)
- data['name'] = 'project-1'
- output = self.app.post('/new/', data=data)
- self.assertEqual(output.status_code, 200)
- self.assertIn('<strong>Create new Project</strong>', output.data)
- self.assertNotIn(
- u'<small>\n This field is required. \n'
- ' </small>', output.data)
- data['csrf_token'] = csrf_token
- output = self.app.post('/new/', data=data)
- self.assertEqual(output.status_code, 200)
- self.assertIn('<strong>Create new Project</strong>', output.data)
- self.assertIn(
- u'</button>\n No user '
- '"username" found\n </div>',
- output.data)
- user.username = 'foo'
- with tests.user_set(pagure.APP, user):
- data['csrf_token'] = csrf_token
- output = self.app.post('/new/', data=data, follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- u'<div class="projectinfo m-t-1 m-b-1">\nProject #1 </div>',
- output.data)
- self.assertIn(u'<p>This repo is brand new!</p>', output.data)
- self.assertIn(
- u'<title>Overview - project-1 - Pagure</title>', output.data)
- # After
- projects = pagure.lib.search_projects(self.session)
- self.assertEqual(len(projects), 1)
- self.assertTrue(os.path.exists(
- os.path.join(self.path, 'project-1.git')))
- self.assertTrue(os.path.exists(
- os.path.join(self.path, 'tickets', 'project-1.git')))
- self.assertTrue(os.path.exists(
- os.path.join(self.path, 'docs', 'project-1.git')))
- self.assertTrue(os.path.exists(
- os.path.join(self.path, 'requests', 'project-1.git')))
- def test_non_ascii_new_project(self):
- """ Test the new_project endpoint with a non-ascii project. """
- # Before
- projects = pagure.lib.search_projects(self.session)
- self.assertEqual(len(projects), 0)
- self.assertFalse(os.path.exists(
- os.path.join(self.path, 'project-1.git')))
- self.assertFalse(os.path.exists(
- os.path.join(self.path, 'tickets', 'project-1.git')))
- self.assertFalse(os.path.exists(
- os.path.join(self.path, 'docs', 'project-1.git')))
- self.assertFalse(os.path.exists(
- os.path.join(self.path, 'requests', 'project-1.git')))
- user = tests.FakeUser()
- user.username = 'foo'
- with tests.user_set(pagure.APP, user):
- output = self.app.get('/new/')
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- b'<strong>Create new Project</strong>', output.data)
- csrf_token = output.data.decode('utf-8').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)
- self.assertIn(
- '<div class="projectinfo m-t-1 m-b-1">\nPrõjéctö #1 </div>',
- output.data if six.PY2 else output.data.decode('utf-8'))
- self.assertIn(
- '''<section class="readme">
- <h1>project-1</h1>
- <p>Prõjéctö #1</p>
- </section>''', output.data if six.PY2 else output.data.decode('utf-8'))
- 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)
- self.assertIn(
- '<div class="projectinfo m-t-1 m-b-1">\nМой первый суперский репозиторий </div>',
- output.data if six.PY2 else output.data.decode('utf-8'))
- self.assertIn(
- '''<section class="readme">
- <h1>project-2</h1>
- <p>Мой первый суперский репозиторий</p>
- </section>''', output.data if six.PY2 else output.data.decode('utf-8'))
- # After
- projects = pagure.lib.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, '%s.git' % project)))
- self.assertTrue(os.path.exists(
- os.path.join(self.path, 'tickets', '%s.git' % project)))
- self.assertTrue(os.path.exists(
- os.path.join(self.path, 'docs', '%s.git' % project)))
- self.assertTrue(os.path.exists(
- os.path.join(self.path, 'requests', '%s.git' % project)))
- @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(pagure.APP, user):
- output = self.app.get('/settings/')
- self.assertEqual(output.status_code, 404)
- self.assertTrue('<h2>Page not found (404)</h2>' in output.data)
- user.username = 'foo'
- with tests.user_set(pagure.APP, user):
- output = self.app.get('/settings/')
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertIn(
- '<textarea class="form-control" id="ssh_key" name="ssh_key">'
- '</textarea>', output.data)
- csrf_token = output.data.split(
- 'name="csrf_token" type="hidden" value="')[1].split('">')[0]
- data = {
- 'ssh_key': 'blah'
- }
- output = self.app.post('/settings/', data=data)
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- data['csrf_token'] = csrf_token
- output = self.app.post(
- '/settings/', data=data, follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertIn('Invalid SSH keys', output.data)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertIn('>blah</textarea>', output.data)
- csrf_token = output.data.split(
- 'name="csrf_token" type="hidden" value="')[1].split('">')[0]
- data = {
- 'ssh_key': 'ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAgQDUkub32fZnNI'
- '1zJYs43vhhx3c6IcYo4yzhw1gQ37BLhrrNeS6x8l5PKX4J8ZP5'
- '1XhViPaLbeOpl94Vm5VSCbLy0xtY9KwLhMkbKj7g6vvfxLm2sT'
- 'Osb15j4jzIkUYYgIE7cHhZMCLWR6UA1c1HEzo6mewMDsvpQ9wk'
- 'cDnAuXjK3Q==',
- 'csrf_token': csrf_token
- }
- output = self.app.post(
- '/settings/', data=data, follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertIn('Public ssh key updated', output.data)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertIn(
- '<textarea class="form-control" id="ssh_key" name="ssh_key">'
- 'ssh-rsa AAAA', output.data)
- ast.return_value = True
- output = self.app.get('/settings/')
- self.assertEqual(output.status_code, 302)
- def test_markdown_preview(self):
- """ Test the markdown_preview endpoint. """
- data = {
- 'content': 'test\n----\n\n * 1\n * item 2'
- }
- # CSRF missing
- output = self.app.post('/markdown/', data=data)
- self.assertEqual(output.status_code, 400)
- user = tests.FakeUser()
- user.username = 'foo'
- with tests.user_set(pagure.APP, user):
- output = self.app.get('/settings/')
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertIn(
- '<textarea class="form-control" id="ssh_key" name="ssh_key">'
- '</textarea>', output.data)
- csrf_token = output.data.split(
- 'name="csrf_token" type="hidden" value="')[1].split('">')[0]
- # With CSRF
- data['csrf_token'] = csrf_token
- output = self.app.post('/markdown/', data=data)
- self.assertEqual(output.status_code, 200)
- exp = """<h2>test</h2>
- <ul>
- <li>1</li>
- <li>item 2</li>
- </ul>"""
- self.assertEqual(output.data, exp)
- @patch('pagure.ui.app.admin_session_timedout')
- def test_remove_user_email(self, ast):
- """ Test the remove_user_email endpoint. """
- ast.return_value = False
- self.test_new_project()
- user = tests.FakeUser()
- with tests.user_set(pagure.APP, user):
- output = self.app.post('/settings/email/drop')
- self.assertEqual(output.status_code, 404)
- self.assertTrue('<h2>Page not found (404)</h2>' in output.data)
- user.username = 'foo'
- with tests.user_set(pagure.APP, user):
- output = self.app.post('/settings/')
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertIn(
- '<textarea class="form-control form-control-error" id="ssh_key" name="ssh_key">'
- '</textarea>', output.data)
- csrf_token = output.data.split(
- 'name="csrf_token" type="hidden" value="')[1].split('">')[0]
- data = {
- 'email': 'foo@pingou.com',
- }
- output = self.app.post(
- '/settings/email/drop', data=data, follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertIn(
- '<textarea class="form-control" id="ssh_key" name="ssh_key">'
- '</textarea>', output.data)
- self.assertIn(
- '</button>\n You must always have at least one email',
- output.data)
- user.username = 'pingou'
- with tests.user_set(pagure.APP, user):
- output = self.app.post('/settings/')
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertIn(
- '<textarea class="form-control form-control-error" id="ssh_key" name="ssh_key">'
- '</textarea>', output.data)
- csrf_token = output.data.split(
- 'name="csrf_token" type="hidden" value="')[1].split('">')[0]
- data = {
- 'email': 'foo@pingou.com',
- }
- output = self.app.post(
- '/settings/email/drop', data=data, follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertEqual(output.data.count('foo@pingou.com'), 4)
- data = {
- 'csrf_token': csrf_token,
- 'email': 'foobar@pingou.com',
- }
- output = self.app.post(
- '/settings/email/drop', data=data, follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertIn(
- '</button>\n You do not have the '
- 'email: foobar@pingou.com, nothing to remove', output.data)
- data = {
- 'csrf_token': csrf_token,
- 'email': 'foo@pingou.com',
- }
- output = self.app.post(
- '/settings/email/drop', data=data, follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertEqual(output.data.count('foo@pingou.com'), 0)
- self.assertEqual(output.data.count('bar@pingou.com'), 3)
- output = self.app.post(
- '/settings/email/drop', data=data, follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertEqual(output.data.count('foo@pingou.com'), 0)
- self.assertEqual(output.data.count('bar@pingou.com'), 3)
- ast.return_value = True
- output = self.app.post('/settings/email/drop', data=data)
- self.assertEqual(output.status_code, 302)
- @patch('pagure.lib.notify.send_email')
- @patch('pagure.ui.app.admin_session_timedout')
- def test_add_user_email(self, ast, send_email):
- """ Test the add_user_email endpoint. """
- send_email.return_value = True
- ast.return_value = False
- self.test_new_project()
- user = tests.FakeUser()
- with tests.user_set(pagure.APP, user):
- output = self.app.post('/settings/email/add')
- self.assertEqual(output.status_code, 404)
- self.assertTrue('<h2>Page not found (404)</h2>' in output.data)
- user.username = 'foo'
- with tests.user_set(pagure.APP, user):
- output = self.app.post('/settings/email/add')
- self.assertEqual(output.status_code, 200)
- self.assertTrue("<strong>Add new email</strong>" in output.data)
- self.assertIn(
- '<input class="form-control form-control-error" id="email" '
- 'name="email" type="text" value="">', output.data)
- user.username = 'pingou'
- with tests.user_set(pagure.APP, user):
- output = self.app.post('/settings/email/add')
- self.assertEqual(output.status_code, 200)
- self.assertTrue("<strong>Add new email</strong>" in output.data)
- self.assertIn(
- '<input class="form-control form-control-error" id="email" '
- 'name="email" type="text" value="">', output.data)
- csrf_token = output.data.split(
- 'name="csrf_token" type="hidden" value="')[1].split('">')[0]
- data = {
- 'email': 'foo2@pingou.com',
- }
- output = self.app.post(
- '/settings/email/add', data=data, follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertTrue("<strong>Add new email</strong>" in output.data)
- self.assertEqual(output.data.count('foo2@pingou.com'), 1)
- # New email
- data = {
- 'csrf_token': csrf_token,
- 'email': 'foobar@pingou.com',
- }
- output = self.app.post(
- '/settings/email/add', data=data, follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertIn(
- '</button>\n Email pending validation',
- output.data)
- self.assertEqual(output.data.count('foo@pingou.com'), 4)
- self.assertEqual(output.data.count('bar@pingou.com'), 5)
- self.assertEqual(output.data.count('foobar@pingou.com'), 2)
- # Email already pending
- output = self.app.post(
- '/settings/email/add', data=data, follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n '
- '<strong>Add new email</strong>', output.data)
- self.assertIn(
- '</button>\n This email is already '
- 'pending confirmation', output.data)
- # User already has this email
- data = {
- 'csrf_token': csrf_token,
- 'email': 'foo@pingou.com',
- }
- output = self.app.post(
- '/settings/email/add', data=data, follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertTrue("<strong>Add new email</strong>" in output.data)
- self.assertTrue(
- 'Invalid value, can't be any of: bar@pingou.com, '
- 'foo@pingou.com. ' in output.data
- or
- 'Invalid value, can't be any of: foo@pingou.com, '
- 'bar@pingou.com. ' in output.data
- )
- self.assertEqual(output.data.count('foo@pingou.com'), 6)
- self.assertEqual(output.data.count('bar@pingou.com'), 5)
- self.assertEqual(output.data.count('foobar@pingou.com'), 0)
- # Email registered by someone else
- data = {
- 'csrf_token': csrf_token,
- 'email': 'foo@bar.com',
- }
- output = self.app.post(
- '/settings/email/add', data=data, follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertTrue("<strong>Add new email</strong>" in output.data)
- self.assertIn(
- 'Invalid value, can't be any of: foo@bar.com. ',
- output.data)
- ast.return_value = True
- output = self.app.post('/settings/email/add', data=data)
- self.assertEqual(output.status_code, 302)
- @patch('pagure.lib.notify.send_email')
- @patch('pagure.ui.app.admin_session_timedout')
- def test_set_default_email(self, ast, send_email):
- """ Test the set_default_email endpoint. """
- send_email.return_value = True
- ast.return_value = False
- self.test_new_project()
- user = tests.FakeUser()
- with tests.user_set(pagure.APP, user):
- output = self.app.post('/settings/email/default')
- self.assertEqual(output.status_code, 404)
- self.assertTrue('<h2>Page not found (404)</h2>' in output.data)
- user.username = 'pingou'
- with tests.user_set(pagure.APP, user):
- output = self.app.get('/settings/')
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertIn(
- '<textarea class="form-control" id="ssh_key" name="ssh_key">'
- '</textarea>', output.data)
- csrf_token = output.data.split(
- 'name="csrf_token" type="hidden" value="')[1].split('">')[0]
- data = {
- 'email': 'foo@pingou.com',
- }
- output = self.app.post(
- '/settings/email/default', data=data, follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertEqual(output.data.count('foo@pingou.com'), 4)
- # Set invalid default email
- data = {
- 'csrf_token': csrf_token,
- 'email': 'foobar@pingou.com',
- }
- output = self.app.post(
- '/settings/email/default', data=data, follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertEqual(output.data.count('foo@pingou.com'), 4)
- self.assertIn(
- '</button>\n You do not have the '
- 'email: foobar@pingou.com, nothing to set',
- output.data)
- # Set default email
- data = {
- 'csrf_token': csrf_token,
- 'email': 'foo@pingou.com',
- }
- output = self.app.post(
- '/settings/email/default', data=data, follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertEqual(output.data.count('foo@pingou.com'), 4)
- self.assertIn(
- '</button>\n Default email set to: '
- 'foo@pingou.com', output.data)
- ast.return_value = True
- output = self.app.post('/settings/email/default', data=data)
- self.assertEqual(output.status_code, 302)
- @patch('pagure.lib.notify.send_email')
- @patch('pagure.ui.app.admin_session_timedout')
- def test_reconfirm_email(self, ast, send_email):
- """ Test the reconfirm_email endpoint. """
- send_email.return_value = True
- ast.return_value = False
- self.test_new_project()
- # Add a pending email to pingou
- userobj = pagure.lib.search_user(self.session, username='pingou')
- self.assertEqual(len(userobj.emails), 2)
- email_pend = pagure.lib.model.UserEmailPending(
- user_id=userobj.id,
- email='foo@fp.o',
- token='abcdef',
- )
- self.session.add(email_pend)
- self.session.commit()
- user = tests.FakeUser()
- with tests.user_set(pagure.APP, user):
- output = self.app.post('/settings/email/resend')
- self.assertEqual(output.status_code, 404)
- self.assertTrue('<h2>Page not found (404)</h2>' in output.data)
- user.username = 'pingou'
- with tests.user_set(pagure.APP, user):
- output = self.app.get('/settings/')
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertIn(
- '<textarea class="form-control" id="ssh_key" name="ssh_key">'
- '</textarea>', output.data)
- csrf_token = output.data.split(
- 'name="csrf_token" type="hidden" value="')[1].split('">')[0]
- data = {
- 'email': 'foo@pingou.com',
- }
- output = self.app.post(
- '/settings/email/resend', data=data, follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertEqual(output.data.count('foo@pingou.com'), 4)
- # Set invalid default email
- data = {
- 'csrf_token': csrf_token,
- 'email': 'foobar@pingou.com',
- }
- output = self.app.post(
- '/settings/email/resend', data=data, follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertEqual(output.data.count('foo@pingou.com'), 4)
- self.assertIn(
- '</button>\n This email address has '
- 'already been confirmed', output.data)
- # Validate a non-validated email
- data = {
- 'csrf_token': csrf_token,
- 'email': 'foo@fp.o',
- }
- output = self.app.post(
- '/settings/email/resend', data=data, follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertEqual(output.data.count('foo@pingou.com'), 4)
- self.assertIn(
- '</button>\n Confirmation email re-sent',
- output.data)
- ast.return_value = True
- output = self.app.post('/settings/email/resend', data=data)
- self.assertEqual(output.status_code, 302)
- @patch('pagure.ui.app.admin_session_timedout')
- def test_confirm_email(self, ast):
- """ Test the confirm_email endpoint. """
- output = self.app.get('/settings/email/confirm/foobar')
- self.assertEqual(output.status_code, 302)
- ast.return_value = False
- # Add a pending email to pingou
- userobj = pagure.lib.search_user(self.session, username='pingou')
- self.assertEqual(len(userobj.emails), 2)
- email_pend = pagure.lib.model.UserEmailPending(
- user_id=userobj.id,
- email='foo@fp.o',
- token='abcdef',
- )
- self.session.add(email_pend)
- self.session.commit()
- user = tests.FakeUser()
- user.username = 'pingou'
- with tests.user_set(pagure.APP, user):
- # Wrong token
- output = self.app.get(
- '/settings/email/confirm/foobar', follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertIn(
- '</button>\n No email associated with this token.',
- output.data)
- # Confirm email
- output = self.app.get(
- '/settings/email/confirm/abcdef', follow_redirects=True)
- self.assertEqual(output.status_code, 200)
- self.assertIn(
- '<div class="card-header">\n Basic Information\n'
- ' </div>', output.data)
- self.assertIn(
- '</button>\n Email validated',
- output.data)
- userobj = pagure.lib.search_user(self.session, username='pingou')
- self.assertEqual(len(userobj.emails), 3)
- ast.return_value = True
- output = self.app.get('/settings/email/confirm/foobar')
- self.assertEqual(output.status_code, 302)
- if __name__ == '__main__':
- SUITE = unittest.TestLoader().loadTestsFromTestCase(PagureFlaskApptests)
- unittest.TextTestRunner(verbosity=2).run(SUITE)
|