Make Response pickleable

Fixes #558 for 2.x

Author: honzakral

elasticsearch_dsl/result.py

@@ -28,6 +28,13 @@ def __repr__(self):
     def __len__(self):
         return len(self.hits)
 
+    def __getstate__(self):
+        return (self._d_, self._callbacks)
+
+    def __setstate__(self, state):
+        super(AttrDict, self).__setattr__('_d_', state[0])
+        super(AttrDict, self).__setattr__('_callbacks', state[1])
+
     def success(self):
         return self._shards.total == self._shards.successful and not self.timed_out
 

test_elasticsearch_dsl/test_result.py

@@ -1,7 +1,17 @@
+import pickle
+
 from pytest import raises
 
 from elasticsearch_dsl import result
 
+def test_response_is_pickleable(dummy_response):
+    res = result.Response(dummy_response)
+    res.hits
+    r = pickle.loads(pickle.dumps(res))
+
+    assert r == res
+    assert r.hits == res.hits
+
 def test_attribute_error_in_hits_is_not_hidden(dummy_response):
     def f(hit):
         raise AttributeError()