async_update

Author: Ricardo Sánchez

mongoengine_plus/aio/async_query_set.py

@@ -15,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.3.dev1'
+__version__ = '0.0.3.dev2'

tests/aio/test_async_query_set.py

@@ -31,3 +31,12 @@ async def test_to_list(cities):
 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'