gacks handle test case

[sfa.git] / tests / testHierarchy.py

diff --git a/tests/testHierarchy.py b/tests/testHierarchy.py
index a094af0..618c1cd 100644 (file)
--- a/tests/testHierarchy.py
+++ b/tests/testHierarchy.py
@@ -1,5 +1,6 @@
 import unittest
 import os
+from excep import *
 from hierarchy import *
 from config import *
 
@@ -17,23 +18,30 @@ 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)
+
+        self.assertRaises(MissingAuthority, h.get_auth_info, name)
+
+        h.create_auth(name, create_parents=True)
+        auth_info = h.get_auth_info(name)
         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)
         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)