Merge pull request #1681 from akatsoulas/basket-tests

Fix basket tests.
mozilla · Apr 3, 2017 · bfa6c31 · bfa6c31
2 parents 737a79e + 1cf11be
commit bfa6c31
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mozillians/users/tests/test_tasks.py b/mozillians/users/tests/test_tasks.py
@@ -1,5 +1,6 @@
 from datetime import datetime
 
+from django.conf import settings
 from django.contrib.auth.models import User
 from django.test.utils import override_settings
 
@@ -225,6 +226,7 @@ def test_subscribe_user_task_no_email_no_newsletters(self, subscribe_mock):
         subscribe_user_task(result=result, email=None)
         subscribe_mock.assert_called_with('result_email@example.com', ['mozilla-phone'],
                                           sync='N', trigger_welcome='N',
+                                          source_url=settings.SITE_URL,
                                           api_key='basket_api_key')
 
     @patch('mozillians.users.tasks.basket.subscribe')
@@ -253,6 +255,7 @@ def test_subscribe_user_task(self, subscribe_mock):
         subscribe_user_task(**kwargs)
         subscribe_mock.assert_called_with('foo@example.com', ['foobar'],
                                           sync='N', trigger_welcome='N',
+                                          source_url=settings.SITE_URL,
                                           api_key='basket_api_key')
 
     @patch('mozillians.users.tasks.basket.subscribe')
@@ -266,6 +269,7 @@ def test_subscribe_user_task_no_result(self, subscribe_mock):
         subscribe_user_task(**kwargs)
         subscribe_mock.assert_called_with('foo@example.com', ['mozilla-phone'],
                                           sync='N', trigger_welcome='N',
+                                          source_url=settings.SITE_URL,
                                           api_key='basket_api_key')
 
     @patch('mozillians.users.tasks.subscribe_user_task.retry')