Merge pull request #11 from cfe-lab/first-last

Add first() and last() to MockSet.

django_mock_queries/query.py

@@ -156,6 +156,17 @@ def earliest(field):
 
     mock_set.earliest = MagicMock(side_effect=earliest)
 
+    def first():
+        for item in items:
+            return item
+
+    mock_set.first = MagicMock(side_effect=first)
+
+    def last():
+        return items and items[-1] or None
+
+    mock_set.last = MagicMock(side_effect=last)
+
     def __iter__():
         return iter([project(i) for i, v in enumerate(items)])
 

tests/test_query.py

@@ -199,6 +199,36 @@ def test_query_latest_returns_the_last_element_from_ordered_set(self):
 
         assert latest == item_3
 
+    def test_query_first_none(self):
+        first = self.mock_set.first()
+
+        assert first is None, first
+
+    def test_query_first(self):
+        item_1 = MockModel(foo=1)
+        item_2 = MockModel(foo=2)
+        item_3 = MockModel(foo=3)
+
+        self.mock_set.add(item_3, item_1, item_2)
+        first = self.mock_set.first()
+
+        assert first == item_3, first
+
+    def test_query_last_none(self):
+        last = self.mock_set.last()
+
+        assert last is None, last
+
+    def test_query_last(self):
+        item_1 = MockModel(foo=1)
+        item_2 = MockModel(foo=2)
+        item_3 = MockModel(foo=3)
+
+        self.mock_set.add(item_3, item_1, item_2)
+        last = self.mock_set.last()
+
+        assert last == item_2, last
+
     def test_query_latest_raises_error_exist_when_empty_set(self):
         self.mock_set.clear()
         self.assertRaises(Exception, self.mock_set.latest, 'foo')