added more tests for records

[sfa.git] / tests / testHierarchy.py

diff --git a/tests/testHierarchy.py b/tests/testHierarchy.py
index a094af0..5262009 100644 (file)
--- a/tests/testHierarchy.py
+++ b/tests/testHierarchy.py
@@ -17,23 +17,27 @@ class TestHierarchy(unittest.TestCase):
     def testGetAuthInfo(self):
         h = Hierarchy(BASEDIR)
 
-        auth_info = h.get_auth_info("planetlab.us.arizona.stork", "sa", can_create=True)
+        name = "planetlab.us.arizona.stork"
+
+        self.assertEqual(h.auth_exists(name), False)
+
+        auth_info = h.get_auth_info(name, can_create=True)
         self.assert_(auth_info)
 
         gid = auth_info.get_gid_object()
         self.assert_(gid)
-        self.assertEqual(gid.get_subject(), "planetlab.us.arizona.stork")
+        self.assertEqual(gid.get_subject(), name)
 
         pubkey = auth_info.get_pkey_object()
         self.assert_(gid)
 
         # try to get it again, make sure it's still there
-        auth_info2 = h.get_auth_info("planetlab.us.arizona.stork", "sa", can_create=False)
+        auth_info2 = h.get_auth_info(name, can_create=False)
         self.assert_(auth_info2)
 
         gid = auth_info2.get_gid_object()
         self.assert_(gid)
-        self.assertEqual(gid.get_subject(), "planetlab.us.arizona.stork")
+        self.assertEqual(gid.get_subject(), name)
 
         pubkey = auth_info2.get_pkey_object()
         self.assert_(gid)