From e213e7180f4f4940df0116a33587cb5251327d60 Mon Sep 17 00:00:00 2001
From: Pierre-Yves Chibon <pingou@pingoured.fr>
Date: Apr 22 2015 07:58:16 +0000
Subject: Add unit-tests for the add_user endpoint behavior with admin_session_timedout


---

diff --git a/tests/test_progit_flask_ui_repo.py b/tests/test_progit_flask_ui_repo.py
index 0433c54..1da7ec3 100644
--- a/tests/test_progit_flask_ui_repo.py
+++ b/tests/test_progit_flask_ui_repo.py
@@ -51,8 +51,10 @@ class PagureFlaskRepotests(tests.Modeltests):
             tests.HERE, 'docs')
         self.app = pagure.APP.test_client()
 
-    def test_add_user(self):
+    @patch('pagure.ui.repo.admin_session_timedout')
+    def test_add_user(self, ast):
         """ Test the add_user endpoint. """
+        ast.return_value = False
 
         output = self.app.get('/foo/adduser')
         self.assertEqual(output.status_code, 302)
@@ -67,6 +69,11 @@ class PagureFlaskRepotests(tests.Modeltests):
             output = self.app.get('/test/adduser')
             self.assertEqual(output.status_code, 403)
 
+            ast.return_value = True
+            output = self.app.get('/test/adduser')
+            self.assertEqual(output.status_code, 302)
+            ast.return_value = False
+
         user.username = 'pingou'
         with tests.user_set(pagure.APP, user):
             output = self.app.get('/test/adduser')