Clean up Python APIs to C_FindObject*() and C_GetSlotList().

1 files changed, 13 insertions, 10 deletions

diff --git a/unit_tests.py b/unit_tests.py
index cd47886..75f2602 100644
--- a/unit_tests.py
+++ b/unit_tests.py
@@ -50,7 +50,7 @@ class TestDevice(unittest.TestCase):
     p11.C_CloseAllSessions(only_slot)
 
   def test_getSlots(self):
-    self.assertEqual(p11.C_GetSlotList(), [only_slot])
+    self.assertEqual(p11.C_GetSlotList(), (only_slot,))
 
   def test_getTokenInfo(self):
     token_info = p11.C_GetTokenInfo(only_slot)
@@ -62,8 +62,7 @@ class TestDevice(unittest.TestCase):
     ro_session = p11.C_OpenSession(only_slot, CKF_SERIAL_SESSION)
 
   def test_sessions_parallel(self):
-    # Cooked API doesn't allow the user to make this mistake, so we
-    # have to use the raw API to test it.
+    # Cooked API doesn't allow this mistake, must use raw API to test
     from ctypes import byref
     handle = CK_SESSION_HANDLE()
     notify = CK_NOTIFY()
@@ -96,6 +95,16 @@ class TestDevice(unittest.TestCase):
     self.assertIsInstance(random, str)
     self.assertEqual(len(random), n)
 
+  def test_findObjects(self):
+    session = p11.C_OpenSession(only_slot)
+    p11.C_FindObjectsInit(session, CKA_CLASS = CKO_PUBLIC_KEY)
+    with self.assertRaises(CKR_OPERATION_ACTIVE):
+      p11.C_FindObjectsInit(session, CKA_CLASS = CKO_PRIVATE_KEY)
+    for handle in p11.C_FindObjects(session):
+      self.assertIsInstance(handle, (int, long))
+    p11.C_FindObjectsFinal(session)
+
+
 class TestKeys(unittest.TestCase):
   """
   Tests involving keys.
@@ -118,7 +127,7 @@ class TestKeys(unittest.TestCase):
     p11.C_Login(self.session, CKU_USER, user_pin)
 
   def tearDown(self):
-    for handle in p11.Find(self.session):
+    for handle in p11.FindObjects(self.session):
       p11.C_DestroyObject(self.session, handle)
     p11.C_CloseAllSessions(only_slot)
     del self.session
@@ -223,9 +232,6 @@ class TestKeys(unittest.TestCase):
       CKA_LABEL         = "EC-P-256 test case from \"Suite B Implementer's Guide to FIPS 186-3\"",
       CKA_ID            = "EC-P-256",
       CKA_VERIFY        = True,
-      CKA_ENCRYPT       = False,
-      CKA_WRAP          = False,
-      CKA_TOKEN         = False,
       CKA_EC_POINT      = Q,
       CKA_EC_PARAMS     = self.oid_p256)
     p11.C_VerifyInit(self.session, CKM_ECDSA, handle)
@@ -244,9 +250,6 @@ class TestKeys(unittest.TestCase):
       CKA_LABEL         = "EC-P-384 test case from \"Suite B Implementer's Guide to FIPS 186-3\"",
       CKA_ID            = "EC-P-384",
       CKA_VERIFY        = True,
-      CKA_ENCRYPT       = False,
-      CKA_WRAP          = False,
-      CKA_TOKEN         = False,
       CKA_EC_POINT      = Q,
       CKA_EC_PARAMS     = self.oid_p384)
     p11.C_VerifyInit(self.session, CKM_ECDSA, handle)