a little nicer wrt pep8

[sfa.git] / tests / testHierarchy.py

diff --git a/tests/testHierarchy.py b/tests/testHierarchy.py
old mode 100644 (file)
new mode 100755 (executable)
index 5262009..379d850
--- a/tests/testHierarchy.py
+++ b/tests/testHierarchy.py
@@ -1,7 +1,8 @@
 import unittest
 import os
-from hierarchy import *
-from config import *
+from sfa.util.faults import *
+from sfa.trust.hierarchy import *
+from sfa.util.config import *
 
 BASEDIR = "test_hierarchy"
 PURGE_BASEDIR = "rm -rf test_hierarchy"
@@ -21,26 +22,29 @@ class TestHierarchy(unittest.TestCase):
 
         self.assertEqual(h.auth_exists(name), False)
 
-        auth_info = h.get_auth_info(name, can_create=True)
-        self.assert_(auth_info)
+        self.assertRaises(MissingAuthority, h.get_auth_info, name)
+
+        h.create_auth(name, create_parents=True)
+        auth_info = h.get_auth_info(name)
+        self.assertTrue(auth_info)
 
         gid = auth_info.get_gid_object()
-        self.assert_(gid)
+        self.assertTrue(gid)
         self.assertEqual(gid.get_subject(), name)
 
         pubkey = auth_info.get_pkey_object()
-        self.assert_(gid)
+        self.assertTrue(gid)
 
         # try to get it again, make sure it's still there
-        auth_info2 = h.get_auth_info(name, can_create=False)
-        self.assert_(auth_info2)
+        auth_info2 = h.get_auth_info(name)
+        self.assertTrue(auth_info2)
 
         gid = auth_info2.get_gid_object()
-        self.assert_(gid)
+        self.assertTrue(gid)
         self.assertEqual(gid.get_subject(), name)
 
         pubkey = auth_info2.get_pkey_object()
-        self.assert_(gid)
+        self.assertTrue(gid)
 
 
 if __name__ == "__main__":