checked in test cases

[sfa.git] / tests / testCred.py

diff --git a/tests/testCred.py b/tests/testCred.py
new file mode 100644 (file)
index 0000000..b1d97b7
--- /dev/null
+++ b/tests/testCred.py
@@ -0,0 +1,57 @@
+import unittest
+from credential import *
+from rights import *
+from gid import *
+
+class TestCred(unittest.TestCase):
+   def setUp(self):
+      pass
+
+   def testCreate(self):
+      cred = Credential(create=True)
+
+   def testDefaults(self):
+      cred = Credential(subject="testCredential")
+
+      self.assertEqual(cred.get_gid_caller(), None)
+      self.assertEqual(cred.get_gid_object(), None)
+
+   def testLoadSave(self):
+      cred = Credential(subject="testCredential")
+
+      gidCaller = GID(subject="caller", uuid=create_uuid(), hrn="foo.caller")
+      gidObject = GID(subject="object", uuid=create_uuid(), hrn="foo.object")
+      lifeTime = 12345
+      delegate = True
+      rights = "embed,bind"
+
+      cred.set_gid_caller(gidCaller)
+      self.assertEqual(cred.get_gid_caller().get_subject(), gidCaller.get_subject())
+
+      cred.set_gid_object(gidObject)
+      self.assertEqual(cred.get_gid_object().get_subject(), gidObject.get_subject())
+
+      cred.set_lifetime(lifeTime)
+      self.assertEqual(cred.get_lifetime(), lifeTime)
+
+      cred.set_delegate(delegate)
+      self.assertEqual(cred.get_delegate(), delegate)
+
+      cred.set_privileges(rights)
+      self.assertEqual(cred.get_privileges().save_to_string(), rights)
+
+      cred.encode()
+
+      cred_str = cred.save_to_string()
+
+      # re-load the credential from a string and make sure it's fields are
+      # intact
+      cred2 = Credential(string = cred_str)
+      self.assertEqual(cred2.get_gid_caller().get_subject(), gidCaller.get_subject())
+      self.assertEqual(cred2.get_gid_object().get_subject(), gidObject.get_subject())
+      self.assertEqual(cred2.get_lifetime(), lifeTime)
+      self.assertEqual(cred2.get_delegate(), delegate)
+      self.assertEqual(cred2.get_privileges().save_to_string(), rights)
+
+if __name__ == "__main__":
+    unittest.main()