Adding a first test

scotwk · scotwk · commit 06667de63c7d · 2015-05-13T14:01:57.000-04:00
diff --git a/elevennote/note/models.py b/elevennote/note/models.py
@@ -14,4 +14,5 @@ class Note(models.Model):
     pub_date = models.DateTimeField('date published')
 
     def was_published_recently(self):
-        return self.pub_date >= timezone.now() - timedelta(days=1)
+        now = timezone.now()
+        return now - timedelta(days=1) <= self.pub_date <= now
diff --git a/elevennote/note/tests.py b/elevennote/note/tests.py
@@ -1,3 +1,18 @@
+import datetime
+
+from django.utils import timezone
 from django.test import TestCase
 
+from .models import Note
+
 # Create your tests here.
+class NoteMethodTests(TestCase):
+
+    def test_was_published_recently(self):
+        """
+        was_published_recently() should return False for notes whose
+        pub_date is in the future.
+        """
+        time = timezone.now() + datetime.timedelta(days=30)
+        future_note = Note(pub_date=time)
+        self.assertEqual(future_note.was_published_recently(), False)
