added§

git-svn-id: https://linkchecker.svn.sourceforge.net/svnroot/linkchecker/trunk/linkchecker@3298 e7d03fd6-7b0d-0410-9947-9c21f3af8025

linkcheck/tests/test_decorators.py

@@ -0,0 +1,56 @@
+# -*- coding: iso-8859-1 -*-
+# Copyright (C) 2005-2006 Bastian Kleineidam
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+"""
+Test decorators.
+"""
+
+import unittest
+import time
+import cStringIO as StringIO
+import linkcheck.decorators
+
+
+class TestDecorators (unittest.TestCase):
+    """
+    Test decorators.
+    """
+
+    def test_timeit (self):
+        @linkcheck.decorators.timed()
+	def f ():
+	    return 42
+	self.assertEquals(f(), 42)
+
+    def test_timeit2 (self):
+        log = StringIO.StringIO()
+        @linkcheck.decorators.timed(log=log, limit=0)
+	def f ():
+	    time.sleep(1)
+	    return 42
+	self.assertEquals(f(), 42)
+	self.assert_(log.getvalue())
+
+
+def test_suite ():
+    """
+    Build and return a TestSuite.
+    """
+    return unittest.makeSuite(TestDecorators)
+
+
+if __name__ == '__main__':
+    unittest.main()