Commit a02a149f by Őry Máté

dashboard: fix test_periodic_task again

parent b20ebb00
...@@ -10,9 +10,13 @@ from ..tasks import local_periodic_tasks ...@@ -10,9 +10,13 @@ from ..tasks import local_periodic_tasks
@patch.object(local_periodic_tasks, 'send_mail') @patch.object(local_periodic_tasks, 'send_mail')
@patch.object(Notification, 'objects') @patch.object(Notification, 'objects')
class EmailNotificationTestCase(unittest.TestCase): class EmailNotificationTestCase(unittest.TestCase):
nextpk = 0
def get_fake_notification(self, user=None, **kwargs): def get_fake_notification(self, user=None, **kwargs):
self.nextpk += 1
if user is None: if user is None:
user = MagicMock(spec=User) user = MagicMock(spec=User, pk=self.nextpk)
user.profile.__unicode__.return_value = "user" user.profile.__unicode__.return_value = "user"
user.email = "mail" user.email = "mail"
user.profile.email_notifications = True user.profile.email_notifications = True
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment