edan_f0/tests/functional/customer/notification_tests.py

from django_dynamic_fixture import G

from oscar.test.testcases import WebTestCase
from oscar.core.compat import get_user_model
from oscar.apps.customer.notifications import services


User = get_user_model()

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'])