Add exclude() to MockSet

django_mock_queries/query.py

@@ -73,6 +73,13 @@ def filter(*args, **attrs):
 
     mock_set.filter = MagicMock(side_effect=filter)
 
+    def exclude(*args, **attrs):
+        excluded = filter(*args, **attrs)
+        results = [item for item in items if item not in excluded]
+        return MockSet(*results, cls=cls)
+
+    mock_set.exclude = MagicMock(side_effect=exclude)
+
     def clear():
         del items[:]
 

tests/test_query.py

@@ -68,6 +68,18 @@ def test_query_filters_items_by_q_object_and(self):
         assert item_2 in results
         assert item_3 not in results
 
+    def test_query_exclude(self):
+        item_1 = MockModel(foo=1, bar='a')
+        item_2 = MockModel(foo=1, bar='b')
+        item_3 = MockModel(foo=2, bar='b')
+
+        self.mock_set.add(item_1, item_2, item_3)
+        results = list(self.mock_set.exclude(foo=1, bar='b'))
+
+        assert item_1 in results, results
+        assert item_2 not in results, results
+        assert item_3 in results, results
+
     def test_query_clears_all_items_from_set(self):
         self.mock_set.add(1, 2, 3)
         self.mock_set.clear()