From 0ebfa3f5e54a30c2d451495cd99090d010fec42f Mon Sep 17 00:00:00 2001
From: Pierre-Yves Chibon <pingou@pingoured.fr>
Date: Mar 02 2015 14:07:41 +0000
Subject: Expand the unit-tests for the front page when the user is logged in


---

diff --git a/tests/test_progit_flask_ui_app.py b/tests/test_progit_flask_ui_app.py
index 9cdb830..45ce9cb 100644
--- a/tests/test_progit_flask_ui_app.py
+++ b/tests/test_progit_flask_ui_app.py
@@ -45,6 +45,23 @@ class ProgitFlaskApptests(tests.Modeltests):
         self.assertEqual(output.status_code, 200)
         self.assertTrue('<h2>All Projects (2)</h2>' in output.data)
 
+        output = self.app.get('/?page=abc')
+        self.assertEqual(output.status_code, 200)
+        self.assertTrue('<h2>All Projects (2)</h2>' in output.data)
+
+        user = tests.FakeUser()
+        with tests.user_set(progit.APP, user):
+            output = self.app.get('/?repopage=abc&forkpage=def')
+            self.assertTrue(
+                '<section class="project_list" id="repos">' in output.data)
+            self.assertTrue('<h3>My Forks (0)</h3>' in output.data)
+            self.assertTrue(
+                '<section class="project_list" id="myforks">' in output.data)
+            self.assertTrue('<h3>My Projects (0)</h3>' in output.data)
+            self.assertTrue(
+                '<section class="project_list" id="myrepos">' in output.data)
+            self.assertTrue('<h3>All Projects (2)</h3>' in output.data)
+
 
 if __name__ == '__main__':
     SUITE = unittest.TestLoader().loadTestsFromTestCase(ProgitFlaskApptests)