Run queries within peewee exception_wrapper() context, so exceptions …

…raised are now same as for sync execution

peewee_async.py

@@ -236,7 +236,7 @@ def create_or_get(self, model, **kwargs):
         """
         try:
             return (yield from self.create(model, **kwargs)), True
-        except: # except peewee.IntegrityError:
+        except peewee.IntegrityError:
             query = []
             for field_name, value in kwargs.items():
                 field = getattr(model, field_name)
@@ -1394,4 +1394,5 @@ def _run_sql(db, operation, *args, **kwargs):
 def _execute_query_async(query):
     """Execute query and return cursor object.
     """
-    return (yield from _run_sql(query.database, *query.sql()))
+    with query.database.exception_wrapper():
+        return (yield from _run_sql(query.database, *query.sql()))

tests/__init__.py

@@ -401,7 +401,7 @@ def test_connect_close(self):
         @asyncio.coroutine
         def get_conn(objects):
             yield from objects.connect()
-            yield from asyncio.sleep(0.125, loop=self.loop)
+            # yield from asyncio.sleep(0.05, loop=self.loop)
             # NOTE: "private" member access
             return objects.database._async_conn