wizard/tests/util_test.py

   1 import traceback
   2
   3 from wizard.util import *
   4
   5 class MyError(Exception):
   6     def __str__(self):
   7         return """
   8
   9 ERROR: Foo
  10 """
  11
  12 def test_get_dir_uid():
  13     if os.getuid(): return # only run if on a scripts server. This is crude
  14     assert get_dir_uid("/mit/ezyang/web_scripts/test-wiki") == 537864399
  15
  16 def test_get_dir_uid_locker():
  17     if os.getuid(): return
  18     assert get_dir_uid("/mit/apo/web_scripts/") == 536956980
  19
  20 def test_get_exception_name():
  21     try:
  22         raise NotImplementedError
  23     except NotImplementedError:
  24         assert get_exception_name(traceback.format_exc()) == "NotImplementedError"
  25
  26 def test_get_exception_name_withstr():
  27     try:
  28         raise MyError
  29     except MyError:
  30         assert get_exception_name(traceback.format_exc()) == "MyError"
  31