Home | History | Annotate | Download | only in test 
      1 #!/usr/bin/env python
      2 #
      3 # Copyright 2009, Google Inc.
      4 # All rights reserved.
      5 #
      6 # Redistribution and use in source and binary forms, with or without
      7 # modification, are permitted provided that the following conditions are
      8 # met:
      9 #
     10 #     * Redistributions of source code must retain the above copyright
     11 # notice, this list of conditions and the following disclaimer.
     12 #     * Redistributions in binary form must reproduce the above
     13 # copyright notice, this list of conditions and the following disclaimer
     14 # in the documentation and/or other materials provided with the
     15 # distribution.
     16 #     * Neither the name of Google Inc. nor the names of its
     17 # contributors may be used to endorse or promote products derived from
     18 # this software without specific prior written permission.
     19 #
     20 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
     21 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
     22 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
     23 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
     24 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
     25 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
     26 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     27 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     28 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     29 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
     30 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     31 
     32 
     33 """Tests for util module."""
     34 
     35 
     36 import unittest
     37 
     38 import config  # This must be imported before mod_pywebsocket.
     39 from mod_pywebsocket import util
     40 
     41 
     42 class UtilTest(unittest.TestCase):
     43     def test_get_stack_trace(self):
     44         self.assertEqual('None\n', util.get_stack_trace())
     45         try:
     46             a = 1 / 0  # Intentionally raise exception.
     47         except Exception:
     48             trace = util.get_stack_trace()
     49             self.failUnless(trace.startswith('Traceback'))
     50             self.failUnless(trace.find('ZeroDivisionError') != -1)
     51 
     52     def test_prepend_message_to_exception(self):
     53         exc = Exception('World')
     54         self.assertEqual('World', str(exc))
     55         util.prepend_message_to_exception('Hello ', exc)
     56         self.assertEqual('Hello World', str(exc))
     57 
     58 
     59 if __name__ == '__main__':
     60     unittest.main()
     61 
     62 
     63 # vi:sts=4 sw=4 et
     64