Update test.

Scifabric · Oct 31, 2017 · e3cbdda · e3cbdda
1 parent a2c71c8
commit e3cbdda
Showing 1 changed file with 18 additions and 18 deletions.
diff --git a/test/test_newsletter.py b/test/test_newsletter.py
@@ -228,24 +228,24 @@ def test_new_user_gets_newsletter(self, newsletter):
     @patch('pybossa.view.account.newsletter', autospec=True)
     def test_new_user_gets_newsletter_only_once(self, newsletter):
         """Test NEWSLETTER user gets newsletter only once works."""
-        newsletter.is_initialized.return_value = True
-        newsletter.ask_user_to_subscribe.return_value = True
-        res = self.register()
-        dom = BeautifulSoup(res.data)
-        user = user_repo.get(1)
-        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
-        assert user.newsletter_prompted is True, err_msg
-
-        self.signout()
-        newsletter.ask_user_to_subscribe.return_value = False
-        res = self.signin()
-        dom = BeautifulSoup(res.data)
-        assert dom.find(id='newsletter') is None, err_msg
-        assert dom.find(id='signmeup') is None, err_msg
-        assert dom.find(id='notinterested') is 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)
+            user = user_repo.get(1)
+            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
+            assert user.newsletter_prompted is True, err_msg
+
+            self.signout()
+            newsletter.ask_user_to_subscribe.return_value = False
+            res = self.signin()
+            dom = BeautifulSoup(res.data)
+            assert dom.find(id='newsletter') is None, err_msg
+            assert dom.find(id='signmeup') is None, err_msg
+            assert dom.find(id='notinterested') is None, err_msg
 
     @with_context
     @patch('pybossa.view.account.newsletter', autospec=True)