edan_f0/tests/functional/customer/notification_tests.py

from django.contrib.auth.models import User
from django_dynamic_fixture import G

from oscar_testsupport.testcases import WebTestCase
from oscar.apps.customer.notifications import services


class TestAUserWithUnreadNotifications(WebTestCase):

    def setUp(self):
        self.user = G(User)
        services.notify_user(self.user, "Test message")

    def test_can_see_them_in_page_header(self):
        homepage = self.app.get('/', user=self.user)
        self.assertTrue('num_unread_notifications' in homepage.context)
        self.assertEqual(1, homepage.context['num_unread_notifications'])