fix formatting, add tests for init_pin

Author: Dominikus Gierlach

pkcs11/types.py

@@ -577,9 +577,9 @@ def set_pin(self, old_pin, new_pin):
 
     def init_pin(self, pin):
         """
-        Initializes the user PIN. 
+        Initializes the user PIN.
 
-        Differs from set_pin in that it sets the user PIN for the first time. 
+        Differs from set_pin in that it sets the user PIN for the first time.
         Once set, the pin can be changed using set_pin.
         """
         raise NotImplementedError()

tests/test_sessions.py

@@ -1,6 +1,7 @@
 """
 PKCS#11 Sessions
 """
+
 import pkcs11
 from pkcs11 import (
     Attribute,
@@ -318,3 +319,25 @@ def test_set_pin(self):
                 session.set_pin(old_token_pin, "")
             with self.assertRaises(PinIncorrect):
                 session.set_pin("", new_token_pin)
+
+    @Only.softhsm2
+    def test_init_pin(self):
+        new_token_pin = f"{TOKEN_PIN}56"
+
+        with self.token.open(rw=True, so_pin=TOKEN_SO_PIN) as session:
+            session.init_pin(new_token_pin)
+
+        with self.token.open(rw=True, user_pin=new_token_pin) as session:
+            self.assertIsInstance(session, pkcs11.Session)
+
+        with self.token.open(rw=True, so_pin=TOKEN_SO_PIN) as session:
+            session.init_pin(TOKEN_PIN)
+
+        with self.token.open(rw=True, user_pin=TOKEN_PIN) as session:
+            self.assertIsInstance(session, pkcs11.Session)
+
+        with self.token.open(rw=True, so_pin=TOKEN_SO_PIN) as session:
+            with self.assertRaises(AttributeError):
+                session.init_pin(None)
+            with self.assertRaises(PinLenRange):
+                session.init_pin("")