tests/testHierarchy.py

   1 import unittest
   2 import os
   3 from hierarchy import *
   4 from config import *
   5
   6 BASEDIR = "test_hierarchy"
   7 PURGE_BASEDIR = "rm -rf test_hierarchy"
   8
   9 class TestHierarchy(unittest.TestCase):
  10     def setUp(self):
  11         os.system(PURGE_BASEDIR)
  12         pass
  13
  14     def testInit(self):
  15         h = Hierarchy(BASEDIR)
  16
  17     def testGetAuthInfo(self):
  18         h = Hierarchy(BASEDIR)
  19
  20         name = "planetlab.us.arizona.stork"
  21
  22         self.assertEqual(h.auth_exists(name), False)
  23
  24         auth_info = h.get_auth_info(name, can_create=True)
  25         self.assert_(auth_info)
  26
  27         gid = auth_info.get_gid_object()
  28         self.assert_(gid)
  29         self.assertEqual(gid.get_subject(), name)
  30
  31         pubkey = auth_info.get_pkey_object()
  32         self.assert_(gid)
  33
  34         # try to get it again, make sure it's still there
  35         auth_info2 = h.get_auth_info(name, can_create=False)
  36         self.assert_(auth_info2)
  37
  38         gid = auth_info2.get_gid_object()
  39         self.assert_(gid)
  40         self.assertEqual(gid.get_subject(), name)
  41
  42         pubkey = auth_info2.get_pkey_object()
  43         self.assert_(gid)
  44
  45
  46 if __name__ == "__main__":
  47     unittest.main()