Update test.

Scifabric · Oct 31, 2017 · e1e53f9 · e1e53f9
1 parent e3cbdda
commit e1e53f9
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/test/test_newsletter.py b/test/test_newsletter.py
@@ -214,14 +214,14 @@ class TestNewsletterViewFunctions(web.Helper):
     @patch('pybossa.view.account.newsletter', autospec=True)
     def test_new_user_gets_newsletter(self, newsletter):
         """Test NEWSLETTER new user works."""
-        newsletter.is_initialized.return_value = True
-        newsletter.ask_user_to_subscribe.return_value = True
-        res = self.register()
-        dom = BeautifulSoup(res.data)
-        err_msg = "There should be a newsletter page."
-        assert dom.find(id='newsletter') is not None, err_msg
-        assert dom.find(id='signmeup') is not None, err_msg
-        assert dom.find(id='notinterested') is not None, err_msg
+        with patch.dict(self.flask_app.config, {'MAILCHIMP_API_KEY': 'key'}):
+            newsletter.ask_user_to_subscribe.return_value = True
+            res = self.register()
+            dom = BeautifulSoup(res.data)
+            err_msg = "There should be a newsletter page."
+            assert dom.find(id='newsletter') is not None, err_msg
+            assert dom.find(id='signmeup') is not None, err_msg
+            assert dom.find(id='notinterested') is not None, err_msg
 
 
     @with_context