diff --git a/tests/test_progit_flask_ui_repo.py b/tests/test_progit_flask_ui_repo.py index bf4f150..db8fd6f 100644 --- a/tests/test_progit_flask_ui_repo.py +++ b/tests/test_progit_flask_ui_repo.py @@ -102,6 +102,46 @@ class ProgitFlaskRepotests(tests.Modeltests): self.assertTrue( '
' in output.data) + def test_remove_user(self): + """ Test the remove_user endpoint. """ + + output = self.app.post('/foo/dropuser/1') + self.assertEqual(output.status_code, 302) + + user = tests.FakeUser() + with tests.user_set(progit.APP, user): + output = self.app.post('/foo/dropuser/1') + self.assertEqual(output.status_code, 404) + + tests.create_projects(self.session) + + output = self.app.post('/test/dropuser/1') + self.assertEqual(output.status_code, 403) + + user.username = 'pingou' + with tests.user_set(progit.APP, user): + output = self.app.post('/test/dropuser/1', follow_redirects=True) + self.assertEqual(output.status_code, 200) + self.assertTrue( + '