Merge pull request #13 from cuenca-mx/asyncio

Asyncio

Author: Ricardo

mongoengine_plus/aio/async_document.py

@@ -6,7 +6,7 @@
 
 class AsyncDocument(Document):
     meta = dict(
-        allow_inheritance=True,
+        abstract=True,
         queryset_class=AsyncQuerySet,
     )
 

mongoengine_plus/aio/async_query_set.py

@@ -4,6 +4,9 @@
 
 
 class AsyncQuerySet(QuerySet):
+    async def async_first(self):
+        return await create_awaitable(self.first)
+
     async def async_get(self, *q_objs, **query):
         return await create_awaitable(self.get, *q_objs, **query)
 
@@ -12,3 +15,6 @@ async def async_count(self, with_limit_and_skip=False):
 
     async def async_to_list(self):
         return await create_awaitable(list, self)
+
+    async def async_update(self, *u_objs, **query):
+        return await create_awaitable(self.update, *u_objs, **query)

mongoengine_plus/version.py

@@ -1 +1 @@
-__version__ = '0.0.2'
+__version__ = '0.0.3'

tests/aio/test_async_query_set.py

@@ -25,3 +25,18 @@ async def test_to_list(cities):
         Q(state='Chiapas') | Q(state='Tabasco')
     ).async_to_list()
     assert len(filtered) == 3
+
+
+@pytest.mark.asyncio
+async def test_first(cities):
+    first_city = await City.objects(state='Tabasco').async_first()
+    assert first_city.state == 'Tabasco'
+
+
+@pytest.mark.asyncio
+async def test_update(cities):
+    await City.objects(name='San Cristobal').async_update(
+        set__name='San Cristobal de las Casas'
+    )
+    sancris = await City.objects.async_get(name__contains='San Cristobal')
+    assert sancris.name == 'San Cristobal de las Casas'