From 2c15daee6e3c13692260fa5c741074ccf64f5bfa Mon Sep 17 00:00:00 2001
From: Pierre-Yves Chibon <pingou@pingoured.fr>
Date: Mar 05 2015 12:27:59 +0000
Subject: Adjust the unit-tests for remove_user now that it is CSRF protected


---

diff --git a/tests/test_progit_flask_ui_repo.py b/tests/test_progit_flask_ui_repo.py
index b6db3c6..f5152b5 100644
--- a/tests/test_progit_flask_ui_repo.py
+++ b/tests/test_progit_flask_ui_repo.py
@@ -120,7 +120,15 @@ class ProgitFlaskRepotests(tests.Modeltests):
 
         user.username = 'pingou'
         with tests.user_set(progit.APP, user):
-            output = self.app.post('/test/dropuser/1', follow_redirects=True)
+            output = self.app.post('/test/settings')
+
+            csrf_token = output.data.split(
+                'name="csrf_token" type="hidden" value="')[1].split('">')[0]
+
+            data = {'csrf_token': csrf_token}
+
+            output = self.app.post(
+                '/test/dropuser/2', data=data, follow_redirects=True)
             self.assertEqual(output.status_code, 200)
             self.assertTrue('<header class="repo">' in output.data)
             self.assertTrue('<h2>Settings</h2>' in output.data)
@@ -144,6 +152,17 @@ class ProgitFlaskRepotests(tests.Modeltests):
             self.assertTrue('<header class="repo">' in output.data)
             self.assertTrue('<h2>Settings</h2>' in output.data)
             self.assertTrue(
+                '<ul id="flashes">\n                </ul>' in output.data)
+            self.assertFalse(
+                '<li class="message">User removed</li>' in output.data)
+
+            data = {'csrf_token': csrf_token}
+            output = self.app.post(
+                '/test/dropuser/2', data=data, follow_redirects=True)
+            self.assertEqual(output.status_code, 200)
+            self.assertTrue('<header class="repo">' in output.data)
+            self.assertTrue('<h2>Settings</h2>' in output.data)
+            self.assertTrue(
                 '<li class="message">User removed</li>' in output.data)
 
     def test_update_description(self):