Test token search working

Author: maxwolfe

tests/test_slots_and_tokens.py

@@ -88,15 +88,7 @@ def test_init_token(self):
             raise AssertionError("No Uninitialized token found")
 
         lib.reinitialize()
-        updated_tokens = lib.get_tokens()
+        token, *_ = lib.get_tokens(token_label=temp_token_label)
 
-        for token in updated_tokens:
-            if pkcs11.TokenFlag.USER_PIN_INITIALIZED not in token.flags and\
-            pkcs11.TokenFlag.TOKEN_INITIALIZED in token.flags:
-                self.assertEqual(token.label, temp_token_label)
-
-        '''
-        self.assertEqual(token.label, temp_token_label)
         self.assertIn(pkcs11.TokenFlag.TOKEN_INITIALIZED, token.flags)
         self.assertNotIn(pkcs11.TokenFlag.USER_PIN_INITIALIZED, token.flags)
-        '''