Fix possibly vulnerable ldap code to use better kernel hooks.

[wizard.git] / wizard / tests / util_test.py

diff --git a/wizard/tests/util_test.py b/wizard/tests/util_test.py
index ad77c8838bc8671daf3120c702087d0acb1a2b3b..c34abba73f4b7bc8ea3ac178dd5c616226b6a28a 100644 (file)
--- a/wizard/tests/util_test.py
+++ b/wizard/tests/util_test.py
@@ -9,14 +9,13 @@ class MyError(Exception):
 ERROR: Foo
 """
 
-def test_get_dir_user():
-    assert get_dir_user("/mit/ezyang/web_scripts/test-wiki") == "ezyang"
+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_user_locker():
-    assert get_dir_user("/mit/apo/web_scripts/") == "apo"
-
-def test_get_dir_url():
-    assert get_dir_url("/mit/ezyang/web_scripts/foo") == "http://ezyang.scripts.mit.edu/foo"
+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: