Fix tests

Author: ei-grad

elasticutils/contrib/django/tests/__init__.py

@@ -13,10 +13,16 @@ def reset_model_cache():
     del _model_cache[0:]
 
 
+class FakePK():
+    name = 'id'
+
+
 class Meta(object):
     def __init__(self, db_table):
         self.db_table = db_table
 
+    pk = FakePK()
+
 
 class SearchQuerySet(object):
     # Yes. This is kind of crazy, but ... whatever.
@@ -28,8 +34,8 @@ def get(self, pk):
         pk = int(pk)
         return [m for m in _model_cache if m.id == pk][0]
 
-    def filter(self, id__in=None):
-        self.steps.append(('filter', id__in))
+    def filter(self, pk__in=None):
+        self.steps.append(('filter', pk__in))
         return self
 
     def order_by(self, *fields):
@@ -89,6 +95,10 @@ def __init__(self, **kw):
             setattr(self, key, kw[key])
         _model_cache.append(self)
 
+    @property
+    def pk(self):
+        return self.id
+
 
 class FakeDjangoMappingType(MappingType, Indexable):
     @classmethod