3 from hierarchy import *
6 BASEDIR = "test_hierarchy"
7 PURGE_BASEDIR = "rm -rf test_hierarchy"
9 class TestHierarchy(unittest.TestCase):
11 os.system(PURGE_BASEDIR)
15 h = Hierarchy(BASEDIR)
17 def testGetAuthInfo(self):
18 h = Hierarchy(BASEDIR)
20 auth_info = h.get_auth_info("planetlab.us.arizona.stork", "sa", can_create=True)
21 self.assert_(auth_info)
23 gid = auth_info.get_gid_object()
25 self.assertEqual(gid.get_subject(), "planetlab.us.arizona.stork")
27 pubkey = auth_info.get_pkey_object()
30 # try to get it again, make sure it's still there
31 auth_info2 = h.get_auth_info("planetlab.us.arizona.stork", "sa", can_create=False)
32 self.assert_(auth_info2)
34 gid = auth_info2.get_gid_object()
36 self.assertEqual(gid.get_subject(), "planetlab.us.arizona.stork")
38 pubkey = auth_info2.get_pkey_object()
42 if __name__ == "__main__":