management.tests.test_pkcs11_aes_key

Tests for PKCS#11 Pkcs11AESKey class.

Classes

TestPkcs11AESKey

Test cases for the Pkcs11AESKey class.

Module Contents

class management.tests.test_pkcs11_aes_key.TestPkcs11AESKey

Test cases for the Pkcs11AESKey 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

aes_key()

Create Pkcs11AESKey instance without initializing session.

Return type:

management.pkcs11_util.Pkcs11AESKey

aes_key_with_session(mock_session)

Create Pkcs11AESKey instance with mocked session.

Parameters:

mock_session (unittest.mock.Mock)

Return type:

management.pkcs11_util.Pkcs11AESKey

test_init(aes_key)

Test Pkcs11AESKey initialization.

Parameters:

aes_key (management.pkcs11_util.Pkcs11AESKey)

Return type:

None

test_supported_key_lengths()

Return type:

None

test_load_key_general_error(aes_key_with_session, mock_session)

Test load_key with general error.

Parameters:

• aes_key_with_session (management.pkcs11_util.Pkcs11AESKey)

• mock_session (unittest.mock.Mock)

Return type:

None

test_load_key_initializes_session(aes_key)

Test load_key initializes session if None.

Parameters:

aes_key (management.pkcs11_util.Pkcs11AESKey)

Return type:

None

test_generate_key_default_length(aes_key_with_session, mock_session)

Test generate_key with default key length.

Parameters:

• aes_key_with_session (management.pkcs11_util.Pkcs11AESKey)

• mock_session (unittest.mock.Mock)

Return type:

None

test_generate_key_custom_length(aes_key_with_session, mock_session)

Test generate_key with custom key length.

Parameters:

• aes_key_with_session (management.pkcs11_util.Pkcs11AESKey)

• mock_session (unittest.mock.Mock)

Return type:

None

test_generate_key_192_length(aes_key_with_session, mock_session)

Test generate_key with 192-bit key length.

Parameters:

• aes_key_with_session (management.pkcs11_util.Pkcs11AESKey)

• mock_session (unittest.mock.Mock)

Return type:

None

test_generate_key_unsupported_length(aes_key_with_session)

Test generate_key with unsupported key length.

Parameters:

aes_key_with_session (management.pkcs11_util.Pkcs11AESKey)

Return type:

None

test_generate_key_another_unsupported_length(aes_key_with_session)

Test generate_key with another unsupported key length.

Parameters:

aes_key_with_session (management.pkcs11_util.Pkcs11AESKey)

Return type:

None

test_generate_key_initializes_session(aes_key)

Test generate_key initializes session if None.

Parameters:

aes_key (management.pkcs11_util.Pkcs11AESKey)

Return type:

None

test_generate_key_failure(aes_key_with_session, mock_session)

Test generate_key when key generation fails.

Parameters:

• aes_key_with_session (management.pkcs11_util.Pkcs11AESKey)

• mock_session (unittest.mock.Mock)

Return type:

None

test_close_with_session(aes_key_with_session, mock_session)

Test close method when session exists.

Parameters:

• aes_key_with_session (management.pkcs11_util.Pkcs11AESKey)

• mock_session (unittest.mock.Mock)

Return type:

None

test_close_no_session(aes_key)

Test close when no session exists.

Parameters:

aes_key (management.pkcs11_util.Pkcs11AESKey)

Return type:

None

test_close_with_exception(aes_key_with_session, mock_session)

Test close when session.close() raises exception.

Parameters:

• aes_key_with_session (management.pkcs11_util.Pkcs11AESKey)

• mock_session (unittest.mock.Mock)

Return type:

None

test_context_manager(aes_key)

Test context manager functionality.

Parameters:

aes_key (management.pkcs11_util.Pkcs11AESKey)

Return type:

None

test_context_manager_with_exception(aes_key)

Test context manager functionality when exception occurs.

Parameters:

aes_key (management.pkcs11_util.Pkcs11AESKey)

Return type:

None

test_multiple_operations_workflow(mock_session)

Test a complete workflow with multiple operations.

Parameters:

mock_session (unittest.mock.Mock)

Return type:

None

test_load_then_generate_error(aes_key_with_session, mock_session)

Test that we can handle both load and generate operations.

Parameters:

• aes_key_with_session (management.pkcs11_util.Pkcs11AESKey)

• mock_session (unittest.mock.Mock)

Return type:

None

test_all_supported_key_lengths(aes_key_with_session, mock_session)

Test that all supported key lengths work.

Parameters:

• aes_key_with_session (management.pkcs11_util.Pkcs11AESKey)

• mock_session (unittest.mock.Mock)

Return type:

None

test_string_representation_attributes(aes_key)

Test that all expected attributes are present.

Parameters:

aes_key (management.pkcs11_util.Pkcs11AESKey)

Return type:

None