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('

All Projects (2)

' in output.data) + output = self.app.get('/?page=abc') + self.assertEqual(output.status_code, 200) + self.assertTrue('

All Projects (2)

' in output.data) + + user = tests.FakeUser() + with tests.user_set(progit.APP, user): + output = self.app.get('/?repopage=abc&forkpage=def') + self.assertTrue( + '
' in output.data) + self.assertTrue('

My Forks (0)

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

My Projects (0)

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

All Projects (2)

' in output.data) + if __name__ == '__main__': SUITE = unittest.TestLoader().loadTestsFromTestCase(ProgitFlaskApptests)