Test token search working

pyauth · Feb 7, 2021 · beb0260 · beb0260
1 parent 2828f03
commit beb0260
Showing 1 changed file with 1 addition and 9 deletions.
diff --git a/tests/test_slots_and_tokens.py b/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)
-        '''