We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8ff87f1 commit 216b0adCopy full SHA for 216b0ad
django_mock_queries/query.py
@@ -1,6 +1,7 @@
1
import datetime
2
import random
3
from collections import OrderedDict, namedtuple
4
+from six import with_metaclass
5
try:
6
from unittest.mock import Mock, MagicMock, PropertyMock
7
except ImportError:
@@ -13,7 +14,14 @@
13
14
)
15
16
-class MockSet(MagicMock):
17
+class MockSetMeta(type):
18
+ def __call__(cls, *initial_items, **kwargs):
19
+ obj = super(MockSetMeta, cls).__call__(**kwargs)
20
+ obj.add(*initial_items)
21
+ return obj
22
+
23
24
+class MockSet(with_metaclass(MockSetMeta, MagicMock)):
25
EVENT_ADDED = 'added'
26
EVENT_UPDATED = 'updated'
27
EVENT_SAVED = 'saved'
0 commit comments