diff --git a/tests/test_pagure_flask_ui_repo.py b/tests/test_pagure_flask_ui_repo.py index daf623a..3cf785f 100644 --- a/tests/test_pagure_flask_ui_repo.py +++ b/tests/test_pagure_flask_ui_repo.py @@ -1490,8 +1490,30 @@ class PagureFlaskRepotests(tests.Modeltests): tests.add_readme_git_repo(os.path.join(self.path, 'repos', 'test.git'), 'README.dummy') self.perfReset() + # Authenticated, the Fork button appears + user = tests.FakeUser(username='pingou') + with tests.user_set(self.app.application, user): + output = self.app.get('/test') + self.assertEqual(output.status_code, 200) + self.assertIn( + 'Fork', + output.data) + self.assertFalse('
This repo is brand new!
' in output.data) + self.assertFalse('Forked from' in output.data) + self.assertFalse('README.txt' in output.data) + self.assertFalse('README.dummy' in output.data) + self.assertIn( + 'This repo is brand new!
' in output.data) self.assertFalse('Forked from' in output.data) self.assertFalse('README.txt' in output.data) @@ -1519,6 +1541,45 @@ class PagureFlaskRepotests(tests.Modeltests): tests.add_readme_git_repo( os.path.join(self.path, 'repos', 'forks', 'pingou', 'test.git')) + # Authenticated and already have a fork, the View Fork button appears + user = tests.FakeUser(username='pingou') + with tests.user_set(self.app.application, user): + output = self.app.get('/fork/pingou/test') + self.assertEqual(output.status_code, 200) + self.assertFalse('This repo is brand new!
' in output.data) + self.assertIn( + 'This repo is brand new!
' in output.data) + self.assertIn( + 'This repo is brand new!
' in output.data) @@ -1526,6 +1587,12 @@ class PagureFlaskRepotests(tests.Modeltests): '