diff --git a/tests/test_pagure_flask_ui_repo.py b/tests/test_pagure_flask_ui_repo.py
index 0adf303..fda50dd 100644
--- a/tests/test_pagure_flask_ui_repo.py
+++ b/tests/test_pagure_flask_ui_repo.py
@@ -1419,6 +1419,13 @@ class PagureFlaskRepotests(tests.Modeltests):
''
'@@ -0,0 +1,3 @@' in output.data)
+ #View the commit when branch name is provided
+ output = self.app.get('/test/c/%s?branch=master' % commit.oid.hex)
+ self.assertEqual(output.status_code, 200)
+ self.assertTrue(
+ ''
+ in output.data)
+
# Add a fork of a fork
item = pagure.lib.model.Project(
user_id=1, # pingou
@@ -1463,6 +1470,14 @@ class PagureFlaskRepotests(tests.Modeltests):
self.assertEqual(output.status_code, 404)
self.assertIn('Project not found
', output.data)
+ #View the commit of the fork when branch name is provided
+ output = self.app.get('/fork/pingou/test3/c/%s?branch=master' % commit.oid.hex)
+ self.assertEqual(output.status_code, 200)
+ self.assertTrue(
+ ''
+ in output.data)
+
+
def test_view_commit_patch(self):
""" Test the view_commit_patch endpoint. """