From 1afcb52d73271bbbd78f885451aa1b0e78c09871 Mon Sep 17 00:00:00 2001
From: matt <matt@xcolour.net>
Date: Mon, 28 Jan 2013 11:32:18 -0500
Subject: Import StringIO so it can be used.

---
 tests/test_profilemiddleware.py | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 tests/test_profilemiddleware.py

(limited to 'tests/test_profilemiddleware.py')

diff --git a/tests/test_profilemiddleware.py b/tests/test_profilemiddleware.py
new file mode 100644
index 0000000..4c189f8
--- /dev/null
+++ b/tests/test_profilemiddleware.py
@@ -0,0 +1,29 @@
+from paste.fixture import *
+try:
+    from paste.debug.profile import *
+    disable = False
+except ImportError:
+    disable = True
+
+if not disable:
+    def simple_app(environ, start_response):
+        start_response('200 OK', [('content-type', 'text/html')])
+        return ['all ok']
+
+    def long_func():
+        for i in range(1000):
+            pass
+        return 'test'
+
+    def test_profile():
+        app = TestApp(ProfileMiddleware(simple_app, {}))
+        res = app.get('/')
+        # The original app:
+        res.mustcontain('all ok')
+        # The profile information:
+        res.mustcontain('<pre')
+
+    def test_decorator():
+        value = profile_decorator()(long_func)()
+        assert value == 'test'
+
-- 
cgit v1.2.1