management.tests.test_pkcs11_private_key

Tests for PKCS#11 Pkcs11PrivateKey class.

Classes

MockPkcs11PrivateKey	Mock implementation of Pkcs11PrivateKey for testing.
TestPkcs11PrivateKey	Test cases for the Pkcs11PrivateKey abstract base class.

Module Contents

class management.tests.test_pkcs11_private_key.MockPkcs11PrivateKey(lib_path, token_label, user_pin, key_label, slot_id=None)

Bases: management.pkcs11_util.Pkcs11PrivateKey

Mock implementation of Pkcs11PrivateKey for testing.

Parameters:

• lib_path (str)

• token_label (str)

• user_pin (str)

• key_label (str)

• slot_id (int | None)

sign(data, *args, **kwargs)

Mock sign method.

Parameters:

• data (bytes)

• args (Any)

• kwargs (Any)

Return type:

bytes

public_key()

Mock public_key method.

Return type:

cryptography.hazmat.primitives.asymmetric.rsa.RSAPublicKey | cryptography.hazmat.primitives.asymmetric.ec.EllipticCurvePublicKey

property key_size: int

Mock key_size property.

Return type:

int

class management.tests.test_pkcs11_private_key.TestPkcs11PrivateKey

Test cases for the Pkcs11PrivateKey abstract base class.

mock_session()

Mock PKCS#11 session.

Return type:

unittest.mock.Mock

mock_token(mock_session)

Mock PKCS#11 token.

Parameters:

mock_session (unittest.mock.Mock)

Return type:

unittest.mock.Mock

mock_lib(mock_token)

Mock PKCS#11 library.

Parameters:

mock_token (unittest.mock.Mock)

Return type:

unittest.mock.Mock

pkcs11_key(mock_session)

Create MockPkcs11PrivateKey instance with mocked dependencies.

Parameters:

mock_session (unittest.mock.Mock)

Return type:

MockPkcs11PrivateKey

test_digest_mechanisms_mapping()

Test that DIGEST_MECHANISMS contains expected mappings.

Return type:

None

test_digest_data_success(pkcs11_key, mock_session)

Test digest_data method with successful operation.

Parameters:

• pkcs11_key (MockPkcs11PrivateKey)

• mock_session (unittest.mock.Mock)

Return type:

None

test_digest_data_unsupported_algorithm(pkcs11_key)

Test digest_data with unsupported algorithm.

Parameters:

pkcs11_key (MockPkcs11PrivateKey)

Return type:

None

test_digest_data_no_session()

Test digest_data when session is None.

Return type:

None

test_copy_key_success(pkcs11_key, mock_session)

Test copy_key method with successful operation.

Parameters:

• pkcs11_key (MockPkcs11PrivateKey)

• mock_session (unittest.mock.Mock)

Return type:

None

test_copy_key_with_template(pkcs11_key, mock_session)

Test copy_key with additional template.

Parameters:

• pkcs11_key (MockPkcs11PrivateKey)

• mock_session (unittest.mock.Mock)

Return type:

None

test_copy_key_no_session()

Test copy_key when session is None.

Return type:

None

test_destroy_object_success(pkcs11_key, mock_session)

Test destroy_object method with successful operation.

Parameters:

• pkcs11_key (MockPkcs11PrivateKey)

• mock_session (unittest.mock.Mock)

Return type:

None

test_destroy_object_not_found(pkcs11_key, mock_session)

Test destroy_object when object not found.

Parameters:

• pkcs11_key (MockPkcs11PrivateKey)

• mock_session (unittest.mock.Mock)

Return type:

None

test_key_exists_true(pkcs11_key, mock_session)

Test _key_exists when key exists.

Parameters:

• pkcs11_key (MockPkcs11PrivateKey)

• mock_session (unittest.mock.Mock)

Return type:

None

test_key_exists_false(pkcs11_key, mock_session)

Test _key_exists when key does not exist.

Parameters:

• pkcs11_key (MockPkcs11PrivateKey)

• mock_session (unittest.mock.Mock)

Return type:

None

test_destroy_key_success(pkcs11_key)

Test destroy_key method with successful operation.

Parameters:

pkcs11_key (MockPkcs11PrivateKey)

Return type:

None

test_destroy_key_no_key(pkcs11_key)

Test destroy_key when no key exists.

Parameters:

pkcs11_key (MockPkcs11PrivateKey)

Return type:

None

test_destroy_key_pkcs11_error(pkcs11_key)

Test destroy_key when PKCS11Error occurs.

Parameters:

pkcs11_key (MockPkcs11PrivateKey)

Return type:

None

test_close_with_session(pkcs11_key, mock_session)

Test close method when session exists.

Parameters:

• pkcs11_key (MockPkcs11PrivateKey)

• mock_session (unittest.mock.Mock)

Return type:

None

test_close_no_session()

Test close when no session exists.

Return type:

None

test_context_manager(pkcs11_key)

Test context manager functionality.

Parameters:

pkcs11_key (MockPkcs11PrivateKey)

Return type:

None

test_abstract_methods()

Test that abstract methods are defined.

Return type:

None

test_mock_implementation_methods(pkcs11_key)

Test that the mock implementation works correctly.

Parameters:

pkcs11_key (MockPkcs11PrivateKey)

Return type:

None

test_initialization_success()

Test successful initialization of Pkcs11PrivateKey.

Return type:

None

test_initialization_user_already_logged_in()

Test initialization when user is already logged in.

Return type:

None

test_initialization_failure()

Test initialization failure.

Return type:

None

test_raise_methods(pkcs11_key)

Test the private _raise methods work correctly.

Parameters:

pkcs11_key (MockPkcs11PrivateKey)

Return type:

None