From: Jason Kirtland <jek@discorporate.us>
Date: Mon, 23 Jul 2007 16:04:37 +0000 (+0000)
Subject: Rearrange profiling a bit
X-Git-Tag: rel_0_4_6~44
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=24c98dc697c13270198c47d27baae53e0a2c9fac;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

Rearrange profiling a bit
---

diff --git a/test/perf/ormsession.py b/test/perf/ormsession.py
index d0f4e6ecbf..a9d310ef68 100644
--- a/test/perf/ormsession.py
+++ b/test/perf/ormsession.py
@@ -199,22 +199,27 @@ def create_purchase():
     session.commit()
     session.expire(customer)
 
-
 def setup_db():
     metadata.drop_all()
     metadata.create_all()
 def cleanup_db():
     metadata.drop_all()
 
+@profiled('default')
+def default():
+    run_queries()
+    create_purchase()
+
 @profiled('all')
 def main():
     metadata.bind = testbase.db
-    define_tables()
-    setup_mappers()
-    setup_db()
-    insert_data()
-    run_queries()
-    create_purchase()
-    cleanup_db()
+    try:
+        define_tables()
+        setup_mappers()
+        setup_db()
+        insert_data()
+        default()
+    finally:
+        cleanup_db()
 
 main()