Fixed asynchronous search query to_dict method

elasticmagic/ext/asyncio/search.py

@@ -5,6 +5,10 @@ class AsyncSearchQuery(BaseSearchQuery):
     """Asynchronous version of the :class:`.SearchQuery`
     """
 
+    async def to_dict(self, compiler=None):
+        compiler = compiler or await self.get_compiler()
+        return compiler.compiled_query(self).body
+
     async def get_compiler(self):
         return await self._index_or_cluster.get_compiler()
 

elasticmagic/search.py

@@ -713,13 +713,6 @@ def get_compiler_context(self):
 
     get_context = get_compiler_context
 
-    def to_dict(self, compiler=None):
-        """Compiles the query and returns python dictionary that can be
-        serialized to json.
-        """
-        compiler = compiler or self.get_compiler()
-        return compiler.compiled_query(self).body
-
     def slice(self, offset, limit):
         """Applies offset and limit to the query."""
         sliced_query, _ = self._prepare_slice(slice(offset, limit))
@@ -793,6 +786,13 @@ def get_compiler(self):
     def get_query_compiler(self):
         return self.get_compiler().compiled_query
 
+    def to_dict(self, compiler=None):
+        """Compiles the query and returns python dictionary that can be
+        serialized to json.
+        """
+        compiler = compiler or self.get_compiler()
+        return compiler.compiled_query(self).body
+
     def get_result(self):
         """Executes current query and returns processed :class:`SearchResult`
         object. Caches result so subsequent calls with the same search query

tests_integ/asyncio/test_search.py

@@ -3,6 +3,18 @@
 from .conftest import Car
 
 
+@pytest.mark.asyncio
+async def test_to_dict(es_index, cars):
+    sq = (
+        es_index.search_query()
+        .limit(1)
+    )
+
+    assert await sq.to_dict() == {
+        'size': 1
+    }
+
+
 @pytest.mark.asyncio
 async def test_get_result(es_index, cars):
     sq = (