Test for aggregate_rows() & get rid of limit(1) in get() (#76)

Test for aggregate_rows() & get rid of limit(1) in get()

peewee_async.py

@@ -168,7 +168,7 @@ async def my_async_func():
             query = query.where(*conditions)
 
         try:
-            result = yield from self.execute(query.limit(1))
+            result = yield from self.execute(query)
             return list(result)[0]
         except IndexError:
             raise model.DoesNotExist
@@ -487,7 +487,7 @@ def get_object(source, *args):
         model = source
 
     # Return first object from query
-    for obj in (yield from select(query.where(*args).limit(1))):
+    for obj in (yield from select(query.where(*args))):
         return obj
 
     # No objects found

tests/__init__.py

@@ -743,6 +743,32 @@ def test(objects):
 
         self.run_with_managers(test)
 
+    def test_aggregate_rows(self):
+        @asyncio.coroutine
+        def test(objects):
+            alpha = yield from objects.create(TestModelAlpha,
+                                              text='Alpha')
+
+            beta_1 = yield from objects.create(TestModelBeta,
+                                               alpha=alpha,
+                                               text='Beta 1')
+            beta_2 = yield from objects.create(TestModelBeta,
+                                               alpha=alpha,
+                                               text='Beta 2')
+
+            result = yield from objects.get((
+                TestModelAlpha
+                .select(TestModelAlpha, TestModelBeta)
+                .join(TestModelBeta)
+                .order_by(TestModelAlpha.text, TestModelBeta.text)
+                .aggregate_rows()))
+
+            self.assertEqual(result, alpha)
+
+            self.assertEqual(result.betas, [beta_1, beta_2])
+
+        self.run_with_managers(test)
+
 
 #####################
 # Python 3.5+ tests #