More bugfixes from running live.

[wizard.git] / wizard / tests / util_test.py
diff --git a/wizard/tests/util_test.py b/wizard/tests/util_test.py

index c34abba73f4b7bc8ea3ac178dd5c616226b6a28a..e4ebe91977d846ec31e0f5599bcc1fa47aaafca9 100644 (file)
--- a/wizard/tests/util_test.py
+++ b/wizard/tests/util_test.py
@@ -9,6 +9,20 @@ class MyError(Exception):
  ERROR: Foo
  """
  
+class MyErrorWithHTML(Exception):
+    def __str__(self):
+        return """
+
+ERROR: Bar
+
+<html>
+    <title>No good!</title>
+</html>
+"""
+
+def test_dictmap():
+    assert dictmap(lambda x: x + 1, {'a': 0, 'b': 1}) == {'a': 1, 'b': 2}
+
  def test_get_dir_uid():
      if os.getuid(): return # only run if on a scripts server. This is crude
      assert get_dir_uid("/mit/ezyang/web_scripts/test-wiki") == 537864399
@@ -29,3 +43,9 @@ def test_get_exception_name_withstr():
      except MyError:
          assert get_exception_name(traceback.format_exc()) == "MyError"
  
+def test_get_exception_name_withhtml():
+    try:
+        raise MyErrorWithHTML
+    except MyErrorWithHTML:
+        assert get_exception_name(traceback.format_exc()) == "MyErrorWithHTML"
+