Add unit tests for notification model

oscar/apps/customer/services.py

@@ -3,7 +3,7 @@ from django.db.models import get_model
 Notification = get_model('notifications', 'Notification')
 
 
-def notify(user, msg, category=None):
+def notify_user(user, msg, category=None):
     """
     Send a simple notification to a user
     """
@@ -12,9 +12,10 @@ def notify(user, msg, category=None):
         subject=msg,
         category=category)
 
-def send_message(user, subject, body, category):
-    Notification.objects.create(
-        recipient=user,
-        subject=subject,
-        body=body,
-        category=category)
+
+def notify_users(users, msg, category=None):
+    """
+    Send a simple notification to an iterable of users
+    """
+    for user in users:
+        notify_user(user, msg, category)

tests/unit/notification_tests.py

@@ -0,0 +1,32 @@
+from django.test import TestCase
+from django.contrib.auth.models import User
+from django_dynamic_fixture import G
+
+from oscar.apps.notifications.models import Notification
+
+
+class TestANewNotification(TestCase):
+
+    def setUp(self):
+        # Don't save models for speed
+        self.notification = Notification(
+            recipient=User(),
+            subject="Hello")
+
+    def test_is_in_a_users_inbox(self):
+        self.assertEqual(Notification.INBOX, self.notification.location)
+
+    def test_is_not_read(self):
+        self.assertFalse(self.notification.is_read)
+
+
+class TestANotification(TestCase):
+
+    def setUp(self):
+        self.notification = Notification.objects.create(
+            recipient=G(User),
+            subject="Hello")
+
+    def test_can_be_archived(self):
+        self.notification.archive()
+        self.assertEqual(Notification.ARCHIVE, self.notification.location)