diff --git a/tests/test_progit_flask_ui_repo.py b/tests/test_progit_flask_ui_repo.py index 968ca94..18f9bfd 100644 --- a/tests/test_progit_flask_ui_repo.py +++ b/tests/test_progit_flask_ui_repo.py @@ -383,6 +383,86 @@ class ProgitFlaskRepotests(tests.Modeltests): self.assertEqual( output.data.count(''), 10) + def test_view_repo_branch(self): + """ Test the view_repo_branch endpoint. """ + + output = self.app.get('/foo/branch/master') + # No project registered in the DB + self.assertEqual(output.status_code, 404) + + tests.create_projects(self.session) + + output = self.app.get('/test/branch/master') + # No git repo associated + self.assertEqual(output.status_code, 404) + + tests.create_projects_git(tests.HERE) + + output = self.app.get('/test/branch/master') + self.assertEqual(output.status_code, 404) + + # Add some content to the git repo + tests.add_content_git_repo(os.path.join(tests.HERE, 'test.git')) + tests.add_readme_git_repo(os.path.join(tests.HERE, 'test.git')) + + output = self.app.get('/test/branch/master') + self.assertEqual(output.status_code, 200) + self.assertFalse('

This repo is brand new!

' in output.data) + self.assertFalse('Forked from' in output.data) + self.assertTrue('

test project #1

' in output.data) + self.assertEqual( + output.data.count(''), 3) + + # Turn that repo into a fork + repo = progit.lib.get_project(self.session, 'test') + repo.parent_id = 2 + self.session.add(repo) + self.session.commit() + + # View the repo in the UI + output = self.app.get('/test/branch/master') + self.assertEqual(output.status_code, 404) + + # Add some content to the git repo + tests.add_content_git_repo( + os.path.join(tests.HERE, 'forks', 'pingou', 'test.git')) + tests.add_readme_git_repo( + os.path.join(tests.HERE, 'forks', 'pingou', 'test.git')) + + output = self.app.get('/fork/pingou/test/branch/master') + self.assertEqual(output.status_code, 200) + self.assertFalse('

This repo is brand new!

' in output.data) + self.assertTrue('

test project #1

' in output.data) + self.assertTrue('Forked from' in output.data) + self.assertEqual( + output.data.count(''), 3) + + # Add a fork of a fork + item = progit.lib.model.Project( + user_id=1, # pingou + name='test3', + description='test project #3', + parent_id=1, + ) + self.session.add(item) + self.session.commit() + + tests.add_content_git_repo( + os.path.join(tests.HERE, 'forks', 'pingou', 'test3.git')) + tests.add_readme_git_repo( + os.path.join(tests.HERE, 'forks', 'pingou', 'test3.git')) + tests.add_commit_git_repo( + os.path.join(tests.HERE, 'forks', 'pingou', 'test3.git'), + ncommits=10) + + output = self.app.get('/fork/pingou/test3/branch/master') + self.assertEqual(output.status_code, 200) + self.assertFalse('

This repo is brand new!

' in output.data) + self.assertTrue('

test project #3

' in output.data) + self.assertTrue('Forked from' in output.data) + self.assertEqual( + output.data.count(''), 10) + if __name__ == '__main__': SUITE = unittest.TestLoader().loadTestsFromTestCase(ProgitFlaskRepotests)