import traceback

from wizard.util import *

class MyError(Exception):
    def __str__(self):
        return """

ERROR: Foo
"""

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

def test_get_dir_uid_locker():
    if os.getuid(): return
    assert get_dir_uid("/mit/apo/web_scripts/") == 536956980

def test_get_exception_name():
    try:
        raise NotImplementedError
    except NotImplementedError:
        assert get_exception_name(traceback.format_exc()) == "NotImplementedError"

def test_get_exception_name_withstr():
    try:
        raise MyError
    except MyError:
        assert get_exception_name(traceback.format_exc()) == "MyError"