fix tests, mock and patch with flash messages

Scifabric · Jan 5, 2017 · 5e08b9f · 5e08b9f
1 parent fca795c
commit 5e08b9f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/test/test_util.py b/test/test_util.py
@@ -214,7 +214,8 @@ def test_handle_content_type_html_error(self, mockjsonify,
     @patch('pybossa.util.request')
     @patch('pybossa.util.render_template')
     @patch('pybossa.util.jsonify')
-    def test_redirect_content_type_json(self, mockjsonify, mockrender, mockrequest):
+    @patch('pybossa.util.last_flashed_message')
+    def test_redirect_content_type_json(self, mocklast, mockjsonify, mockrender, mockrequest):
         mockrequest.headers.__getitem__.return_value = 'application/json'
         mockjsonify.side_effect = myjsonify
         res = util.redirect_content_type('http://next.uri')
@@ -227,7 +228,8 @@ def test_redirect_content_type_json(self, mockjsonify, mockrender, mockrequest):
     @patch('pybossa.util.request')
     @patch('pybossa.util.render_template')
     @patch('pybossa.util.jsonify')
-    def test_redirect_content_type_json_message(self, mockjsonify, mockrender, mockrequest):
+    @patch('pybossa.util.last_flashed_message')
+    def test_redirect_content_type_json_message(self, mocklast, mockjsonify, mockrender, mockrequest):
         mockrequest.headers.__getitem__.return_value = 'application/json'
         mockjsonify.side_effect = myjsonify
         res = util.redirect_content_type('http://next.uri', message='hallo123')