Update payload management

This change updates payload management, streamlining the import
process for kmip.core.messages.payloads. Now any request or
response payload is accessible by importing payloads. All code
importing and using individual payload modules has been updated
to use this new approach.
This commit is contained in:
Peter Hamilton 2017-07-31 15:48:29 -04:00
parent 7f6f4d004f
commit 82a7b8a741
23 changed files with 1092 additions and 1087 deletions

View File

@ -14,83 +14,64 @@
# under the License.
from kmip.core.factories.payloads import PayloadFactory
from kmip.core.messages.payloads import activate
from kmip.core.messages.payloads import create
from kmip.core.messages.payloads import create_key_pair
from kmip.core.messages.payloads import decrypt
from kmip.core.messages.payloads import derive_key
from kmip.core.messages.payloads import destroy
from kmip.core.messages.payloads import discover_versions
from kmip.core.messages.payloads import encrypt
from kmip.core.messages.payloads import get
from kmip.core.messages.payloads import get_attribute_list
from kmip.core.messages.payloads import get_attributes
from kmip.core.messages.payloads import locate
from kmip.core.messages.payloads import query
from kmip.core.messages.payloads import rekey_key_pair
from kmip.core.messages.payloads import register
from kmip.core.messages.payloads import revoke
from kmip.core.messages.payloads import sign
from kmip.core.messages.payloads import signature_verify
from kmip.core.messages.payloads import mac
from kmip.core.messages import payloads
class RequestPayloadFactory(PayloadFactory):
def _create_create_payload(self):
return create.CreateRequestPayload()
return payloads.CreateRequestPayload()
def _create_create_key_pair_payload(self):
return create_key_pair.CreateKeyPairRequestPayload()
return payloads.CreateKeyPairRequestPayload()
def _create_register_payload(self):
return register.RegisterRequestPayload()
return payloads.RegisterRequestPayload()
def _create_derive_key_payload(self):
return derive_key.DeriveKeyRequestPayload()
return payloads.DeriveKeyRequestPayload()
def _create_rekey_key_pair_payload(self):
return rekey_key_pair.RekeyKeyPairRequestPayload()
return payloads.RekeyKeyPairRequestPayload()
def _create_locate_payload(self):
return locate.LocateRequestPayload()
return payloads.LocateRequestPayload()
def _create_get_payload(self):
return get.GetRequestPayload()
return payloads.GetRequestPayload()
def _create_get_attribute_list_payload(self):
return get_attribute_list.GetAttributeListRequestPayload()
return payloads.GetAttributeListRequestPayload()
def _create_get_attributes_payload(self):
return get_attributes.GetAttributesRequestPayload()
return payloads.GetAttributesRequestPayload()
def _create_destroy_payload(self):
return destroy.DestroyRequestPayload()
return payloads.DestroyRequestPayload()
def _create_query_payload(self):
return query.QueryRequestPayload()
return payloads.QueryRequestPayload()
def _create_discover_versions_payload(self):
return discover_versions.DiscoverVersionsRequestPayload()
return payloads.DiscoverVersionsRequestPayload()
def _create_activate_payload(self):
return activate.ActivateRequestPayload()
return payloads.ActivateRequestPayload()
def _create_revoke_payload(self):
return revoke.RevokeRequestPayload()
return payloads.RevokeRequestPayload()
def _create_mac_payload(self):
return mac.MACRequestPayload()
return payloads.MACRequestPayload()
def _create_encrypt_payload(self):
return encrypt.EncryptRequestPayload()
return payloads.EncryptRequestPayload()
def _create_decrypt_payload(self):
return decrypt.DecryptRequestPayload()
return payloads.DecryptRequestPayload()
def _create_sign_payload(self):
return sign.SignRequestPayload()
return payloads.SignRequestPayload()
def _create_signature_verify_payload(self):
return signature_verify.SignatureVerifyRequestPayload()
return payloads.SignatureVerifyRequestPayload()

View File

@ -14,83 +14,64 @@
# under the License.
from kmip.core.factories.payloads import PayloadFactory
from kmip.core.messages.payloads import activate
from kmip.core.messages.payloads import create
from kmip.core.messages.payloads import create_key_pair
from kmip.core.messages.payloads import decrypt
from kmip.core.messages.payloads import destroy
from kmip.core.messages.payloads import derive_key
from kmip.core.messages.payloads import discover_versions
from kmip.core.messages.payloads import encrypt
from kmip.core.messages.payloads import get
from kmip.core.messages.payloads import get_attribute_list
from kmip.core.messages.payloads import get_attributes
from kmip.core.messages.payloads import locate
from kmip.core.messages.payloads import query
from kmip.core.messages.payloads import rekey_key_pair
from kmip.core.messages.payloads import register
from kmip.core.messages.payloads import revoke
from kmip.core.messages.payloads import mac
from kmip.core.messages.payloads import sign
from kmip.core.messages.payloads import signature_verify
from kmip.core.messages import payloads
class ResponsePayloadFactory(PayloadFactory):
def _create_create_payload(self):
return create.CreateResponsePayload()
return payloads.CreateResponsePayload()
def _create_create_key_pair_payload(self):
return create_key_pair.CreateKeyPairResponsePayload()
return payloads.CreateKeyPairResponsePayload()
def _create_register_payload(self):
return register.RegisterResponsePayload()
return payloads.RegisterResponsePayload()
def _create_derive_key_payload(self):
return derive_key.DeriveKeyResponsePayload()
return payloads.DeriveKeyResponsePayload()
def _create_rekey_key_pair_payload(self):
return rekey_key_pair.RekeyKeyPairResponsePayload()
return payloads.RekeyKeyPairResponsePayload()
def _create_locate_payload(self):
return locate.LocateResponsePayload()
return payloads.LocateResponsePayload()
def _create_get_payload(self):
return get.GetResponsePayload()
return payloads.GetResponsePayload()
def _create_get_attribute_list_payload(self):
return get_attribute_list.GetAttributeListResponsePayload()
return payloads.GetAttributeListResponsePayload()
def _create_get_attributes_payload(self):
return get_attributes.GetAttributesResponsePayload()
return payloads.GetAttributesResponsePayload()
def _create_destroy_payload(self):
return destroy.DestroyResponsePayload()
return payloads.DestroyResponsePayload()
def _create_query_payload(self):
return query.QueryResponsePayload()
return payloads.QueryResponsePayload()
def _create_discover_versions_payload(self):
return discover_versions.DiscoverVersionsResponsePayload()
return payloads.DiscoverVersionsResponsePayload()
def _create_activate_payload(self):
return activate.ActivateResponsePayload()
return payloads.ActivateResponsePayload()
def _create_revoke_payload(self):
return revoke.RevokeResponsePayload()
return payloads.RevokeResponsePayload()
def _create_mac_payload(self):
return mac.MACResponsePayload()
return payloads.MACResponsePayload()
def _create_encrypt_payload(self):
return encrypt.EncryptResponsePayload()
return payloads.EncryptResponsePayload()
def _create_decrypt_payload(self):
return decrypt.DecryptResponsePayload()
return payloads.DecryptResponsePayload()
def _create_sign_payload(self):
return sign.SignResponsePayload()
return payloads.SignResponsePayload()
def _create_signature_verify_payload(self):
return signature_verify.SignatureVerifyResponsePayload()
return payloads.SignatureVerifyResponsePayload()

View File

@ -13,4 +13,121 @@
# License for the specific language governing permissions and limitations
# under the License.
__all__ = ['create', 'destroy', 'get', 'locate', 'register']
from kmip.core.messages.payloads.activate import (
ActivateRequestPayload,
ActivateResponsePayload
)
from kmip.core.messages.payloads.create import (
CreateRequestPayload,
CreateResponsePayload
)
from kmip.core.messages.payloads.create_key_pair import (
CreateKeyPairRequestPayload,
CreateKeyPairResponsePayload
)
from kmip.core.messages.payloads.decrypt import (
DecryptRequestPayload,
DecryptResponsePayload
)
from kmip.core.messages.payloads.derive_key import (
DeriveKeyRequestPayload,
DeriveKeyResponsePayload
)
from kmip.core.messages.payloads.destroy import (
DestroyRequestPayload,
DestroyResponsePayload
)
from kmip.core.messages.payloads.discover_versions import (
DiscoverVersionsRequestPayload,
DiscoverVersionsResponsePayload
)
from kmip.core.messages.payloads.encrypt import (
EncryptRequestPayload,
EncryptResponsePayload
)
from kmip.core.messages.payloads.get import (
GetRequestPayload,
GetResponsePayload
)
from kmip.core.messages.payloads.get_attribute_list import (
GetAttributeListRequestPayload,
GetAttributeListResponsePayload
)
from kmip.core.messages.payloads.get_attributes import (
GetAttributesRequestPayload,
GetAttributesResponsePayload
)
from kmip.core.messages.payloads.locate import (
LocateRequestPayload,
LocateResponsePayload
)
from kmip.core.messages.payloads.mac import (
MACRequestPayload,
MACResponsePayload
)
from kmip.core.messages.payloads.query import (
QueryRequestPayload,
QueryResponsePayload
)
from kmip.core.messages.payloads.register import (
RegisterRequestPayload,
RegisterResponsePayload
)
from kmip.core.messages.payloads.rekey_key_pair import (
RekeyKeyPairRequestPayload,
RekeyKeyPairResponsePayload
)
from kmip.core.messages.payloads.revoke import (
RevokeRequestPayload,
RevokeResponsePayload
)
from kmip.core.messages.payloads.sign import (
SignRequestPayload,
SignResponsePayload
)
from kmip.core.messages.payloads.signature_verify import (
SignatureVerifyRequestPayload,
SignatureVerifyResponsePayload
)
__all__ = [
"ActivateRequestPayload",
"ActivateResponsePayload",
"CreateRequestPayload",
"CreateResponsePayload",
"CreateKeyPairRequestPayload",
"CreateKeyPairResponsePayload",
"DecryptRequestPayload",
"DecryptResponsePayload",
"DeriveKeyRequestPayload",
"DeriveKeyResponsePayload",
"DestroyRequestPayload",
"DestroyResponsePayload",
"DiscoverVersionsRequestPayload",
"DiscoverVersionsResponsePayload",
"EncryptRequestPayload",
"EncryptResponsePayload",
"GetRequestPayload",
"GetResponsePayload",
"GetAttributeListRequestPayload",
"GetAttributeListResponsePayload",
"GetAttributesRequestPayload",
"GetAttributesResponsePayload",
"LocateRequestPayload",
"LocateResponsePayload",
"MACRequestPayload",
"MACResponsePayload",
"QueryRequestPayload",
"QueryResponsePayload",
"RegisterRequestPayload",
"RegisterResponsePayload",
"RekeyKeyPairRequestPayload",
"RekeyKeyPairResponsePayload",
"RevokeRequestPayload",
"RevokeResponsePayload",
"SignRequestPayload",
"SignResponsePayload",
"SignatureVerifyRequestPayload",
"SignatureVerifyResponsePayload"
]

View File

@ -48,25 +48,7 @@ from kmip.core.messages.contents import ProtocolVersion
from kmip.core.messages import messages
from kmip.core.messages.payloads import activate
from kmip.core.messages.payloads import create
from kmip.core.messages.payloads import create_key_pair
from kmip.core.messages.payloads import decrypt
from kmip.core.messages.payloads import derive_key
from kmip.core.messages.payloads import destroy
from kmip.core.messages.payloads import discover_versions
from kmip.core.messages.payloads import encrypt
from kmip.core.messages.payloads import get
from kmip.core.messages.payloads import get_attributes
from kmip.core.messages.payloads import get_attribute_list
from kmip.core.messages.payloads import locate
from kmip.core.messages.payloads import query
from kmip.core.messages.payloads import rekey_key_pair
from kmip.core.messages.payloads import register
from kmip.core.messages.payloads import revoke
from kmip.core.messages.payloads import sign
from kmip.core.messages.payloads import signature_verify
from kmip.core.messages.payloads import mac
from kmip.core.messages import payloads
from kmip.services.server.kmip_protocol import KMIPProtocol
@ -349,7 +331,7 @@ class KMIPProxy(KMIP):
| context for the operation result.
"""
operation = Operation(OperationEnum.DERIVE_KEY)
request_payload = derive_key.DeriveKeyRequestPayload(
request_payload = payloads.DeriveKeyRequestPayload(
object_type=object_type,
unique_identifiers=unique_identifiers,
derivation_method=derivation_method,
@ -563,7 +545,7 @@ class KMIPProxy(KMIP):
"""
operation = Operation(OperationEnum.ENCRYPT)
request_payload = encrypt.EncryptRequestPayload(
request_payload = payloads.EncryptRequestPayload(
unique_identifier=unique_identifier,
data=data,
cryptographic_parameters=cryptographic_parameters,
@ -639,7 +621,7 @@ class KMIPProxy(KMIP):
"""
operation = Operation(OperationEnum.DECRYPT)
request_payload = decrypt.DecryptRequestPayload(
request_payload = payloads.DecryptRequestPayload(
unique_identifier=unique_identifier,
data=data,
cryptographic_parameters=cryptographic_parameters,
@ -714,7 +696,7 @@ class KMIPProxy(KMIP):
"""
operation = Operation(OperationEnum.SIGNATURE_VERIFY)
request_payload = signature_verify.SignatureVerifyRequestPayload(
request_payload = payloads.SignatureVerifyRequestPayload(
unique_identifier=unique_identifier,
cryptographic_parameters=cryptographic_parameters,
data=message,
@ -781,7 +763,7 @@ class KMIPProxy(KMIP):
"""
operation = Operation(OperationEnum.SIGN)
request_payload = sign.SignRequestPayload(
request_payload = payloads.SignRequestPayload(
unique_identifier=unique_identifier,
cryptographic_parameters=cryptographic_parameters,
data=data
@ -831,7 +813,7 @@ class KMIPProxy(KMIP):
if object_type is None:
raise ValueError('object_type cannot be None')
req_pl = create.CreateRequestPayload(
req_pl = payloads.CreateRequestPayload(
object_type=object_type,
template_attribute=template_attribute)
batch_item = messages.RequestBatchItem(operation=operation,
@ -867,7 +849,7 @@ class KMIPProxy(KMIP):
private_key_template_attribute=None,
public_key_template_attribute=None):
operation = Operation(OperationEnum.CREATE_KEY_PAIR)
payload = create_key_pair.CreateKeyPairRequestPayload(
payload = payloads.CreateKeyPairRequestPayload(
common_template_attribute=common_template_attribute,
private_key_template_attribute=private_key_template_attribute,
public_key_template_attribute=public_key_template_attribute)
@ -881,7 +863,7 @@ class KMIPProxy(KMIP):
private_key_template_attribute=None,
public_key_template_attribute=None):
operation = Operation(OperationEnum.REKEY_KEY_PAIR)
payload = rekey_key_pair.RekeyKeyPairRequestPayload(
payload = payloads.RekeyKeyPairRequestPayload(
private_key_uuid, offset,
common_template_attribute=common_template_attribute,
private_key_template_attribute=private_key_template_attribute,
@ -892,7 +874,7 @@ class KMIPProxy(KMIP):
def _build_query_batch_item(self, query_functions=None):
operation = Operation(OperationEnum.QUERY)
payload = query.QueryRequestPayload(query_functions)
payload = payloads.QueryRequestPayload(query_functions)
batch_item = messages.RequestBatchItem(
operation=operation, request_payload=payload)
return batch_item
@ -903,7 +885,7 @@ class KMIPProxy(KMIP):
attribute_names=None
):
operation = Operation(OperationEnum.GET_ATTRIBUTES)
payload = get_attributes.GetAttributesRequestPayload(
payload = payloads.GetAttributesRequestPayload(
uuid,
attribute_names
)
@ -915,7 +897,7 @@ class KMIPProxy(KMIP):
def _build_get_attribute_list_batch_item(self, uid=None):
operation = Operation(OperationEnum.GET_ATTRIBUTE_LIST)
payload = get_attribute_list.GetAttributeListRequestPayload(uid)
payload = payloads.GetAttributeListRequestPayload(uid)
batch_item = messages.RequestBatchItem(
operation=operation, request_payload=payload)
return batch_item
@ -923,7 +905,7 @@ class KMIPProxy(KMIP):
def _build_discover_versions_batch_item(self, protocol_versions=None):
operation = Operation(OperationEnum.DISCOVER_VERSIONS)
payload = discover_versions.DiscoverVersionsRequestPayload(
payload = payloads.DiscoverVersionsRequestPayload(
protocol_versions)
batch_item = messages.RequestBatchItem(
@ -1084,7 +1066,7 @@ class KMIPProxy(KMIP):
if key_wrapping_specification is not None:
kws = objects.KeyWrappingSpecification(key_wrapping_specification)
req_pl = get.GetRequestPayload(
req_pl = payloads.GetRequestPayload(
unique_identifier=unique_identifier,
key_format_type=key_format_type,
key_compression_type=key_compression_type,
@ -1126,7 +1108,7 @@ class KMIPProxy(KMIP):
if unique_identifier is not None:
uuid = attr.UniqueIdentifier(unique_identifier)
payload = activate.ActivateRequestPayload(unique_identifier=uuid)
payload = payloads.ActivateRequestPayload(unique_identifier=uuid)
batch_item = messages.RequestBatchItem(operation=operation,
request_payload=payload)
@ -1159,7 +1141,7 @@ class KMIPProxy(KMIP):
if unique_identifier is not None:
uuid = attr.UniqueIdentifier(unique_identifier)
payload = destroy.DestroyRequestPayload(unique_identifier=uuid)
payload = payloads.DestroyRequestPayload(unique_identifier=uuid)
batch_item = messages.RequestBatchItem(operation=operation,
request_payload=payload)
@ -1194,7 +1176,7 @@ class KMIPProxy(KMIP):
if unique_identifier is not None:
uuid = attr.UniqueIdentifier(unique_identifier)
payload = revoke.RevokeRequestPayload(
payload = payloads.RevokeRequestPayload(
unique_identifier=uuid,
revocation_reason=reason,
compromise_occurrence_date=compromise_occurrence_date)
@ -1231,7 +1213,7 @@ class KMIPProxy(KMIP):
if object_type is None:
raise ValueError('object_type cannot be None')
req_pl = register.RegisterRequestPayload(
req_pl = payloads.RegisterRequestPayload(
object_type=object_type,
template_attribute=template_attribute,
secret=secret)
@ -1271,18 +1253,18 @@ class KMIPProxy(KMIP):
objgrp = None
if maximum_items is not None:
mxi = locate.LocateRequestPayload.MaximumItems(maximum_items)
mxi = payloads.LocateRequestPayload.MaximumItems(maximum_items)
if storage_status_mask is not None:
m = storage_status_mask
ssmask = locate.LocateRequestPayload.StorageStatusMask(m)
ssmask = payloads.LocateRequestPayload.StorageStatusMask(m)
if object_group_member is not None:
o = object_group_member
objgrp = locate.LocateRequestPayload.ObjectGroupMember(o)
objgrp = payloads.LocateRequestPayload.ObjectGroupMember(o)
payload = locate.LocateRequestPayload(maximum_items=mxi,
storage_status_mask=ssmask,
object_group_member=objgrp,
attributes=attributes)
payload = payloads.LocateRequestPayload(maximum_items=mxi,
storage_status_mask=ssmask,
object_group_member=objgrp,
attributes=attributes)
batch_item = messages.RequestBatchItem(operation=operation,
request_payload=payload)
@ -1315,7 +1297,7 @@ class KMIPProxy(KMIP):
credential=None):
operation = Operation(OperationEnum.MAC)
req_pl = mac.MACRequestPayload(
req_pl = payloads.MACRequestPayload(
unique_identifier=attr.UniqueIdentifier(unique_identifier),
cryptographic_parameters=cryptographic_parameters,
data=objects.Data(data))

View File

@ -38,24 +38,7 @@ from kmip.core.factories import secrets
from kmip.core.messages import contents
from kmip.core.messages import messages
from kmip.core.messages.payloads import activate
from kmip.core.messages.payloads import revoke
from kmip.core.messages.payloads import create
from kmip.core.messages.payloads import create_key_pair
from kmip.core.messages.payloads import decrypt
from kmip.core.messages.payloads import derive_key
from kmip.core.messages.payloads import destroy
from kmip.core.messages.payloads import discover_versions
from kmip.core.messages.payloads import encrypt
from kmip.core.messages.payloads import get
from kmip.core.messages.payloads import get_attributes
from kmip.core.messages.payloads import get_attribute_list
from kmip.core.messages.payloads import query
from kmip.core.messages.payloads import register
from kmip.core.messages.payloads import mac
from kmip.core.messages.payloads import locate
from kmip.core.messages.payloads import sign
from kmip.core.messages.payloads import signature_verify
from kmip.core.messages import payloads
from kmip.core import misc
@ -1086,7 +1069,7 @@ class KmipEngine(object):
)
)
response_payload = create.CreateResponsePayload(
response_payload = payloads.CreateResponsePayload(
object_type=payload.object_type,
unique_identifier=attributes.UniqueIdentifier(
str(managed_object.unique_identifier)
@ -1262,7 +1245,7 @@ class KmipEngine(object):
)
)
response_payload = create_key_pair.CreateKeyPairResponsePayload(
response_payload = payloads.CreateKeyPairResponsePayload(
private_key_uuid=attributes.PrivateKeyUniqueIdentifier(
str(private_key.unique_identifier)
),
@ -1332,7 +1315,7 @@ class KmipEngine(object):
)
)
response_payload = register.RegisterResponsePayload(
response_payload = payloads.RegisterResponsePayload(
unique_identifier=attributes.UniqueIdentifier(
str(managed_object.unique_identifier)
)
@ -1521,7 +1504,7 @@ class KmipEngine(object):
)
self._id_placeholder = str(managed_object.unique_identifier)
response_payload = derive_key.DeriveKeyResponsePayload(
response_payload = payloads.DeriveKeyResponsePayload(
unique_identifier=str(managed_object.unique_identifier)
)
return response_payload
@ -1562,7 +1545,7 @@ class KmipEngine(object):
str(managed_object.unique_identifier))
for managed_object in managed_objects]
response_payload = locate.LocateResponsePayload(
response_payload = payloads.LocateResponsePayload(
unique_identifiers=unique_identifiers
)
@ -1716,7 +1699,7 @@ class KmipEngine(object):
else:
core_secret = self._build_core_object(managed_object)
response_payload = get.GetResponsePayload(
response_payload = payloads.GetResponsePayload(
object_type=managed_object._object_type,
unique_identifier=unique_identifier,
secret=core_secret
@ -1742,7 +1725,7 @@ class KmipEngine(object):
payload.attribute_names
)
response_payload = get_attributes.GetAttributesResponsePayload(
response_payload = payloads.GetAttributesResponsePayload(
unique_identifier=unique_identifier,
attributes=attrs
)
@ -1771,7 +1754,7 @@ class KmipEngine(object):
for object_attribute in object_attributes:
attribute_names.append(object_attribute.attribute_name.value)
response_payload = get_attribute_list.GetAttributeListResponsePayload(
response_payload = payloads.GetAttributeListResponsePayload(
unique_identifier=unique_identifier,
attribute_names=attribute_names
)
@ -1809,7 +1792,7 @@ class KmipEngine(object):
managed_object.state = enums.State.ACTIVE
self._data_session.commit()
response_payload = activate.ActivateResponsePayload(
response_payload = payloads.ActivateResponsePayload(
unique_identifier=attributes.UniqueIdentifier(unique_identifier)
)
@ -1864,7 +1847,7 @@ class KmipEngine(object):
managed_object.state = enums.State.DEACTIVATED
self._data_session.commit()
response_payload = revoke.RevokeResponsePayload(
response_payload = payloads.RevokeResponsePayload(
unique_identifier=attributes.UniqueIdentifier(unique_identifier)
)
@ -1904,7 +1887,7 @@ class KmipEngine(object):
objects.ManagedObject.unique_identifier == unique_identifier
).delete()
response_payload = destroy.DestroyResponsePayload(
response_payload = payloads.DestroyResponsePayload(
unique_identifier=attributes.UniqueIdentifier(unique_identifier)
)
@ -1968,7 +1951,7 @@ class KmipEngine(object):
if enums.QueryFunction.QUERY_EXTENSION_MAP in queries:
extensions = list()
response_payload = query.QueryResponsePayload(
response_payload = payloads.QueryResponsePayload(
operations=operations,
object_types=objects,
vendor_identification=vendor_identification,
@ -1991,7 +1974,7 @@ class KmipEngine(object):
else:
supported_versions = self._protocol_versions
response_payload = discover_versions.DiscoverVersionsResponsePayload(
response_payload = payloads.DiscoverVersionsResponsePayload(
protocol_versions=supported_versions
)
@ -2053,7 +2036,7 @@ class KmipEngine(object):
iv_nonce=payload.iv_counter_nonce
)
response_payload = encrypt.EncryptResponsePayload(
response_payload = payloads.EncryptResponsePayload(
unique_identifier,
result.get('cipher_text'),
result.get('iv_nonce')
@ -2116,7 +2099,7 @@ class KmipEngine(object):
iv_nonce=payload.iv_counter_nonce
)
response_payload = decrypt.DecryptResponsePayload(
response_payload = payloads.DecryptResponsePayload(
unique_identifier,
result
)
@ -2184,7 +2167,7 @@ class KmipEngine(object):
else:
validity = enums.ValidityIndicator.INVALID
response_payload = signature_verify.SignatureVerifyResponsePayload(
response_payload = payloads.SignatureVerifyResponsePayload(
unique_identifier=unique_identifier,
validity_indicator=validity
)
@ -2256,7 +2239,7 @@ class KmipEngine(object):
data
)
response_payload = mac.MACResponsePayload(
response_payload = payloads.MACResponsePayload(
unique_identifier=attributes.UniqueIdentifier(unique_identifier),
mac_data=MACData(result)
)
@ -2312,7 +2295,7 @@ class KmipEngine(object):
data=payload.data
)
response_payload = sign.SignResponsePayload(
response_payload = payloads.SignResponsePayload(
unique_identifier=unique_identifier,
signature_data=result
)

View File

@ -18,25 +18,7 @@ import testtools
from kmip.core import enums
from kmip.core.factories.payloads.request import RequestPayloadFactory
from kmip.core.messages.payloads import activate
from kmip.core.messages.payloads import create
from kmip.core.messages.payloads import create_key_pair
from kmip.core.messages.payloads import decrypt
from kmip.core.messages.payloads import destroy
from kmip.core.messages.payloads import derive_key
from kmip.core.messages.payloads import discover_versions
from kmip.core.messages.payloads import encrypt
from kmip.core.messages.payloads import get
from kmip.core.messages.payloads import get_attribute_list
from kmip.core.messages.payloads import get_attributes
from kmip.core.messages.payloads import locate
from kmip.core.messages.payloads import query
from kmip.core.messages.payloads import rekey_key_pair
from kmip.core.messages.payloads import register
from kmip.core.messages.payloads import revoke
from kmip.core.messages.payloads import sign
from kmip.core.messages.payloads import signature_verify
from kmip.core.messages.payloads import mac
from kmip.core.messages import payloads
class TestRequestPayloadFactory(testtools.TestCase):
@ -57,25 +39,25 @@ class TestRequestPayloadFactory(testtools.TestCase):
def test_create_create_payload(self):
payload = self.factory.create(enums.Operation.CREATE)
self._test_payload_type(payload, create.CreateRequestPayload)
self._test_payload_type(payload, payloads.CreateRequestPayload)
def test_create_create_key_pair_payload(self):
payload = self.factory.create(enums.Operation.CREATE_KEY_PAIR)
self._test_payload_type(
payload,
create_key_pair.CreateKeyPairRequestPayload
payloads.CreateKeyPairRequestPayload
)
def test_create_register_payload(self):
payload = self.factory.create(enums.Operation.REGISTER)
self._test_payload_type(payload, register.RegisterRequestPayload)
self._test_payload_type(payload, payloads.RegisterRequestPayload)
def test_create_rekey_payload(self):
self._test_not_implemented(self.factory.create, enums.Operation.REKEY)
def test_create_derive_key_payload(self):
payload = self.factory.create(enums.Operation.DERIVE_KEY)
self._test_payload_type(payload, derive_key.DeriveKeyRequestPayload)
self._test_payload_type(payload, payloads.DeriveKeyRequestPayload)
def test_create_certify_payload(self):
self._test_not_implemented(
@ -91,27 +73,27 @@ class TestRequestPayloadFactory(testtools.TestCase):
def test_create_locate_payload(self):
payload = self.factory.create(enums.Operation.LOCATE)
self._test_payload_type(payload, locate.LocateRequestPayload)
self._test_payload_type(payload, payloads.LocateRequestPayload)
def test_create_check_payload(self):
self._test_not_implemented(self.factory.create, enums.Operation.CHECK)
def test_create_get_payload(self):
payload = self.factory.create(enums.Operation.GET)
self._test_payload_type(payload, get.GetRequestPayload)
self._test_payload_type(payload, payloads.GetRequestPayload)
def test_create_get_attributes_payload(self):
payload = self.factory.create(enums.Operation.GET_ATTRIBUTES)
self._test_payload_type(
payload,
get_attributes.GetAttributesRequestPayload
payloads.GetAttributesRequestPayload
)
def test_create_get_attributes_list_payload(self):
payload = self.factory.create(enums.Operation.GET_ATTRIBUTE_LIST)
self._test_payload_type(
payload,
get_attribute_list.GetAttributeListRequestPayload
payloads.GetAttributeListRequestPayload
)
def test_create_add_attribute_payload(self):
@ -146,15 +128,15 @@ class TestRequestPayloadFactory(testtools.TestCase):
def test_create_activate_payload(self):
payload = self.factory.create(enums.Operation.ACTIVATE)
self._test_payload_type(payload, activate.ActivateRequestPayload)
self._test_payload_type(payload, payloads.ActivateRequestPayload)
def test_create_revoke_payload(self):
payload = self.factory.create(enums.Operation.REVOKE)
self._test_payload_type(payload, revoke.RevokeRequestPayload)
self._test_payload_type(payload, payloads.RevokeRequestPayload)
def test_create_destroy_payload(self):
payload = self.factory.create(enums.Operation.DESTROY)
self._test_payload_type(payload, destroy.DestroyRequestPayload)
self._test_payload_type(payload, payloads.DestroyRequestPayload)
def test_create_archive_payload(self):
self._test_not_implemented(
@ -176,7 +158,7 @@ class TestRequestPayloadFactory(testtools.TestCase):
def test_create_query_payload(self):
payload = self.factory.create(enums.Operation.QUERY)
self._test_payload_type(payload, query.QueryRequestPayload)
self._test_payload_type(payload, payloads.QueryRequestPayload)
def test_create_cancel_payload(self):
self._test_not_implemented(
@ -200,38 +182,38 @@ class TestRequestPayloadFactory(testtools.TestCase):
payload = self.factory.create(enums.Operation.REKEY_KEY_PAIR)
self._test_payload_type(
payload,
rekey_key_pair.RekeyKeyPairRequestPayload
payloads.RekeyKeyPairRequestPayload
)
def test_create_discover_versions_payload(self):
payload = self.factory.create(enums.Operation.DISCOVER_VERSIONS)
self._test_payload_type(
payload,
discover_versions.DiscoverVersionsRequestPayload
payloads.DiscoverVersionsRequestPayload
)
def test_create_encrypt_payload(self):
payload = self.factory.create(enums.Operation.ENCRYPT)
self._test_payload_type(payload, encrypt.EncryptRequestPayload)
self._test_payload_type(payload, payloads.EncryptRequestPayload)
def test_create_decrypt_payload(self):
payload = self.factory.create(enums.Operation.DECRYPT)
self._test_payload_type(payload, decrypt.DecryptRequestPayload)
self._test_payload_type(payload, payloads.DecryptRequestPayload)
def test_create_sign_payload(self):
payload = self.factory.create(enums.Operation.SIGN)
self._test_payload_type(payload, sign.SignRequestPayload)
self._test_payload_type(payload, payloads.SignRequestPayload)
def test_create_signature_verify_payload(self):
payload = self.factory.create(enums.Operation.SIGNATURE_VERIFY)
self._test_payload_type(
payload,
signature_verify.SignatureVerifyRequestPayload
payloads.SignatureVerifyRequestPayload
)
def test_create_mac_payload(self):
payload = self.factory.create(enums.Operation.MAC)
self._test_payload_type(payload, mac.MACRequestPayload)
self._test_payload_type(payload, payloads.MACRequestPayload)
def test_create_mac_verify_payload(self):
self._test_not_implemented(

View File

@ -18,25 +18,7 @@ import testtools
from kmip.core import enums
from kmip.core.factories.payloads.response import ResponsePayloadFactory
from kmip.core.messages.payloads import activate
from kmip.core.messages.payloads import create
from kmip.core.messages.payloads import create_key_pair
from kmip.core.messages.payloads import decrypt
from kmip.core.messages.payloads import destroy
from kmip.core.messages.payloads import derive_key
from kmip.core.messages.payloads import discover_versions
from kmip.core.messages.payloads import encrypt
from kmip.core.messages.payloads import get
from kmip.core.messages.payloads import get_attribute_list
from kmip.core.messages.payloads import get_attributes
from kmip.core.messages.payloads import locate
from kmip.core.messages.payloads import query
from kmip.core.messages.payloads import rekey_key_pair
from kmip.core.messages.payloads import register
from kmip.core.messages.payloads import revoke
from kmip.core.messages.payloads import sign
from kmip.core.messages.payloads import signature_verify
from kmip.core.messages.payloads import mac
from kmip.core.messages import payloads
class TestResponsePayloadFactory(testtools.TestCase):
@ -57,25 +39,25 @@ class TestResponsePayloadFactory(testtools.TestCase):
def test_create_create_payload(self):
payload = self.factory.create(enums.Operation.CREATE)
self._test_payload_type(payload, create.CreateResponsePayload)
self._test_payload_type(payload, payloads.CreateResponsePayload)
def test_create_create_key_pair_payload(self):
payload = self.factory.create(enums.Operation.CREATE_KEY_PAIR)
self._test_payload_type(
payload,
create_key_pair.CreateKeyPairResponsePayload
payloads.CreateKeyPairResponsePayload
)
def test_create_register_payload(self):
payload = self.factory.create(enums.Operation.REGISTER)
self._test_payload_type(payload, register.RegisterResponsePayload)
self._test_payload_type(payload, payloads.RegisterResponsePayload)
def test_create_rekey_payload(self):
self._test_not_implemented(self.factory.create, enums.Operation.REKEY)
def test_create_derive_key_payload(self):
payload = self.factory.create(enums.Operation.DERIVE_KEY)
self._test_payload_type(payload, derive_key.DeriveKeyResponsePayload)
self._test_payload_type(payload, payloads.DeriveKeyResponsePayload)
def test_create_certify_payload(self):
self._test_not_implemented(
@ -91,27 +73,27 @@ class TestResponsePayloadFactory(testtools.TestCase):
def test_create_locate_payload(self):
payload = self.factory.create(enums.Operation.LOCATE)
self._test_payload_type(payload, locate.LocateResponsePayload)
self._test_payload_type(payload, payloads.LocateResponsePayload)
def test_create_check_payload(self):
self._test_not_implemented(self.factory.create, enums.Operation.CHECK)
def test_create_get_payload(self):
payload = self.factory.create(enums.Operation.GET)
self._test_payload_type(payload, get.GetResponsePayload)
self._test_payload_type(payload, payloads.GetResponsePayload)
def test_create_get_attributes_payload(self):
payload = self.factory.create(enums.Operation.GET_ATTRIBUTES)
self._test_payload_type(
payload,
get_attributes.GetAttributesResponsePayload
payloads.GetAttributesResponsePayload
)
def test_create_get_attributes_list_payload(self):
payload = self.factory.create(enums.Operation.GET_ATTRIBUTE_LIST)
self._test_payload_type(
payload,
get_attribute_list.GetAttributeListResponsePayload
payloads.GetAttributeListResponsePayload
)
def test_create_add_attribute_payload(self):
@ -144,15 +126,15 @@ class TestResponsePayloadFactory(testtools.TestCase):
def test_create_activate_payload(self):
payload = self.factory.create(enums.Operation.ACTIVATE)
self._test_payload_type(payload, activate.ActivateResponsePayload)
self._test_payload_type(payload, payloads.ActivateResponsePayload)
def test_create_revoke_payload(self):
payload = self.factory.create(enums.Operation.REVOKE)
self._test_payload_type(payload, revoke.RevokeResponsePayload)
self._test_payload_type(payload, payloads.RevokeResponsePayload)
def test_create_destroy_payload(self):
payload = self.factory.create(enums.Operation.DESTROY)
self._test_payload_type(payload, destroy.DestroyResponsePayload)
self._test_payload_type(payload, payloads.DestroyResponsePayload)
def test_create_archive_payload(self):
self._test_not_implemented(
@ -174,7 +156,7 @@ class TestResponsePayloadFactory(testtools.TestCase):
def test_create_query_payload(self):
payload = self.factory.create(enums.Operation.QUERY)
self._test_payload_type(payload, query.QueryResponsePayload)
self._test_payload_type(payload, payloads.QueryResponsePayload)
def test_create_cancel_payload(self):
self._test_not_implemented(
@ -198,40 +180,40 @@ class TestResponsePayloadFactory(testtools.TestCase):
payload = self.factory.create(enums.Operation.REKEY_KEY_PAIR)
self._test_payload_type(
payload,
rekey_key_pair.RekeyKeyPairResponsePayload
payloads.RekeyKeyPairResponsePayload
)
def test_create_discover_versions_payload(self):
payload = self.factory.create(enums.Operation.DISCOVER_VERSIONS)
self._test_payload_type(
payload,
discover_versions.DiscoverVersionsResponsePayload
payloads.DiscoverVersionsResponsePayload
)
def test_create_encrypt_payload(self):
payload = self.factory.create(enums.Operation.ENCRYPT)
self._test_payload_type(payload, encrypt.EncryptResponsePayload)
self._test_payload_type(payload, payloads.EncryptResponsePayload)
def test_create_decrypt_payload(self):
payload = self.factory.create(enums.Operation.DECRYPT)
self._test_payload_type(payload, decrypt.DecryptResponsePayload)
self._test_payload_type(payload, payloads.DecryptResponsePayload)
def test_create_sign_payload(self):
payload = self.factory.create(enums.Operation.SIGN)
self._test_payload_type(payload, sign.SignResponsePayload)
self._test_payload_type(payload, payloads.SignResponsePayload)
def test_create_signature_verify_payload(self):
payload = self.factory.create(enums.Operation.SIGNATURE_VERIFY)
self._test_payload_type(
payload,
signature_verify.SignatureVerifyResponsePayload
payloads.SignatureVerifyResponsePayload
)
def test_create_mac_payload(self):
payload = self.factory.create(enums.Operation.MAC)
self._test_payload_type(
payload,
mac.MACResponsePayload
payloads.MACResponsePayload
)
def test_create_mac_verify_payload(self):

View File

@ -18,7 +18,7 @@ from testtools import TestCase
from kmip.core import utils
from kmip.core import attributes
from kmip.core.messages.payloads import activate
from kmip.core.messages import payloads
class TestActivateRequestPayload(TestCase):
@ -49,14 +49,14 @@ class TestActivateRequestPayload(TestCase):
Test that a ActivateRequestPayload object can be constructed with no
specified value.
"""
activate.ActivateRequestPayload()
payloads.ActivateRequestPayload()
def test_init_with_args(self):
"""
Test that a ActivateRequestPayload object can be constructed with valid
values.
"""
activate.ActivateRequestPayload(unique_identifier=self.uuid)
payloads.ActivateRequestPayload(unique_identifier=self.uuid)
def test_validate_with_bad_uuid_type(self):
"""
@ -65,14 +65,14 @@ class TestActivateRequestPayload(TestCase):
"""
self.assertRaisesRegexp(
TypeError, "invalid unique identifier",
activate.ActivateRequestPayload, "not-a-uuid")
payloads.ActivateRequestPayload, "not-a-uuid")
def test_read_with_known_uuid(self):
"""
Test that a ActivateRequestPayload object with known UUID can be read
from a data stream.
"""
payload = activate.ActivateRequestPayload()
payload = payloads.ActivateRequestPayload()
payload.read(self.encoding_a)
expected = '668eff89-3010-4258-bc0e-8c402309c746'
observed = payload.unique_identifier.value
@ -88,7 +88,7 @@ class TestActivateRequestPayload(TestCase):
written to a data stream.
"""
stream = utils.BytearrayStream()
payload = activate.ActivateRequestPayload(self.uuid)
payload = payloads.ActivateRequestPayload(self.uuid)
payload.write(stream)
length_expected = len(self.encoding_a)
@ -132,14 +132,14 @@ class TestActivateResponsePayload(TestCase):
Test that a ActivateResponsePayload object can be constructed with no
specified value.
"""
activate.ActivateResponsePayload()
payloads.ActivateResponsePayload()
def test_init_with_args(self):
"""
Test that a ActivateResponsePayload object can be constructed with
valid values.
"""
activate.ActivateResponsePayload(unique_identifier=self.uuid)
payloads.ActivateResponsePayload(unique_identifier=self.uuid)
def test_validate_with_invalid_uuid(self):
"""
@ -148,14 +148,14 @@ class TestActivateResponsePayload(TestCase):
"""
self.assertRaisesRegexp(
TypeError, "invalid unique identifier",
activate.ActivateResponsePayload, "not-a-uuid")
payloads.ActivateResponsePayload, "not-a-uuid")
def test_read_with_known_uuid(self):
"""
Test that a ActivateResponsePayload object with known UUID can be read
from a data stream.
"""
payload = activate.ActivateResponsePayload()
payload = payloads.ActivateResponsePayload()
payload.read(self.encoding_a)
expected = '668eff89-3010-4258-bc0e-8c402309c746'
observed = payload.unique_identifier.value
@ -171,7 +171,7 @@ class TestActivateResponsePayload(TestCase):
written to a data stream.
"""
stream = utils.BytearrayStream()
payload = activate.ActivateResponsePayload(self.uuid)
payload = payloads.ActivateResponsePayload(self.uuid)
payload.write(stream)
length_expected = len(self.encoding_a)

View File

@ -19,7 +19,7 @@ from kmip.core import attributes
from kmip.core import objects
from kmip.core import utils
from kmip.core.messages.payloads import create_key_pair
from kmip.core.messages import payloads
class TestCreateKeyPairRequestPayload(TestCase):
@ -44,10 +44,10 @@ class TestCreateKeyPairRequestPayload(TestCase):
super(TestCreateKeyPairRequestPayload, self).tearDown()
def test_init_with_none(self):
create_key_pair.CreateKeyPairRequestPayload()
payloads.CreateKeyPairRequestPayload()
def test_init_with_args(self):
create_key_pair.CreateKeyPairRequestPayload(
payloads.CreateKeyPairRequestPayload(
self.common_template_attribute,
self.private_key_template_attribute,
self.public_key_template_attribute)
@ -58,7 +58,7 @@ class TestCreateKeyPairRequestPayload(TestCase):
'public_key_template_attribute': None}
self.assertRaisesRegexp(
TypeError, "invalid common template attribute",
create_key_pair.CreateKeyPairRequestPayload, **kwargs)
payloads.CreateKeyPairRequestPayload, **kwargs)
def test_validate_with_invalid_private_key_template_attribute(self):
kwargs = {'common_template_attribute': None,
@ -66,7 +66,7 @@ class TestCreateKeyPairRequestPayload(TestCase):
'public_key_template_attribute': None}
self.assertRaisesRegexp(
TypeError, "invalid private key template attribute",
create_key_pair.CreateKeyPairRequestPayload, **kwargs)
payloads.CreateKeyPairRequestPayload, **kwargs)
def test_validate_with_invalid_public_key_template_attribute(self):
kwargs = {'common_template_attribute': None,
@ -74,7 +74,7 @@ class TestCreateKeyPairRequestPayload(TestCase):
'public_key_template_attribute': 'invalid'}
self.assertRaises(
TypeError, "invalid public key template attribute",
create_key_pair.CreateKeyPairRequestPayload, **kwargs)
payloads.CreateKeyPairRequestPayload, **kwargs)
def _test_read(self, stream, payload, common_template_attribute,
private_key_template_attribute,
@ -103,13 +103,13 @@ class TestCreateKeyPairRequestPayload(TestCase):
def test_read_with_none(self):
stream = self.encoding_empty
payload = create_key_pair.CreateKeyPairRequestPayload()
payload = payloads.CreateKeyPairRequestPayload()
self._test_read(stream, payload, None, None, None)
def test_read_with_args(self):
stream = self.encoding_full
payload = create_key_pair.CreateKeyPairRequestPayload()
payload = payloads.CreateKeyPairRequestPayload()
self._test_read(stream, payload, self.common_template_attribute,
self.private_key_template_attribute,
@ -133,13 +133,13 @@ class TestCreateKeyPairRequestPayload(TestCase):
def test_write_with_none(self):
stream = utils.BytearrayStream()
payload = create_key_pair.CreateKeyPairRequestPayload()
payload = payloads.CreateKeyPairRequestPayload()
self._test_write(stream, payload, self.encoding_empty)
def test_write_with_args(self):
stream = utils.BytearrayStream()
payload = create_key_pair.CreateKeyPairRequestPayload(
payload = payloads.CreateKeyPairRequestPayload(
self.common_template_attribute,
self.private_key_template_attribute,
self.public_key_template_attribute)
@ -182,10 +182,10 @@ class TestCreateKeyPairResponsePayload(TestCase):
super(TestCreateKeyPairResponsePayload, self).tearDown()
def test_init_with_none(self):
create_key_pair.CreateKeyPairResponsePayload()
payloads.CreateKeyPairResponsePayload()
def test_init_with_args(self):
create_key_pair.CreateKeyPairResponsePayload(
payloads.CreateKeyPairResponsePayload(
self.private_key_uuid, self.public_key_uuid,
self.private_key_template_attribute,
self.public_key_template_attribute)
@ -197,7 +197,7 @@ class TestCreateKeyPairResponsePayload(TestCase):
'public_key_template_attribute': None}
self.assertRaisesRegexp(
TypeError, "invalid private key unique identifier",
create_key_pair.CreateKeyPairResponsePayload, **kwargs)
payloads.CreateKeyPairResponsePayload, **kwargs)
def test_validate_with_invalid_public_key_unique_identifier(self):
kwargs = {'private_key_uuid': None,
@ -206,7 +206,7 @@ class TestCreateKeyPairResponsePayload(TestCase):
'public_key_template_attribute': None}
self.assertRaisesRegexp(
TypeError, "invalid public key unique identifier",
create_key_pair.CreateKeyPairResponsePayload, **kwargs)
payloads.CreateKeyPairResponsePayload, **kwargs)
def test_validate_with_invalid_private_key_template_attribute(self):
kwargs = {'private_key_uuid': self.private_key_uuid,
@ -215,7 +215,7 @@ class TestCreateKeyPairResponsePayload(TestCase):
'public_key_template_attribute': None}
self.assertRaisesRegexp(
TypeError, "invalid private key template attribute",
create_key_pair.CreateKeyPairResponsePayload, **kwargs)
payloads.CreateKeyPairResponsePayload, **kwargs)
def test_validate_with_invalid_public_key_template_attribute(self):
kwargs = {'private_key_uuid': self.private_key_uuid,
@ -224,7 +224,7 @@ class TestCreateKeyPairResponsePayload(TestCase):
'public_key_template_attribute': 'invalid'}
self.assertRaisesRegexp(
TypeError, "invalid public key template attribute",
create_key_pair.CreateKeyPairResponsePayload, **kwargs)
payloads.CreateKeyPairResponsePayload, **kwargs)
def _test_read(self, stream, payload, private_key_uuid, public_key_uuid,
private_key_template_attribute,
@ -257,14 +257,14 @@ class TestCreateKeyPairResponsePayload(TestCase):
def test_read_with_none(self):
stream = self.encoding_empty
payload = create_key_pair.CreateKeyPairResponsePayload()
payload = payloads.CreateKeyPairResponsePayload()
self._test_read(stream, payload, self.empty_private_key_uuid,
self.empty_public_key_uuid, None, None)
def test_read_with_args(self):
stream = self.encoding_full
payload = create_key_pair.CreateKeyPairResponsePayload(
payload = payloads.CreateKeyPairResponsePayload(
self.private_key_uuid, self.public_key_uuid,
self.private_key_template_attribute,
self.public_key_template_attribute)
@ -292,13 +292,13 @@ class TestCreateKeyPairResponsePayload(TestCase):
def test_write_with_none(self):
stream = utils.BytearrayStream()
payload = create_key_pair.CreateKeyPairResponsePayload()
payload = payloads.CreateKeyPairResponsePayload()
self._test_write(stream, payload, self.encoding_empty)
def test_write_with_args(self):
stream = utils.BytearrayStream()
payload = create_key_pair.CreateKeyPairResponsePayload(
payload = payloads.CreateKeyPairResponsePayload(
self.private_key_uuid, self.public_key_uuid,
self.private_key_template_attribute,
self.public_key_template_attribute)

View File

@ -19,7 +19,7 @@ from kmip.core import attributes
from kmip.core import enums
from kmip.core import utils
from kmip.core.messages.payloads import decrypt
from kmip.core.messages import payloads
class TestDecryptRequestPayload(testtools.TestCase):
@ -98,7 +98,7 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that a Decrypt request payload can be constructed with no
arguments.
"""
payload = decrypt.DecryptRequestPayload()
payload = payloads.DecryptRequestPayload()
self.assertEqual(None, payload.unique_identifier)
self.assertEqual(None, payload.cryptographic_parameters)
@ -110,7 +110,7 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that a Decrypt request payload can be constructed with valid
values
"""
payload = decrypt.DecryptRequestPayload(
payload = payloads.DecryptRequestPayload(
unique_identifier='00000000-1111-2222-3333-444444444444',
cryptographic_parameters=attributes.CryptographicParameters(),
data=b'\x01\x02\x03',
@ -133,7 +133,7 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that a TypeError is raised when an invalid value is used to set
the unique identifier of a Decrypt request payload.
"""
payload = decrypt.DecryptRequestPayload()
payload = payloads.DecryptRequestPayload()
args = (payload, 'unique_identifier', 0)
self.assertRaisesRegexp(
TypeError,
@ -147,7 +147,7 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that a TypeError is raised when an invalid value is used to set
the cryptographic parameters of a Decrypt request payload.
"""
payload = decrypt.DecryptRequestPayload()
payload = payloads.DecryptRequestPayload()
args = (payload, 'cryptographic_parameters', 'invalid')
self.assertRaisesRegexp(
TypeError,
@ -162,7 +162,7 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that a TypeError is raised when an invalid value is used to set
the data of a Decrypt request payload.
"""
payload = decrypt.DecryptRequestPayload()
payload = payloads.DecryptRequestPayload()
args = (payload, 'data', 0)
self.assertRaisesRegexp(
TypeError,
@ -176,7 +176,7 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that a TypeError is raised when an invalid value is used to set
the IV/counter/nonce of a Decrypt request payload.
"""
payload = decrypt.DecryptRequestPayload()
payload = payloads.DecryptRequestPayload()
args = (payload, 'iv_counter_nonce', 0)
self.assertRaisesRegexp(
TypeError,
@ -189,7 +189,7 @@ class TestDecryptRequestPayload(testtools.TestCase):
"""
Test that a Decrypt request payload can be read from a data stream.
"""
payload = decrypt.DecryptRequestPayload()
payload = payloads.DecryptRequestPayload()
self.assertEqual(None, payload.unique_identifier)
self.assertEqual(None, payload.cryptographic_parameters)
@ -251,7 +251,7 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that a Decrypt request payload can be read from a partial data
stream containing the minimum required attributes.
"""
payload = decrypt.DecryptRequestPayload()
payload = payloads.DecryptRequestPayload()
self.assertEqual(None, payload.unique_identifier)
self.assertEqual(None, payload.cryptographic_parameters)
@ -270,7 +270,7 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that a ValueError gets raised when a required Decrypt request
payload attribute is missing from the payload encoding.
"""
payload = decrypt.DecryptRequestPayload()
payload = payloads.DecryptRequestPayload()
args = (self.empty_encoding, )
self.assertRaisesRegexp(
ValueError,
@ -283,7 +283,7 @@ class TestDecryptRequestPayload(testtools.TestCase):
"""
Test that a Decrypt request payload can be written to a data stream.
"""
payload = decrypt.DecryptRequestPayload(
payload = payloads.DecryptRequestPayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
cryptographic_parameters=attributes.CryptographicParameters(
block_cipher_mode=enums.BlockCipherMode.CBC,
@ -315,7 +315,7 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that a partially defined Decrypt request payload can be written
to a data stream.
"""
payload = decrypt.DecryptRequestPayload(
payload = payloads.DecryptRequestPayload(
data=b'\x01\x02\x03\x04\x05\x06\x07\x08'
)
stream = utils.BytearrayStream()
@ -329,7 +329,7 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that a ValueError gets raised when a required Decrypt request
payload attribute is missing when encoding the payload.
"""
payload = decrypt.DecryptRequestPayload()
payload = payloads.DecryptRequestPayload()
stream = utils.BytearrayStream()
args = (stream, )
self.assertRaisesRegexp(
@ -344,13 +344,13 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that the equality operator returns True when comparing two
Decrypt request payloads with the same data.
"""
a = decrypt.DecryptRequestPayload()
b = decrypt.DecryptRequestPayload()
a = payloads.DecryptRequestPayload()
b = payloads.DecryptRequestPayload()
self.assertTrue(a == b)
self.assertTrue(b == a)
a = decrypt.DecryptRequestPayload(
a = payloads.DecryptRequestPayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
cryptographic_parameters=attributes.CryptographicParameters(
block_cipher_mode=enums.BlockCipherMode.CBC,
@ -371,7 +371,7 @@ class TestDecryptRequestPayload(testtools.TestCase):
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF',
iv_counter_nonce=b'\x01'
)
b = decrypt.DecryptRequestPayload(
b = payloads.DecryptRequestPayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
cryptographic_parameters=attributes.CryptographicParameters(
block_cipher_mode=enums.BlockCipherMode.CBC,
@ -401,10 +401,10 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
Decrypt request payloads with different unique identifiers.
"""
a = decrypt.DecryptRequestPayload(
a = payloads.DecryptRequestPayload(
unique_identifier='a'
)
b = decrypt.DecryptRequestPayload(
b = payloads.DecryptRequestPayload(
unique_identifier='b'
)
@ -416,12 +416,12 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
Decrypt request payloads with different cryptographic parameters.
"""
a = decrypt.DecryptRequestPayload(
a = payloads.DecryptRequestPayload(
cryptographic_parameters=attributes.CryptographicParameters(
block_cipher_mode=enums.BlockCipherMode.CBC
)
)
b = decrypt.DecryptRequestPayload(
b = payloads.DecryptRequestPayload(
cryptographic_parameters=attributes.CryptographicParameters(
hashing_algorithm=enums.HashingAlgorithm.MD5
)
@ -435,8 +435,8 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
Decrypt request payloads with different data.
"""
a = decrypt.DecryptRequestPayload(data=b'\x11')
b = decrypt.DecryptRequestPayload(data=b'\xFF')
a = payloads.DecryptRequestPayload(data=b'\x11')
b = payloads.DecryptRequestPayload(data=b'\xFF')
self.assertFalse(a == b)
self.assertFalse(b == a)
@ -446,8 +446,8 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
Decrypt request payloads with different IV/counter/nonce values.
"""
a = decrypt.DecryptRequestPayload(iv_counter_nonce=b'\x22')
b = decrypt.DecryptRequestPayload(iv_counter_nonce=b'\xAA')
a = payloads.DecryptRequestPayload(iv_counter_nonce=b'\x22')
b = payloads.DecryptRequestPayload(iv_counter_nonce=b'\xAA')
self.assertFalse(a == b)
self.assertFalse(b == a)
@ -457,7 +457,7 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
Decrypt request payloads with different types.
"""
a = decrypt.DecryptRequestPayload()
a = payloads.DecryptRequestPayload()
b = 'invalid'
self.assertFalse(a == b)
@ -468,13 +468,13 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that the inequality operator returns False when comparing two
Decrypt request payloads with the same data.
"""
a = decrypt.DecryptRequestPayload()
b = decrypt.DecryptRequestPayload()
a = payloads.DecryptRequestPayload()
b = payloads.DecryptRequestPayload()
self.assertFalse(a != b)
self.assertFalse(b != a)
a = decrypt.DecryptRequestPayload(
a = payloads.DecryptRequestPayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
cryptographic_parameters=attributes.CryptographicParameters(
block_cipher_mode=enums.BlockCipherMode.CBC,
@ -495,7 +495,7 @@ class TestDecryptRequestPayload(testtools.TestCase):
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF',
iv_counter_nonce=b'\x01'
)
b = decrypt.DecryptRequestPayload(
b = payloads.DecryptRequestPayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
cryptographic_parameters=attributes.CryptographicParameters(
block_cipher_mode=enums.BlockCipherMode.CBC,
@ -525,10 +525,10 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
Decrypt request payloads with different unique identifiers.
"""
a = decrypt.DecryptRequestPayload(
a = payloads.DecryptRequestPayload(
unique_identifier='a'
)
b = decrypt.DecryptRequestPayload(
b = payloads.DecryptRequestPayload(
unique_identifier='b'
)
@ -540,12 +540,12 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
Decrypt request payloads with different cryptographic parameters.
"""
a = decrypt.DecryptRequestPayload(
a = payloads.DecryptRequestPayload(
cryptographic_parameters=attributes.CryptographicParameters(
block_cipher_mode=enums.BlockCipherMode.CBC
)
)
b = decrypt.DecryptRequestPayload(
b = payloads.DecryptRequestPayload(
cryptographic_parameters=attributes.CryptographicParameters(
hashing_algorithm=enums.HashingAlgorithm.MD5
)
@ -559,8 +559,8 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
Decrypt request payloads with different data.
"""
a = decrypt.DecryptRequestPayload(data=b'\x11')
b = decrypt.DecryptRequestPayload(data=b'\xFF')
a = payloads.DecryptRequestPayload(data=b'\x11')
b = payloads.DecryptRequestPayload(data=b'\xFF')
self.assertTrue(a != b)
self.assertTrue(b != a)
@ -570,8 +570,8 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
Decrypt request payloads with different IV/counter/nonce values.
"""
a = decrypt.DecryptRequestPayload(iv_counter_nonce=b'\x22')
b = decrypt.DecryptRequestPayload(iv_counter_nonce=b'\xAA')
a = payloads.DecryptRequestPayload(iv_counter_nonce=b'\x22')
b = payloads.DecryptRequestPayload(iv_counter_nonce=b'\xAA')
self.assertTrue(a != b)
self.assertTrue(b != a)
@ -581,7 +581,7 @@ class TestDecryptRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
Decrypt request payloads with different types.
"""
a = decrypt.DecryptRequestPayload()
a = payloads.DecryptRequestPayload()
b = 'invalid'
self.assertTrue(a != b)
@ -591,7 +591,7 @@ class TestDecryptRequestPayload(testtools.TestCase):
"""
Test that repr can be applied to an Decrypt request payload.
"""
payload = decrypt.DecryptRequestPayload(
payload = payloads.DecryptRequestPayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
cryptographic_parameters=attributes.CryptographicParameters(
block_cipher_mode=enums.BlockCipherMode.CBC,
@ -657,7 +657,7 @@ class TestDecryptRequestPayload(testtools.TestCase):
counter_length=0,
initial_counter_value=1
)
payload = decrypt.DecryptRequestPayload(
payload = payloads.DecryptRequestPayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
cryptographic_parameters=cryptographic_parameters,
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF',
@ -724,7 +724,7 @@ class TestDecryptResponsePayload(testtools.TestCase):
Test that an Decrypt response payload can be constructed with no
arguments.
"""
payload = decrypt.DecryptResponsePayload()
payload = payloads.DecryptResponsePayload()
self.assertEqual(None, payload.unique_identifier)
self.assertEqual(None, payload.data)
@ -734,7 +734,7 @@ class TestDecryptResponsePayload(testtools.TestCase):
Test that a Decrypt response payload can be constructed with valid
values
"""
payload = decrypt.DecryptResponsePayload(
payload = payloads.DecryptResponsePayload(
unique_identifier='00000000-1111-2222-3333-444444444444',
data=b'\x01\x02\x03'
)
@ -750,7 +750,7 @@ class TestDecryptResponsePayload(testtools.TestCase):
Test that a TypeError is raised when an invalid value is used to set
the unique identifier of a Decrypt response payload.
"""
payload = decrypt.DecryptResponsePayload()
payload = payloads.DecryptResponsePayload()
args = (payload, 'unique_identifier', 0)
self.assertRaisesRegexp(
TypeError,
@ -764,7 +764,7 @@ class TestDecryptResponsePayload(testtools.TestCase):
Test that a TypeError is raised when an invalid value is used to set
the data of a Decrypt response payload.
"""
payload = decrypt.DecryptResponsePayload()
payload = payloads.DecryptResponsePayload()
args = (payload, 'data', 0)
self.assertRaisesRegexp(
TypeError,
@ -777,7 +777,7 @@ class TestDecryptResponsePayload(testtools.TestCase):
"""
Test that a Decrypt response payload can be read from a data stream.
"""
payload = decrypt.DecryptResponsePayload()
payload = payloads.DecryptResponsePayload()
self.assertEqual(None, payload.unique_identifier)
self.assertEqual(None, payload.data)
@ -795,7 +795,7 @@ class TestDecryptResponsePayload(testtools.TestCase):
Test that a ValueError gets raised when required Decrypt response
payload attributes are missing from the payload encoding.
"""
payload = decrypt.DecryptResponsePayload()
payload = payloads.DecryptResponsePayload()
args = (self.empty_encoding, )
self.assertRaisesRegexp(
ValueError,
@ -804,7 +804,7 @@ class TestDecryptResponsePayload(testtools.TestCase):
*args
)
payload = decrypt.DecryptResponsePayload()
payload = payloads.DecryptResponsePayload()
args = (self.incomplete_encoding, )
self.assertRaisesRegexp(
ValueError,
@ -817,7 +817,7 @@ class TestDecryptResponsePayload(testtools.TestCase):
"""
Test that a Decrypt response payload can be written to a data stream.
"""
payload = decrypt.DecryptResponsePayload(
payload = payloads.DecryptResponsePayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF'
)
@ -832,7 +832,7 @@ class TestDecryptResponsePayload(testtools.TestCase):
Test that a ValueError gets raised when required Decrypt response
payload attributes are missing when encoding the payload.
"""
payload = decrypt.DecryptResponsePayload()
payload = payloads.DecryptResponsePayload()
self.assertIsNone(payload.unique_identifier)
stream = utils.BytearrayStream()
args = (stream, )
@ -843,7 +843,7 @@ class TestDecryptResponsePayload(testtools.TestCase):
*args
)
payload = decrypt.DecryptResponsePayload(
payload = payloads.DecryptResponsePayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959'
)
stream = utils.BytearrayStream()
@ -860,17 +860,17 @@ class TestDecryptResponsePayload(testtools.TestCase):
Test that the equality operator returns True when comparing two
Decrypt response payloads with the same data.
"""
a = decrypt.DecryptResponsePayload()
b = decrypt.DecryptResponsePayload()
a = payloads.DecryptResponsePayload()
b = payloads.DecryptResponsePayload()
self.assertTrue(a == b)
self.assertTrue(b == a)
a = decrypt.DecryptResponsePayload(
a = payloads.DecryptResponsePayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF'
)
b = decrypt.DecryptResponsePayload(
b = payloads.DecryptResponsePayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF'
)
@ -883,10 +883,10 @@ class TestDecryptResponsePayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
Decrypt response payloads with different unique identifiers.
"""
a = decrypt.DecryptResponsePayload(
a = payloads.DecryptResponsePayload(
unique_identifier='a'
)
b = decrypt.DecryptResponsePayload(
b = payloads.DecryptResponsePayload(
unique_identifier='b'
)
@ -898,8 +898,8 @@ class TestDecryptResponsePayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
Decrypt response payloads with different data.
"""
a = decrypt.DecryptResponsePayload(data=b'\x11')
b = decrypt.DecryptResponsePayload(data=b'\xFF')
a = payloads.DecryptResponsePayload(data=b'\x11')
b = payloads.DecryptResponsePayload(data=b'\xFF')
self.assertFalse(a == b)
self.assertFalse(b == a)
@ -909,7 +909,7 @@ class TestDecryptResponsePayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
Decrypt response payloads with different types.
"""
a = decrypt.DecryptResponsePayload()
a = payloads.DecryptResponsePayload()
b = 'invalid'
self.assertFalse(a == b)
@ -920,17 +920,17 @@ class TestDecryptResponsePayload(testtools.TestCase):
Test that the inequality operator returns False when comparing two
Decrypt response payloads with the same data.
"""
a = decrypt.DecryptResponsePayload()
b = decrypt.DecryptResponsePayload()
a = payloads.DecryptResponsePayload()
b = payloads.DecryptResponsePayload()
self.assertFalse(a != b)
self.assertFalse(b != a)
a = decrypt.DecryptResponsePayload(
a = payloads.DecryptResponsePayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF'
)
b = decrypt.DecryptResponsePayload(
b = payloads.DecryptResponsePayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF'
)
@ -943,10 +943,10 @@ class TestDecryptResponsePayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
Decrypt response payloads with different unique identifiers.
"""
a = decrypt.DecryptResponsePayload(
a = payloads.DecryptResponsePayload(
unique_identifier='a'
)
b = decrypt.DecryptResponsePayload(
b = payloads.DecryptResponsePayload(
unique_identifier='b'
)
@ -958,8 +958,8 @@ class TestDecryptResponsePayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
Decrypt response payloads with different data.
"""
a = decrypt.DecryptResponsePayload(data=b'\x11')
b = decrypt.DecryptResponsePayload(data=b'\xFF')
a = payloads.DecryptResponsePayload(data=b'\x11')
b = payloads.DecryptResponsePayload(data=b'\xFF')
self.assertTrue(a != b)
self.assertTrue(b != a)
@ -969,7 +969,7 @@ class TestDecryptResponsePayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
Decrypt response payloads with different types.
"""
a = decrypt.DecryptResponsePayload()
a = payloads.DecryptResponsePayload()
b = 'invalid'
self.assertTrue(a != b)
@ -979,7 +979,7 @@ class TestDecryptResponsePayload(testtools.TestCase):
"""
Test that repr can be applied to a Decrypt response payload.
"""
payload = decrypt.DecryptResponsePayload(
payload = payloads.DecryptResponsePayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF'
)
@ -996,7 +996,7 @@ class TestDecryptResponsePayload(testtools.TestCase):
"""
Test that str can be applied to a Decrypt response payload
"""
payload = decrypt.DecryptResponsePayload(
payload = payloads.DecryptResponsePayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF'
)

View File

@ -21,7 +21,7 @@ from kmip.core import objects
from kmip.core import primitives
from kmip.core import utils
from kmip.core.messages.payloads import derive_key
from kmip.core.messages import payloads
class TestDeriveKeyRequestPayload(testtools.TestCase):
@ -150,7 +150,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that a DeriveKey request payload can be constructed with no
arguments.
"""
payload = derive_key.DeriveKeyRequestPayload()
payload = payloads.DeriveKeyRequestPayload()
self.assertEqual(None, payload.object_type)
self.assertEqual(None, payload.unique_identifiers)
@ -163,7 +163,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that a DeriveKey request payload can be constructed with valid
values
"""
payload = derive_key.DeriveKeyRequestPayload(
payload = payloads.DeriveKeyRequestPayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifiers=['00000000-1111-2222-3333-444444444444'],
derivation_method=enums.DerivationMethod.HASH,
@ -197,7 +197,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that a TypeError is raised when an invalid value is used to set
the object type of a DeriveKey request payload.
"""
payload = derive_key.DeriveKeyRequestPayload()
payload = payloads.DeriveKeyRequestPayload()
args = (payload, 'object_type', 'invalid')
self.assertRaisesRegexp(
TypeError,
@ -211,7 +211,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that a TypeError is raised when invalid values are used to set
the unique identifiers of a DeriveKey request payload.
"""
payload = derive_key.DeriveKeyRequestPayload()
payload = payloads.DeriveKeyRequestPayload()
args = (payload, 'unique_identifiers', 'invalid')
self.assertRaisesRegexp(
TypeError,
@ -241,7 +241,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that a TypeError is raised when an invalid value is used to set
the derivation method of a DeriveKey request payload.
"""
payload = derive_key.DeriveKeyRequestPayload()
payload = payloads.DeriveKeyRequestPayload()
args = (payload, 'derivation_method', 'invalid')
self.assertRaisesRegexp(
TypeError,
@ -255,7 +255,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that a TypeError is raised when an invalid value is used to set
the derivation parameters of a DeriveKey request payload.
"""
payload = derive_key.DeriveKeyRequestPayload()
payload = payloads.DeriveKeyRequestPayload()
args = (payload, 'derivation_parameters', 'invalid')
self.assertRaisesRegexp(
TypeError,
@ -269,7 +269,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that a TypeError is raised when an invalid value is used to set
the template attribute of a DeriveKey request payload.
"""
payload = derive_key.DeriveKeyRequestPayload()
payload = payloads.DeriveKeyRequestPayload()
args = (payload, 'template_attribute', 'invalid')
self.assertRaisesRegexp(
TypeError,
@ -282,7 +282,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
"""
Test that a DeriveKey request payload can be read from a data stream.
"""
payload = derive_key.DeriveKeyRequestPayload()
payload = payloads.DeriveKeyRequestPayload()
self.assertEqual(None, payload.object_type)
self.assertEqual(None, payload.unique_identifiers)
@ -347,7 +347,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that a ValueError gets raised when decoding a DeriveKey request
payload encoding missing the object type.
"""
payload = derive_key.DeriveKeyRequestPayload()
payload = payloads.DeriveKeyRequestPayload()
self.assertEqual(None, payload.object_type)
self.assertEqual(None, payload.unique_identifiers)
@ -368,7 +368,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that a ValueError gets raised when decoding a DeriveKey request
payload encoding missing the unique identifiers.
"""
payload = derive_key.DeriveKeyRequestPayload()
payload = payloads.DeriveKeyRequestPayload()
self.assertEqual(None, payload.object_type)
self.assertEqual(None, payload.unique_identifiers)
@ -389,7 +389,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that a ValueError gets raised when decoding a DeriveKey request
payload encoding missing the derivation method.
"""
payload = derive_key.DeriveKeyRequestPayload()
payload = payloads.DeriveKeyRequestPayload()
self.assertEqual(None, payload.object_type)
self.assertEqual(None, payload.unique_identifiers)
@ -410,7 +410,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that a ValueError gets raised when decoding a DeriveKey request
payload encoding missing the derivation parameters.
"""
payload = derive_key.DeriveKeyRequestPayload()
payload = payloads.DeriveKeyRequestPayload()
self.assertEqual(None, payload.object_type)
self.assertEqual(None, payload.unique_identifiers)
@ -431,7 +431,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that a ValueError gets raised when decoding a DeriveKey request
payload encoding missing the template attribute.
"""
payload = derive_key.DeriveKeyRequestPayload()
payload = payloads.DeriveKeyRequestPayload()
self.assertEqual(None, payload.object_type)
self.assertEqual(None, payload.unique_identifiers)
@ -451,7 +451,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
"""
Test that a DeriveKey request payload can be written to a data stream.
"""
payload = derive_key.DeriveKeyRequestPayload(
payload = payloads.DeriveKeyRequestPayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifiers=[
'fb4b5b9c-6188-4c63-8142-fe9c328129fc',
@ -501,7 +501,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that a ValueError gets raised when encoding a DeriveKey request
payload missing the object type.
"""
payload = derive_key.DeriveKeyRequestPayload()
payload = payloads.DeriveKeyRequestPayload()
args = (utils.BytearrayStream(), )
self.assertRaisesRegexp(
@ -516,7 +516,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that a ValueError gets raised when encoding a DeriveKey request
payload missing the unique identifiers.
"""
payload = derive_key.DeriveKeyRequestPayload(
payload = payloads.DeriveKeyRequestPayload(
object_type=enums.ObjectType.SYMMETRIC_KEY
)
@ -533,7 +533,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that a ValueError gets raised when encoding a DeriveKey request
payload missing the derivation method.
"""
payload = derive_key.DeriveKeyRequestPayload(
payload = payloads.DeriveKeyRequestPayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifiers=[
'fb4b5b9c-6188-4c63-8142-fe9c328129fc',
@ -555,7 +555,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that a ValueError gets raised when encoding a DeriveKey request
payload missing the derivation parameters.
"""
payload = derive_key.DeriveKeyRequestPayload(
payload = payloads.DeriveKeyRequestPayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifiers=[
'fb4b5b9c-6188-4c63-8142-fe9c328129fc',
@ -578,7 +578,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that a ValueError gets raised when encoding a DeriveKey request
payload missing the template attribute.
"""
payload = derive_key.DeriveKeyRequestPayload(
payload = payloads.DeriveKeyRequestPayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifiers=[
'fb4b5b9c-6188-4c63-8142-fe9c328129fc',
@ -608,13 +608,13 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that the equality operator returns True when comparing two
DeriveKey request payloads with the same data.
"""
a = derive_key.DeriveKeyRequestPayload()
b = derive_key.DeriveKeyRequestPayload()
a = payloads.DeriveKeyRequestPayload()
b = payloads.DeriveKeyRequestPayload()
self.assertTrue(a == b)
self.assertTrue(b == a)
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifiers=[
'fb4b5b9c-6188-4c63-8142-fe9c328129fc',
@ -653,7 +653,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
]
)
)
b = derive_key.DeriveKeyRequestPayload(
b = payloads.DeriveKeyRequestPayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifiers=[
'fb4b5b9c-6188-4c63-8142-fe9c328129fc',
@ -701,10 +701,10 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
DeriveKey request payloads with different object types.
"""
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
object_type=enums.ObjectType.SYMMETRIC_KEY
)
b = derive_key.DeriveKeyRequestPayload(
b = payloads.DeriveKeyRequestPayload(
object_type=enums.ObjectType.SECRET_DATA
)
@ -716,24 +716,24 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
DeriveKey request payloads with different sets of unique identifiers.
"""
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
unique_identifiers=['fb4b5b9c-6188-4c63-8142-fe9c328129fc']
)
b = derive_key.DeriveKeyRequestPayload(
b = payloads.DeriveKeyRequestPayload(
unique_identifiers=['5c9b81ef-4ee5-42cd-ba2d-c002fdd0c7b3']
)
self.assertFalse(a == b)
self.assertFalse(b == a)
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
unique_identifiers=[
'fb4b5b9c-6188-4c63-8142-fe9c328129fc',
'5c9b81ef-4ee5-42cd-ba2d-c002fdd0c7b3',
'1703250b-4d40-4de2-93a0-c494a1d4ae40'
]
)
b = derive_key.DeriveKeyRequestPayload(
b = payloads.DeriveKeyRequestPayload(
unique_identifiers=[
'1703250b-4d40-4de2-93a0-c494a1d4ae40',
'5c9b81ef-4ee5-42cd-ba2d-c002fdd0c7b3',
@ -744,14 +744,14 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
self.assertFalse(a == b)
self.assertFalse(b == a)
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
unique_identifiers=[
'fb4b5b9c-6188-4c63-8142-fe9c328129fc',
'5c9b81ef-4ee5-42cd-ba2d-c002fdd0c7b3',
'1703250b-4d40-4de2-93a0-c494a1d4ae40'
]
)
b = derive_key.DeriveKeyRequestPayload(unique_identifiers=[])
b = payloads.DeriveKeyRequestPayload(unique_identifiers=[])
self.assertFalse(a == b)
self.assertFalse(b == a)
@ -761,10 +761,10 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
DeriveKey request payloads with different derivation methods.
"""
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
derivation_method=enums.DerivationMethod.HASH
)
b = derive_key.DeriveKeyRequestPayload(
b = payloads.DeriveKeyRequestPayload(
derivation_method=enums.DerivationMethod.PBKDF2
)
@ -776,7 +776,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
DeriveKey request payloads with different derivation parameters.
"""
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
derivation_parameters=attributes.DerivationParameters(
cryptographic_parameters=attributes.CryptographicParameters(
hashing_algorithm=enums.HashingAlgorithm.SHA_256
@ -785,7 +785,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
derivation_data=b'\xFA\xD9\x8B\x6A\xCA\x6D\x87\xDD'
)
)
b = derive_key.DeriveKeyRequestPayload(
b = payloads.DeriveKeyRequestPayload(
derivation_parameters=attributes.DerivationParameters(
cryptographic_parameters=attributes.CryptographicParameters(
hashing_algorithm=enums.HashingAlgorithm.SHA_1
@ -798,7 +798,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
self.assertFalse(a == b)
self.assertFalse(b == a)
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
derivation_parameters=attributes.DerivationParameters(
cryptographic_parameters=attributes.CryptographicParameters(
hashing_algorithm=enums.HashingAlgorithm.SHA_256
@ -807,15 +807,15 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
derivation_data=b'\xFA\xD9\x8B\x6A\xCA\x6D\x87\xDD'
)
)
b = derive_key.DeriveKeyRequestPayload(
b = payloads.DeriveKeyRequestPayload(
derivation_parameters=attributes.DerivationParameters()
)
self.assertFalse(a == b)
self.assertFalse(b == a)
a = derive_key.DeriveKeyRequestPayload(derivation_parameters=None)
b = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(derivation_parameters=None)
b = payloads.DeriveKeyRequestPayload(
derivation_parameters=attributes.DerivationParameters(
cryptographic_parameters=attributes.CryptographicParameters(
hashing_algorithm=enums.HashingAlgorithm.SHA_256
@ -833,7 +833,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
DeriveKey request payloads with different template attributes.
"""
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
template_attribute=objects.TemplateAttribute(
attributes=[
objects.Attribute(
@ -858,7 +858,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
]
)
)
b = derive_key.DeriveKeyRequestPayload(
b = payloads.DeriveKeyRequestPayload(
template_attribute=objects.TemplateAttribute(
attributes=[
objects.Attribute(
@ -887,7 +887,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
self.assertFalse(a == b)
self.assertFalse(b == a)
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
template_attribute=objects.TemplateAttribute(
attributes=[
objects.Attribute(
@ -912,15 +912,15 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
]
)
)
b = derive_key.DeriveKeyRequestPayload(
b = payloads.DeriveKeyRequestPayload(
template_attribute=objects.TemplateAttribute()
)
self.assertFalse(a == b)
self.assertFalse(b == a)
a = derive_key.DeriveKeyRequestPayload(template_attribute=None)
b = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(template_attribute=None)
b = payloads.DeriveKeyRequestPayload(
template_attribute=objects.TemplateAttribute(
attributes=[
objects.Attribute(
@ -954,7 +954,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
DeriveKey request payloads with different types.
"""
a = derive_key.DeriveKeyRequestPayload()
a = payloads.DeriveKeyRequestPayload()
b = 'invalid'
self.assertFalse(a == b)
@ -965,13 +965,13 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that the inequality operator returns False when comparing two
DeriveKey request payloads with the same data.
"""
a = derive_key.DeriveKeyRequestPayload()
b = derive_key.DeriveKeyRequestPayload()
a = payloads.DeriveKeyRequestPayload()
b = payloads.DeriveKeyRequestPayload()
self.assertFalse(a != b)
self.assertFalse(b != a)
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifiers=[
'fb4b5b9c-6188-4c63-8142-fe9c328129fc',
@ -1010,7 +1010,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
]
)
)
b = derive_key.DeriveKeyRequestPayload(
b = payloads.DeriveKeyRequestPayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifiers=[
'fb4b5b9c-6188-4c63-8142-fe9c328129fc',
@ -1058,10 +1058,10 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
DeriveKey request payloads with different object types.
"""
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
object_type=enums.ObjectType.SYMMETRIC_KEY
)
b = derive_key.DeriveKeyRequestPayload(
b = payloads.DeriveKeyRequestPayload(
object_type=enums.ObjectType.SECRET_DATA
)
@ -1073,24 +1073,24 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
DeriveKey request payloads with different sets of unique identifiers.
"""
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
unique_identifiers=['fb4b5b9c-6188-4c63-8142-fe9c328129fc']
)
b = derive_key.DeriveKeyRequestPayload(
b = payloads.DeriveKeyRequestPayload(
unique_identifiers=['5c9b81ef-4ee5-42cd-ba2d-c002fdd0c7b3']
)
self.assertTrue(a != b)
self.assertTrue(b != a)
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
unique_identifiers=[
'fb4b5b9c-6188-4c63-8142-fe9c328129fc',
'5c9b81ef-4ee5-42cd-ba2d-c002fdd0c7b3',
'1703250b-4d40-4de2-93a0-c494a1d4ae40'
]
)
b = derive_key.DeriveKeyRequestPayload(
b = payloads.DeriveKeyRequestPayload(
unique_identifiers=[
'1703250b-4d40-4de2-93a0-c494a1d4ae40',
'5c9b81ef-4ee5-42cd-ba2d-c002fdd0c7b3',
@ -1101,14 +1101,14 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
self.assertTrue(a != b)
self.assertTrue(b != a)
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
unique_identifiers=[
'fb4b5b9c-6188-4c63-8142-fe9c328129fc',
'5c9b81ef-4ee5-42cd-ba2d-c002fdd0c7b3',
'1703250b-4d40-4de2-93a0-c494a1d4ae40'
]
)
b = derive_key.DeriveKeyRequestPayload(unique_identifiers=[])
b = payloads.DeriveKeyRequestPayload(unique_identifiers=[])
self.assertTrue(a != b)
self.assertTrue(b != a)
@ -1118,10 +1118,10 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
DeriveKey request payloads with different derivation methods.
"""
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
derivation_method=enums.DerivationMethod.HASH
)
b = derive_key.DeriveKeyRequestPayload(
b = payloads.DeriveKeyRequestPayload(
derivation_method=enums.DerivationMethod.PBKDF2
)
@ -1133,7 +1133,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
DeriveKey request payloads with different derivation parameters.
"""
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
derivation_parameters=attributes.DerivationParameters(
cryptographic_parameters=attributes.CryptographicParameters(
hashing_algorithm=enums.HashingAlgorithm.SHA_256
@ -1142,7 +1142,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
derivation_data=b'\xFA\xD9\x8B\x6A\xCA\x6D\x87\xDD'
)
)
b = derive_key.DeriveKeyRequestPayload(
b = payloads.DeriveKeyRequestPayload(
derivation_parameters=attributes.DerivationParameters(
cryptographic_parameters=attributes.CryptographicParameters(
hashing_algorithm=enums.HashingAlgorithm.SHA_1
@ -1155,7 +1155,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
self.assertTrue(a != b)
self.assertTrue(b != a)
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
derivation_parameters=attributes.DerivationParameters(
cryptographic_parameters=attributes.CryptographicParameters(
hashing_algorithm=enums.HashingAlgorithm.SHA_256
@ -1164,15 +1164,15 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
derivation_data=b'\xFA\xD9\x8B\x6A\xCA\x6D\x87\xDD'
)
)
b = derive_key.DeriveKeyRequestPayload(
b = payloads.DeriveKeyRequestPayload(
derivation_parameters=attributes.DerivationParameters()
)
self.assertTrue(a != b)
self.assertTrue(b != a)
a = derive_key.DeriveKeyRequestPayload(derivation_parameters=None)
b = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(derivation_parameters=None)
b = payloads.DeriveKeyRequestPayload(
derivation_parameters=attributes.DerivationParameters(
cryptographic_parameters=attributes.CryptographicParameters(
hashing_algorithm=enums.HashingAlgorithm.SHA_256
@ -1190,7 +1190,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
DeriveKey request payloads with different template attribute.
"""
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
template_attribute=objects.TemplateAttribute(
attributes=[
objects.Attribute(
@ -1215,7 +1215,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
]
)
)
b = derive_key.DeriveKeyRequestPayload(
b = payloads.DeriveKeyRequestPayload(
template_attribute=objects.TemplateAttribute(
attributes=[
objects.Attribute(
@ -1244,7 +1244,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
self.assertTrue(a != b)
self.assertTrue(b != a)
a = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(
template_attribute=objects.TemplateAttribute(
attributes=[
objects.Attribute(
@ -1269,15 +1269,15 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
]
)
)
b = derive_key.DeriveKeyRequestPayload(
b = payloads.DeriveKeyRequestPayload(
template_attribute=objects.TemplateAttribute()
)
self.assertTrue(a != b)
self.assertTrue(b != a)
a = derive_key.DeriveKeyRequestPayload(template_attribute=None)
b = derive_key.DeriveKeyRequestPayload(
a = payloads.DeriveKeyRequestPayload(template_attribute=None)
b = payloads.DeriveKeyRequestPayload(
template_attribute=objects.TemplateAttribute(
attributes=[
objects.Attribute(
@ -1311,7 +1311,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
DeriveKey request payloads with different types.
"""
a = derive_key.DeriveKeyRequestPayload()
a = payloads.DeriveKeyRequestPayload()
b = 'invalid'
self.assertTrue(a != b)
@ -1351,7 +1351,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
)
]
)
payload = derive_key.DeriveKeyRequestPayload(
payload = payloads.DeriveKeyRequestPayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifiers=[
'fb4b5b9c-6188-4c63-8142-fe9c328129fc',
@ -1417,7 +1417,7 @@ class TestDeriveKeyRequestPayload(testtools.TestCase):
)
]
)
payload = derive_key.DeriveKeyRequestPayload(
payload = payloads.DeriveKeyRequestPayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifiers=[
'fb4b5b9c-6188-4c63-8142-fe9c328129fc',
@ -1506,7 +1506,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
Test that a DeriveKey response payload can be constructed with no
arguments.
"""
payload = derive_key.DeriveKeyResponsePayload()
payload = payloads.DeriveKeyResponsePayload()
self.assertEqual(None, payload.unique_identifier)
self.assertEqual(None, payload.template_attribute)
@ -1516,7 +1516,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
Test that a DeriveKey response payload can be constructed with valid
values
"""
payload = derive_key.DeriveKeyResponsePayload(
payload = payloads.DeriveKeyResponsePayload(
unique_identifier='00000000-1111-2222-3333-444444444444',
template_attribute=objects.TemplateAttribute()
)
@ -1535,7 +1535,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
Test that a TypeError is raised when invalid values are used to set
the unique identifier of a DeriveKey request payload.
"""
payload = derive_key.DeriveKeyResponsePayload()
payload = payloads.DeriveKeyResponsePayload()
args = (payload, 'unique_identifier', 0)
self.assertRaisesRegexp(
TypeError,
@ -1549,7 +1549,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
Test that a TypeError is raised when an invalid value is used to set
the template attribute of a DeriveKey response payload.
"""
payload = derive_key.DeriveKeyResponsePayload()
payload = payloads.DeriveKeyResponsePayload()
args = (payload, 'template_attribute', 'invalid')
self.assertRaisesRegexp(
TypeError,
@ -1562,7 +1562,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
"""
Test that a DeriveKey response payload can be read from a data stream.
"""
payload = derive_key.DeriveKeyResponsePayload()
payload = payloads.DeriveKeyResponsePayload()
self.assertEqual(None, payload.unique_identifier)
self.assertEqual(None, payload.template_attribute)
@ -1605,7 +1605,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
Test that a ValueError gets raised when decoding a DeriveKey response
payload encoding missing the unique identifier.
"""
payload = derive_key.DeriveKeyResponsePayload()
payload = payloads.DeriveKeyResponsePayload()
self.assertEqual(None, payload.unique_identifier)
self.assertEqual(None, payload.template_attribute)
@ -1624,7 +1624,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
Test that a DeriveKey response payload missing a template attribute
can be read from a data stream.
"""
payload = derive_key.DeriveKeyResponsePayload()
payload = payloads.DeriveKeyResponsePayload()
self.assertEqual(None, payload.unique_identifier)
self.assertEqual(None, payload.template_attribute)
@ -1641,7 +1641,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
"""
Test that a DeriveKey response payload can be written to a data stream.
"""
payload = derive_key.DeriveKeyResponsePayload(
payload = payloads.DeriveKeyResponsePayload(
unique_identifier='fb4b5b9c-6188-4c63-8142-fe9c328129fc',
template_attribute=objects.TemplateAttribute(
attributes=[
@ -1678,7 +1678,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
Test that a ValueError gets raised when encoding a DeriveKey response
payload missing the unique identifier.
"""
payload = derive_key.DeriveKeyResponsePayload()
payload = payloads.DeriveKeyResponsePayload()
args = (utils.BytearrayStream(), )
self.assertRaisesRegexp(
@ -1693,7 +1693,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
Test that a ValueError gets raised when encoding a DeriveKey response
payload missing the template attribute.
"""
payload = derive_key.DeriveKeyResponsePayload(
payload = payloads.DeriveKeyResponsePayload(
unique_identifier='fb4b5b9c-6188-4c63-8142-fe9c328129fc'
)
stream = utils.BytearrayStream()
@ -1714,13 +1714,13 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
Test that the equality operator returns True when comparing two
DeriveKey response payloads with the same data.
"""
a = derive_key.DeriveKeyResponsePayload()
b = derive_key.DeriveKeyResponsePayload()
a = payloads.DeriveKeyResponsePayload()
b = payloads.DeriveKeyResponsePayload()
self.assertTrue(a == b)
self.assertTrue(b == a)
a = derive_key.DeriveKeyResponsePayload(
a = payloads.DeriveKeyResponsePayload(
unique_identifier='fb4b5b9c-6188-4c63-8142-fe9c328129fc',
template_attribute=objects.TemplateAttribute(
attributes=[
@ -1746,7 +1746,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
]
)
)
b = derive_key.DeriveKeyResponsePayload(
b = payloads.DeriveKeyResponsePayload(
unique_identifier='fb4b5b9c-6188-4c63-8142-fe9c328129fc',
template_attribute=objects.TemplateAttribute(
attributes=[
@ -1781,20 +1781,20 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
DeriveKey response payloads with different unique identifiers.
"""
a = derive_key.DeriveKeyResponsePayload(
a = payloads.DeriveKeyResponsePayload(
unique_identifier='fb4b5b9c-6188-4c63-8142-fe9c328129fc'
)
b = derive_key.DeriveKeyResponsePayload(
b = payloads.DeriveKeyResponsePayload(
unique_identifier='5c9b81ef-4ee5-42cd-ba2d-c002fdd0c7b3'
)
self.assertFalse(a == b)
self.assertFalse(b == a)
a = derive_key.DeriveKeyResponsePayload(
a = payloads.DeriveKeyResponsePayload(
unique_identifier='1703250b-4d40-4de2-93a0-c494a1d4ae40'
)
b = derive_key.DeriveKeyResponsePayload(unique_identifier=None)
b = payloads.DeriveKeyResponsePayload(unique_identifier=None)
self.assertFalse(a == b)
self.assertFalse(b == a)
@ -1804,7 +1804,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
DeriveKey response payloads with different template attributes.
"""
a = derive_key.DeriveKeyResponsePayload(
a = payloads.DeriveKeyResponsePayload(
template_attribute=objects.TemplateAttribute(
attributes=[
objects.Attribute(
@ -1829,7 +1829,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
]
)
)
b = derive_key.DeriveKeyResponsePayload(
b = payloads.DeriveKeyResponsePayload(
template_attribute=objects.TemplateAttribute(
attributes=[
objects.Attribute(
@ -1858,7 +1858,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
self.assertFalse(a == b)
self.assertFalse(b == a)
a = derive_key.DeriveKeyResponsePayload(
a = payloads.DeriveKeyResponsePayload(
template_attribute=objects.TemplateAttribute(
attributes=[
objects.Attribute(
@ -1883,15 +1883,15 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
]
)
)
b = derive_key.DeriveKeyResponsePayload(
b = payloads.DeriveKeyResponsePayload(
template_attribute=objects.TemplateAttribute()
)
self.assertFalse(a == b)
self.assertFalse(b == a)
a = derive_key.DeriveKeyResponsePayload(template_attribute=None)
b = derive_key.DeriveKeyResponsePayload(
a = payloads.DeriveKeyResponsePayload(template_attribute=None)
b = payloads.DeriveKeyResponsePayload(
template_attribute=objects.TemplateAttribute(
attributes=[
objects.Attribute(
@ -1925,7 +1925,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
DeriveKey response payloads with different types.
"""
a = derive_key.DeriveKeyResponsePayload()
a = payloads.DeriveKeyResponsePayload()
b = 'invalid'
self.assertFalse(a == b)
@ -1936,13 +1936,13 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
Test that the inequality operator returns False when comparing two
DeriveKey response payloads with the same data.
"""
a = derive_key.DeriveKeyResponsePayload()
b = derive_key.DeriveKeyResponsePayload()
a = payloads.DeriveKeyResponsePayload()
b = payloads.DeriveKeyResponsePayload()
self.assertFalse(a != b)
self.assertFalse(b != a)
a = derive_key.DeriveKeyResponsePayload(
a = payloads.DeriveKeyResponsePayload(
unique_identifier='fb4b5b9c-6188-4c63-8142-fe9c328129fc',
template_attribute=objects.TemplateAttribute(
attributes=[
@ -1968,7 +1968,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
]
)
)
b = derive_key.DeriveKeyResponsePayload(
b = payloads.DeriveKeyResponsePayload(
unique_identifier='fb4b5b9c-6188-4c63-8142-fe9c328129fc',
template_attribute=objects.TemplateAttribute(
attributes=[
@ -2006,20 +2006,20 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
DeriveKey response payloads with different unique identifiers.
"""
a = derive_key.DeriveKeyResponsePayload(
a = payloads.DeriveKeyResponsePayload(
unique_identifier='fb4b5b9c-6188-4c63-8142-fe9c328129fc'
)
b = derive_key.DeriveKeyResponsePayload(
b = payloads.DeriveKeyResponsePayload(
unique_identifier='5c9b81ef-4ee5-42cd-ba2d-c002fdd0c7b3'
)
self.assertTrue(a != b)
self.assertTrue(b != a)
a = derive_key.DeriveKeyResponsePayload(
a = payloads.DeriveKeyResponsePayload(
unique_identifier='1703250b-4d40-4de2-93a0-c494a1d4ae40'
)
b = derive_key.DeriveKeyResponsePayload(unique_identifier=None)
b = payloads.DeriveKeyResponsePayload(unique_identifier=None)
self.assertTrue(a != b)
self.assertTrue(b != a)
@ -2029,7 +2029,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
DeriveKey response payloads with different template attribute.
"""
a = derive_key.DeriveKeyResponsePayload(
a = payloads.DeriveKeyResponsePayload(
template_attribute=objects.TemplateAttribute(
attributes=[
objects.Attribute(
@ -2054,7 +2054,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
]
)
)
b = derive_key.DeriveKeyResponsePayload(
b = payloads.DeriveKeyResponsePayload(
template_attribute=objects.TemplateAttribute(
attributes=[
objects.Attribute(
@ -2083,7 +2083,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
self.assertTrue(a != b)
self.assertTrue(b != a)
a = derive_key.DeriveKeyResponsePayload(
a = payloads.DeriveKeyResponsePayload(
template_attribute=objects.TemplateAttribute(
attributes=[
objects.Attribute(
@ -2108,15 +2108,15 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
]
)
)
b = derive_key.DeriveKeyResponsePayload(
b = payloads.DeriveKeyResponsePayload(
template_attribute=objects.TemplateAttribute()
)
self.assertTrue(a != b)
self.assertTrue(b != a)
a = derive_key.DeriveKeyResponsePayload(template_attribute=None)
b = derive_key.DeriveKeyResponsePayload(
a = payloads.DeriveKeyResponsePayload(template_attribute=None)
b = payloads.DeriveKeyResponsePayload(
template_attribute=objects.TemplateAttribute(
attributes=[
objects.Attribute(
@ -2150,7 +2150,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
DeriveKey response payloads with different types.
"""
a = derive_key.DeriveKeyResponsePayload()
a = payloads.DeriveKeyResponsePayload()
b = 'invalid'
self.assertTrue(a != b)
@ -2183,7 +2183,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
)
]
)
payload = derive_key.DeriveKeyResponsePayload(
payload = payloads.DeriveKeyResponsePayload(
unique_identifier='fb4b5b9c-6188-4c63-8142-fe9c328129fc',
template_attribute=template_attribute
)
@ -2228,7 +2228,7 @@ class TestDeriveKeyResponsePayload(testtools.TestCase):
)
]
)
payload = derive_key.DeriveKeyResponsePayload(
payload = payloads.DeriveKeyResponsePayload(
unique_identifier='fb4b5b9c-6188-4c63-8142-fe9c328129fc',
template_attribute=template_attribute
)

View File

@ -20,7 +20,7 @@ from testtools import TestCase
from kmip.core import utils
from kmip.core.messages.contents import ProtocolVersion
from kmip.core.messages.payloads import discover_versions
from kmip.core.messages import payloads
class TestDiscoverVersionsRequestPayload(TestCase):
@ -54,23 +54,23 @@ class TestDiscoverVersionsRequestPayload(TestCase):
super(TestDiscoverVersionsRequestPayload, self).tearDown()
def test_init_with_none(self):
discover_versions.DiscoverVersionsRequestPayload()
payloads.DiscoverVersionsRequestPayload()
def test_init_with_args(self):
discover_versions.DiscoverVersionsRequestPayload(
payloads.DiscoverVersionsRequestPayload(
self.protocol_versions_empty)
def test_validate_with_invalid_protocol_versions(self):
kwargs = {'protocol_versions': 'invalid'}
self.assertRaisesRegexp(
TypeError, "invalid protocol versions list",
discover_versions.DiscoverVersionsRequestPayload, **kwargs)
payloads.DiscoverVersionsRequestPayload, **kwargs)
def test_validate_with_invalid_protocol_version(self):
kwargs = {'protocol_versions': ['invalid']}
self.assertRaisesRegexp(
TypeError, "invalid protocol version",
discover_versions.DiscoverVersionsRequestPayload, **kwargs)
payloads.DiscoverVersionsRequestPayload, **kwargs)
def _test_read(self, stream, payload, protocol_versions):
payload.read(stream)
@ -92,21 +92,21 @@ class TestDiscoverVersionsRequestPayload(TestCase):
def test_read_with_empty_protocol_list(self):
stream = self.encoding_empty
payload = discover_versions.DiscoverVersionsRequestPayload()
payload = payloads.DiscoverVersionsRequestPayload()
protocol_versions = self.protocol_versions_empty
self._test_read(stream, payload, protocol_versions)
def test_read_with_one_protocol_version(self):
stream = self.encoding_one
payload = discover_versions.DiscoverVersionsRequestPayload()
payload = payloads.DiscoverVersionsRequestPayload()
protocol_versions = self.protocol_versions_one
self._test_read(stream, payload, protocol_versions)
def test_read_with_two_protocol_versions(self):
stream = self.encoding_two
payload = discover_versions.DiscoverVersionsRequestPayload()
payload = payloads.DiscoverVersionsRequestPayload()
protocol_versions = self.protocol_versions_two
self._test_read(stream, payload, protocol_versions)
@ -129,21 +129,21 @@ class TestDiscoverVersionsRequestPayload(TestCase):
self.assertEqual(expected, stream, msg)
def test_write_with_empty_protocol_list(self):
payload = discover_versions.DiscoverVersionsRequestPayload(
payload = payloads.DiscoverVersionsRequestPayload(
self.protocol_versions_empty)
expected = self.encoding_empty
self._test_write(payload, expected)
def test_write_with_one_protocol_version(self):
payload = discover_versions.DiscoverVersionsRequestPayload(
payload = payloads.DiscoverVersionsRequestPayload(
self.protocol_versions_one)
expected = self.encoding_one
self._test_write(payload, expected)
def test_write_with_two_protocol_versions(self):
payload = discover_versions.DiscoverVersionsRequestPayload(
payload = payloads.DiscoverVersionsRequestPayload(
self.protocol_versions_two)
expected = self.encoding_two
@ -181,23 +181,23 @@ class TestDiscoverVersionsResponsePayload(TestCase):
super(TestDiscoverVersionsResponsePayload, self).tearDown()
def test_init_with_none(self):
discover_versions.DiscoverVersionsResponsePayload()
payloads.DiscoverVersionsResponsePayload()
def test_init_with_args(self):
discover_versions.DiscoverVersionsResponsePayload(
payloads.DiscoverVersionsResponsePayload(
self.protocol_versions_empty)
def test_validate_with_invalid_protocol_versions(self):
kwargs = {'protocol_versions': 'invalid'}
self.assertRaisesRegexp(
TypeError, "invalid protocol versions list",
discover_versions.DiscoverVersionsResponsePayload, **kwargs)
payloads.DiscoverVersionsResponsePayload, **kwargs)
def test_validate_with_invalid_protocol_version(self):
kwargs = {'protocol_versions': ['invalid']}
self.assertRaisesRegexp(
TypeError, "invalid protocol version",
discover_versions.DiscoverVersionsResponsePayload, **kwargs)
payloads.DiscoverVersionsResponsePayload, **kwargs)
def _test_read(self, stream, payload, protocol_versions):
payload.read(stream)
@ -219,21 +219,21 @@ class TestDiscoverVersionsResponsePayload(TestCase):
def test_read_with_empty_protocol_list(self):
stream = self.encoding_empty
payload = discover_versions.DiscoverVersionsResponsePayload()
payload = payloads.DiscoverVersionsResponsePayload()
protocol_versions = self.protocol_versions_empty
self._test_read(stream, payload, protocol_versions)
def test_read_with_one_protocol_version(self):
stream = self.encoding_one
payload = discover_versions.DiscoverVersionsResponsePayload()
payload = payloads.DiscoverVersionsResponsePayload()
protocol_versions = self.protocol_versions_one
self._test_read(stream, payload, protocol_versions)
def test_read_with_two_protocol_versions(self):
stream = self.encoding_two
payload = discover_versions.DiscoverVersionsResponsePayload()
payload = payloads.DiscoverVersionsResponsePayload()
protocol_versions = self.protocol_versions_two
self._test_read(stream, payload, protocol_versions)
@ -256,21 +256,21 @@ class TestDiscoverVersionsResponsePayload(TestCase):
self.assertEqual(expected, stream, msg)
def test_write_with_empty_protocol_list(self):
payload = discover_versions.DiscoverVersionsResponsePayload(
payload = payloads.DiscoverVersionsResponsePayload(
self.protocol_versions_empty)
expected = self.encoding_empty
self._test_write(payload, expected)
def test_write_with_one_protocol_version(self):
payload = discover_versions.DiscoverVersionsResponsePayload(
payload = payloads.DiscoverVersionsResponsePayload(
self.protocol_versions_one)
expected = self.encoding_one
self._test_write(payload, expected)
def test_write_with_two_protocol_versions(self):
payload = discover_versions.DiscoverVersionsResponsePayload(
payload = payloads.DiscoverVersionsResponsePayload(
self.protocol_versions_two)
expected = self.encoding_two

View File

@ -19,7 +19,7 @@ from kmip.core import attributes
from kmip.core import enums
from kmip.core import utils
from kmip.core.messages.payloads import encrypt
from kmip.core.messages import payloads
class TestEncryptRequestPayload(testtools.TestCase):
@ -98,7 +98,7 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that an Encrypt request payload can be constructed with no
arguments.
"""
payload = encrypt.EncryptRequestPayload()
payload = payloads.EncryptRequestPayload()
self.assertEqual(None, payload.unique_identifier)
self.assertEqual(None, payload.cryptographic_parameters)
@ -110,7 +110,7 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that an Encrypt request payload can be constructed with valid
values
"""
payload = encrypt.EncryptRequestPayload(
payload = payloads.EncryptRequestPayload(
unique_identifier='00000000-1111-2222-3333-444444444444',
cryptographic_parameters=attributes.CryptographicParameters(),
data=b'\x01\x02\x03',
@ -133,7 +133,7 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that a TypeError is raised when an invalid value is used to set
the unique identifier of an Encrypt request payload.
"""
payload = encrypt.EncryptRequestPayload()
payload = payloads.EncryptRequestPayload()
args = (payload, 'unique_identifier', 0)
self.assertRaisesRegexp(
TypeError,
@ -147,7 +147,7 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that a TypeError is raised when an invalid value is used to set
the cryptographic parameters of an Encrypt request payload.
"""
payload = encrypt.EncryptRequestPayload()
payload = payloads.EncryptRequestPayload()
args = (payload, 'cryptographic_parameters', 'invalid')
self.assertRaisesRegexp(
TypeError,
@ -162,7 +162,7 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that a TypeError is raised when an invalid value is used to set
the data of an Encrypt request payload.
"""
payload = encrypt.EncryptRequestPayload()
payload = payloads.EncryptRequestPayload()
args = (payload, 'data', 0)
self.assertRaisesRegexp(
TypeError,
@ -176,7 +176,7 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that a TypeError is raised when an invalid value is used to set
the IV/counter/nonce of an Encrypt request payload.
"""
payload = encrypt.EncryptRequestPayload()
payload = payloads.EncryptRequestPayload()
args = (payload, 'iv_counter_nonce', 0)
self.assertRaisesRegexp(
TypeError,
@ -189,7 +189,7 @@ class TestEncryptRequestPayload(testtools.TestCase):
"""
Test that an Encrypt request payload can be read from a data stream.
"""
payload = encrypt.EncryptRequestPayload()
payload = payloads.EncryptRequestPayload()
self.assertEqual(None, payload.unique_identifier)
self.assertEqual(None, payload.cryptographic_parameters)
@ -251,7 +251,7 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that an Encrypt request payload can be read from a partial data
stream containing the minimum required attributes.
"""
payload = encrypt.EncryptRequestPayload()
payload = payloads.EncryptRequestPayload()
self.assertEqual(None, payload.unique_identifier)
self.assertEqual(None, payload.cryptographic_parameters)
@ -270,7 +270,7 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that a ValueError gets raised when a required Encrypt request
payload attribute is missing from the payload encoding.
"""
payload = encrypt.EncryptRequestPayload()
payload = payloads.EncryptRequestPayload()
args = (self.empty_encoding, )
self.assertRaisesRegexp(
ValueError,
@ -283,7 +283,7 @@ class TestEncryptRequestPayload(testtools.TestCase):
"""
Test that an Encrypt request payload can be written to a data stream.
"""
payload = encrypt.EncryptRequestPayload(
payload = payloads.EncryptRequestPayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
cryptographic_parameters=attributes.CryptographicParameters(
block_cipher_mode=enums.BlockCipherMode.CBC,
@ -315,7 +315,7 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that a partially defined Encrypt request payload can be written
to a data stream.
"""
payload = encrypt.EncryptRequestPayload(
payload = payloads.EncryptRequestPayload(
data=b'\x01\x02\x03\x04\x05\x06\x07\x08'
)
stream = utils.BytearrayStream()
@ -329,7 +329,7 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that a ValueError gets raised when a required Encrypt request
payload attribute is missing when encoding the payload.
"""
payload = encrypt.EncryptRequestPayload()
payload = payloads.EncryptRequestPayload()
stream = utils.BytearrayStream()
args = (stream, )
self.assertRaisesRegexp(
@ -344,13 +344,13 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that the equality operator returns True when comparing two
Encrypt request payloads with the same data.
"""
a = encrypt.EncryptRequestPayload()
b = encrypt.EncryptRequestPayload()
a = payloads.EncryptRequestPayload()
b = payloads.EncryptRequestPayload()
self.assertTrue(a == b)
self.assertTrue(b == a)
a = encrypt.EncryptRequestPayload(
a = payloads.EncryptRequestPayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
cryptographic_parameters=attributes.CryptographicParameters(
block_cipher_mode=enums.BlockCipherMode.CBC,
@ -371,7 +371,7 @@ class TestEncryptRequestPayload(testtools.TestCase):
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF',
iv_counter_nonce=b'\x01'
)
b = encrypt.EncryptRequestPayload(
b = payloads.EncryptRequestPayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
cryptographic_parameters=attributes.CryptographicParameters(
block_cipher_mode=enums.BlockCipherMode.CBC,
@ -401,10 +401,10 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
Encrypt request payloads with different unique identifiers.
"""
a = encrypt.EncryptRequestPayload(
a = payloads.EncryptRequestPayload(
unique_identifier='a'
)
b = encrypt.EncryptRequestPayload(
b = payloads.EncryptRequestPayload(
unique_identifier='b'
)
@ -416,12 +416,12 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
Encrypt request payloads with different cryptographic parameters.
"""
a = encrypt.EncryptRequestPayload(
a = payloads.EncryptRequestPayload(
cryptographic_parameters=attributes.CryptographicParameters(
block_cipher_mode=enums.BlockCipherMode.CBC
)
)
b = encrypt.EncryptRequestPayload(
b = payloads.EncryptRequestPayload(
cryptographic_parameters=attributes.CryptographicParameters(
hashing_algorithm=enums.HashingAlgorithm.MD5
)
@ -435,8 +435,8 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
Encrypt request payloads with different data.
"""
a = encrypt.EncryptRequestPayload(data=b'\x11')
b = encrypt.EncryptRequestPayload(data=b'\xFF')
a = payloads.EncryptRequestPayload(data=b'\x11')
b = payloads.EncryptRequestPayload(data=b'\xFF')
self.assertFalse(a == b)
self.assertFalse(b == a)
@ -446,8 +446,8 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
Encrypt request payloads with different IV/counter/nonce values.
"""
a = encrypt.EncryptRequestPayload(iv_counter_nonce=b'\x22')
b = encrypt.EncryptRequestPayload(iv_counter_nonce=b'\xAA')
a = payloads.EncryptRequestPayload(iv_counter_nonce=b'\x22')
b = payloads.EncryptRequestPayload(iv_counter_nonce=b'\xAA')
self.assertFalse(a == b)
self.assertFalse(b == a)
@ -457,7 +457,7 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
Encrypt request payloads with different types.
"""
a = encrypt.EncryptRequestPayload()
a = payloads.EncryptRequestPayload()
b = 'invalid'
self.assertFalse(a == b)
@ -468,13 +468,13 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that the inequality operator returns False when comparing two
Encrypt request payloads with the same data.
"""
a = encrypt.EncryptRequestPayload()
b = encrypt.EncryptRequestPayload()
a = payloads.EncryptRequestPayload()
b = payloads.EncryptRequestPayload()
self.assertFalse(a != b)
self.assertFalse(b != a)
a = encrypt.EncryptRequestPayload(
a = payloads.EncryptRequestPayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
cryptographic_parameters=attributes.CryptographicParameters(
block_cipher_mode=enums.BlockCipherMode.CBC,
@ -495,7 +495,7 @@ class TestEncryptRequestPayload(testtools.TestCase):
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF',
iv_counter_nonce=b'\x01'
)
b = encrypt.EncryptRequestPayload(
b = payloads.EncryptRequestPayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
cryptographic_parameters=attributes.CryptographicParameters(
block_cipher_mode=enums.BlockCipherMode.CBC,
@ -525,10 +525,10 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
Encrypt request payloads with different unique identifiers.
"""
a = encrypt.EncryptRequestPayload(
a = payloads.EncryptRequestPayload(
unique_identifier='a'
)
b = encrypt.EncryptRequestPayload(
b = payloads.EncryptRequestPayload(
unique_identifier='b'
)
@ -540,12 +540,12 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
Encrypt request payloads with different cryptographic parameters.
"""
a = encrypt.EncryptRequestPayload(
a = payloads.EncryptRequestPayload(
cryptographic_parameters=attributes.CryptographicParameters(
block_cipher_mode=enums.BlockCipherMode.CBC
)
)
b = encrypt.EncryptRequestPayload(
b = payloads.EncryptRequestPayload(
cryptographic_parameters=attributes.CryptographicParameters(
hashing_algorithm=enums.HashingAlgorithm.MD5
)
@ -559,8 +559,8 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
Encrypt request payloads with different data.
"""
a = encrypt.EncryptRequestPayload(data=b'\x11')
b = encrypt.EncryptRequestPayload(data=b'\xFF')
a = payloads.EncryptRequestPayload(data=b'\x11')
b = payloads.EncryptRequestPayload(data=b'\xFF')
self.assertTrue(a != b)
self.assertTrue(b != a)
@ -570,8 +570,8 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
Encrypt request payloads with different IV/counter/nonce values.
"""
a = encrypt.EncryptRequestPayload(iv_counter_nonce=b'\x22')
b = encrypt.EncryptRequestPayload(iv_counter_nonce=b'\xAA')
a = payloads.EncryptRequestPayload(iv_counter_nonce=b'\x22')
b = payloads.EncryptRequestPayload(iv_counter_nonce=b'\xAA')
self.assertTrue(a != b)
self.assertTrue(b != a)
@ -581,7 +581,7 @@ class TestEncryptRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
Encrypt request payloads with different types.
"""
a = encrypt.EncryptRequestPayload()
a = payloads.EncryptRequestPayload()
b = 'invalid'
self.assertTrue(a != b)
@ -591,7 +591,7 @@ class TestEncryptRequestPayload(testtools.TestCase):
"""
Test that repr can be applied to an Encrypt request payload.
"""
payload = encrypt.EncryptRequestPayload(
payload = payloads.EncryptRequestPayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
cryptographic_parameters=attributes.CryptographicParameters(
block_cipher_mode=enums.BlockCipherMode.CBC,
@ -657,7 +657,7 @@ class TestEncryptRequestPayload(testtools.TestCase):
counter_length=0,
initial_counter_value=1
)
payload = encrypt.EncryptRequestPayload(
payload = payloads.EncryptRequestPayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
cryptographic_parameters=cryptographic_parameters,
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF',
@ -738,7 +738,7 @@ class TestEncryptResponsePayload(testtools.TestCase):
Test that an Encrypt response payload can be constructed with no
arguments.
"""
payload = encrypt.EncryptResponsePayload()
payload = payloads.EncryptResponsePayload()
self.assertEqual(None, payload.unique_identifier)
self.assertEqual(None, payload.data)
@ -749,7 +749,7 @@ class TestEncryptResponsePayload(testtools.TestCase):
Test that an Encrypt response payload can be constructed with valid
values
"""
payload = encrypt.EncryptResponsePayload(
payload = payloads.EncryptResponsePayload(
unique_identifier='00000000-1111-2222-3333-444444444444',
data=b'\x01\x02\x03',
iv_counter_nonce=b'\x01'
@ -767,7 +767,7 @@ class TestEncryptResponsePayload(testtools.TestCase):
Test that a TypeError is raised when an invalid value is used to set
the unique identifier of an Encrypt response payload.
"""
payload = encrypt.EncryptResponsePayload()
payload = payloads.EncryptResponsePayload()
args = (payload, 'unique_identifier', 0)
self.assertRaisesRegexp(
TypeError,
@ -781,7 +781,7 @@ class TestEncryptResponsePayload(testtools.TestCase):
Test that a TypeError is raised when an invalid value is used to set
the data of an Encrypt response payload.
"""
payload = encrypt.EncryptResponsePayload()
payload = payloads.EncryptResponsePayload()
args = (payload, 'data', 0)
self.assertRaisesRegexp(
TypeError,
@ -795,7 +795,7 @@ class TestEncryptResponsePayload(testtools.TestCase):
Test that a TypeError is raised when an invalid value is used to set
the IV/counter/nonce of an Encrypt response payload.
"""
payload = encrypt.EncryptResponsePayload()
payload = payloads.EncryptResponsePayload()
args = (payload, 'iv_counter_nonce', 0)
self.assertRaisesRegexp(
TypeError,
@ -808,7 +808,7 @@ class TestEncryptResponsePayload(testtools.TestCase):
"""
Test that an Encrypt response payload can be read from a data stream.
"""
payload = encrypt.EncryptResponsePayload()
payload = payloads.EncryptResponsePayload()
self.assertEqual(None, payload.unique_identifier)
self.assertEqual(None, payload.data)
@ -828,7 +828,7 @@ class TestEncryptResponsePayload(testtools.TestCase):
Test that an Encrypt response payload can be read from a partial data
stream containing the minimum required attributes.
"""
payload = encrypt.EncryptResponsePayload()
payload = payloads.EncryptResponsePayload()
self.assertEqual(None, payload.unique_identifier)
self.assertEqual(None, payload.data)
@ -848,7 +848,7 @@ class TestEncryptResponsePayload(testtools.TestCase):
Test that a ValueError gets raised when required Encrypt response
payload attributes are missing from the payload encoding.
"""
payload = encrypt.EncryptResponsePayload()
payload = payloads.EncryptResponsePayload()
args = (self.empty_encoding, )
self.assertRaisesRegexp(
ValueError,
@ -857,7 +857,7 @@ class TestEncryptResponsePayload(testtools.TestCase):
*args
)
payload = encrypt.EncryptResponsePayload()
payload = payloads.EncryptResponsePayload()
args = (self.incomplete_encoding, )
self.assertRaisesRegexp(
ValueError,
@ -870,7 +870,7 @@ class TestEncryptResponsePayload(testtools.TestCase):
"""
Test that an Encrypt response payload can be written to a data stream.
"""
payload = encrypt.EncryptResponsePayload(
payload = payloads.EncryptResponsePayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF',
iv_counter_nonce=b'\x01'
@ -886,7 +886,7 @@ class TestEncryptResponsePayload(testtools.TestCase):
Test that a partially defined Encrypt response payload can be written
to a data stream.
"""
payload = encrypt.EncryptResponsePayload(
payload = payloads.EncryptResponsePayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
data=b'\x01\x02\x03\x04\x05\x06\x07\x08'
)
@ -901,7 +901,7 @@ class TestEncryptResponsePayload(testtools.TestCase):
Test that a ValueError gets raised when required Encrypt response
payload attributes are missing when encoding the payload.
"""
payload = encrypt.EncryptResponsePayload()
payload = payloads.EncryptResponsePayload()
self.assertIsNone(payload.unique_identifier)
stream = utils.BytearrayStream()
args = (stream, )
@ -912,7 +912,7 @@ class TestEncryptResponsePayload(testtools.TestCase):
*args
)
payload = encrypt.EncryptResponsePayload(
payload = payloads.EncryptResponsePayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959'
)
stream = utils.BytearrayStream()
@ -929,18 +929,18 @@ class TestEncryptResponsePayload(testtools.TestCase):
Test that the equality operator returns True when comparing two
Encrypt response payloads with the same data.
"""
a = encrypt.EncryptResponsePayload()
b = encrypt.EncryptResponsePayload()
a = payloads.EncryptResponsePayload()
b = payloads.EncryptResponsePayload()
self.assertTrue(a == b)
self.assertTrue(b == a)
a = encrypt.EncryptResponsePayload(
a = payloads.EncryptResponsePayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF',
iv_counter_nonce=b'\x01'
)
b = encrypt.EncryptResponsePayload(
b = payloads.EncryptResponsePayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF',
iv_counter_nonce=b'\x01'
@ -954,10 +954,10 @@ class TestEncryptResponsePayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
Encrypt response payloads with different unique identifiers.
"""
a = encrypt.EncryptResponsePayload(
a = payloads.EncryptResponsePayload(
unique_identifier='a'
)
b = encrypt.EncryptResponsePayload(
b = payloads.EncryptResponsePayload(
unique_identifier='b'
)
@ -969,8 +969,8 @@ class TestEncryptResponsePayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
Encrypt response payloads with different data.
"""
a = encrypt.EncryptResponsePayload(data=b'\x11')
b = encrypt.EncryptResponsePayload(data=b'\xFF')
a = payloads.EncryptResponsePayload(data=b'\x11')
b = payloads.EncryptResponsePayload(data=b'\xFF')
self.assertFalse(a == b)
self.assertFalse(b == a)
@ -980,8 +980,8 @@ class TestEncryptResponsePayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
Encrypt response payloads with different IV/counter/nonce values.
"""
a = encrypt.EncryptResponsePayload(iv_counter_nonce=b'\x22')
b = encrypt.EncryptResponsePayload(iv_counter_nonce=b'\xAA')
a = payloads.EncryptResponsePayload(iv_counter_nonce=b'\x22')
b = payloads.EncryptResponsePayload(iv_counter_nonce=b'\xAA')
self.assertFalse(a == b)
self.assertFalse(b == a)
@ -991,7 +991,7 @@ class TestEncryptResponsePayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
Encrypt response payloads with different types.
"""
a = encrypt.EncryptResponsePayload()
a = payloads.EncryptResponsePayload()
b = 'invalid'
self.assertFalse(a == b)
@ -1002,18 +1002,18 @@ class TestEncryptResponsePayload(testtools.TestCase):
Test that the inequality operator returns False when comparing two
Encrypt response payloads with the same data.
"""
a = encrypt.EncryptResponsePayload()
b = encrypt.EncryptResponsePayload()
a = payloads.EncryptResponsePayload()
b = payloads.EncryptResponsePayload()
self.assertFalse(a != b)
self.assertFalse(b != a)
a = encrypt.EncryptResponsePayload(
a = payloads.EncryptResponsePayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF',
iv_counter_nonce=b'\x01'
)
b = encrypt.EncryptResponsePayload(
b = payloads.EncryptResponsePayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF',
iv_counter_nonce=b'\x01'
@ -1027,10 +1027,10 @@ class TestEncryptResponsePayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
Encrypt response payloads with different unique identifiers.
"""
a = encrypt.EncryptResponsePayload(
a = payloads.EncryptResponsePayload(
unique_identifier='a'
)
b = encrypt.EncryptResponsePayload(
b = payloads.EncryptResponsePayload(
unique_identifier='b'
)
@ -1042,8 +1042,8 @@ class TestEncryptResponsePayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
Encrypt response payloads with different data.
"""
a = encrypt.EncryptResponsePayload(data=b'\x11')
b = encrypt.EncryptResponsePayload(data=b'\xFF')
a = payloads.EncryptResponsePayload(data=b'\x11')
b = payloads.EncryptResponsePayload(data=b'\xFF')
self.assertTrue(a != b)
self.assertTrue(b != a)
@ -1053,8 +1053,8 @@ class TestEncryptResponsePayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
Encrypt response payloads with different IV/counter/nonce values.
"""
a = encrypt.EncryptResponsePayload(iv_counter_nonce=b'\x22')
b = encrypt.EncryptResponsePayload(iv_counter_nonce=b'\xAA')
a = payloads.EncryptResponsePayload(iv_counter_nonce=b'\x22')
b = payloads.EncryptResponsePayload(iv_counter_nonce=b'\xAA')
self.assertTrue(a != b)
self.assertTrue(b != a)
@ -1064,7 +1064,7 @@ class TestEncryptResponsePayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
Encrypt response payloads with different types.
"""
a = encrypt.EncryptResponsePayload()
a = payloads.EncryptResponsePayload()
b = 'invalid'
self.assertTrue(a != b)
@ -1074,7 +1074,7 @@ class TestEncryptResponsePayload(testtools.TestCase):
"""
Test that repr can be applied to an Encrypt response payload.
"""
payload = encrypt.EncryptResponsePayload(
payload = payloads.EncryptResponsePayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF',
iv_counter_nonce=b'\x01'
@ -1093,7 +1093,7 @@ class TestEncryptResponsePayload(testtools.TestCase):
"""
Test that str can be applied to an Encrypt response payload
"""
payload = encrypt.EncryptResponsePayload(
payload = payloads.EncryptResponsePayload(
unique_identifier='b4faee10-aa2a-4446-8ad4-0881f3422959',
data=b'\x01\x23\x45\x67\x89\xAB\xCD\xEF',
iv_counter_nonce=b'\x01'

View File

@ -22,7 +22,7 @@ from kmip.core import objects
from kmip.core import secrets
from kmip.core import utils
from kmip.core.messages.payloads import get
from kmip.core.messages import payloads
class TestGetRequestPayload(testtools.TestCase):
@ -95,7 +95,7 @@ class TestGetRequestPayload(testtools.TestCase):
"""
Test that a Get request payload can be constructed with no arguments.
"""
payload = get.GetRequestPayload()
payload = payloads.GetRequestPayload()
self.assertEqual(None, payload.unique_identifier)
self.assertEqual(None, payload.key_format_type)
@ -106,7 +106,7 @@ class TestGetRequestPayload(testtools.TestCase):
"""
Test that a Get request payload can be constructed with valid values.
"""
payload = get.GetRequestPayload(
payload = payloads.GetRequestPayload(
unique_identifier='00000000-2222-4444-6666-888888888888',
key_format_type=enums.KeyFormatType.RAW,
key_compression_type=enums.KeyCompressionType.
@ -143,11 +143,11 @@ class TestGetRequestPayload(testtools.TestCase):
self.assertRaisesRegexp(
TypeError,
"Unique identifier must be a string.",
get.GetRequestPayload,
payloads.GetRequestPayload,
**kwargs
)
args = (get.GetRequestPayload(), 'unique_identifier', 0)
args = (payloads.GetRequestPayload(), 'unique_identifier', 0)
self.assertRaisesRegexp(
TypeError,
"Unique identifier must be a string.",
@ -164,11 +164,11 @@ class TestGetRequestPayload(testtools.TestCase):
self.assertRaisesRegexp(
TypeError,
"Key format type must be a KeyFormatType enumeration.",
get.GetRequestPayload,
payloads.GetRequestPayload,
**kwargs
)
args = (get.GetRequestPayload(), 'key_format_type', 'invalid')
args = (payloads.GetRequestPayload(), 'key_format_type', 'invalid')
self.assertRaisesRegexp(
TypeError,
"Key format type must be a KeyFormatType enumeration.",
@ -185,11 +185,15 @@ class TestGetRequestPayload(testtools.TestCase):
self.assertRaisesRegexp(
TypeError,
"Key compression type must be a KeyCompressionType enumeration.",
get.GetRequestPayload,
payloads.GetRequestPayload,
**kwargs
)
args = (get.GetRequestPayload(), 'key_compression_type', 'invalid')
args = (
payloads.GetRequestPayload(),
'key_compression_type',
'invalid'
)
self.assertRaisesRegexp(
TypeError,
"Key compression type must be a KeyCompressionType enumeration.",
@ -207,12 +211,12 @@ class TestGetRequestPayload(testtools.TestCase):
TypeError,
"Key wrapping specification must be a KeyWrappingSpecification "
"struct.",
get.GetRequestPayload,
payloads.GetRequestPayload,
**kwargs
)
args = (
get.GetRequestPayload(),
payloads.GetRequestPayload(),
'key_wrapping_specification',
'invalid'
)
@ -228,7 +232,7 @@ class TestGetRequestPayload(testtools.TestCase):
"""
Test that a GetRequestPayload struct can be read from a data stream.
"""
payload = get.GetRequestPayload()
payload = payloads.GetRequestPayload()
self.assertEqual(None, payload.unique_identifier)
self.assertEqual(None, payload.key_format_type)
@ -282,7 +286,7 @@ class TestGetRequestPayload(testtools.TestCase):
Test that a GetRequestPayload struct can be read from a partial data
stream.
"""
payload = get.GetRequestPayload()
payload = payloads.GetRequestPayload()
self.assertEqual(None, payload.unique_identifier)
self.assertEqual(None, payload.key_format_type)
@ -304,7 +308,7 @@ class TestGetRequestPayload(testtools.TestCase):
Test that a GetRequestPayload struct can be read from an empty data
stream.
"""
payload = get.GetRequestPayload()
payload = payloads.GetRequestPayload()
self.assertEqual(None, payload.unique_identifier)
self.assertEqual(None, payload.key_format_type)
@ -322,7 +326,7 @@ class TestGetRequestPayload(testtools.TestCase):
"""
Test that a GetRequestPayload struct can be written to a data stream.
"""
payload = get.GetRequestPayload(
payload = payloads.GetRequestPayload(
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038',
key_format_type=enums.KeyFormatType.RAW,
key_compression_type=enums.KeyCompressionType.
@ -351,7 +355,7 @@ class TestGetRequestPayload(testtools.TestCase):
Test that a partially defined GetRequestPayload struct can be written
to a data stream.
"""
payload = get.GetRequestPayload(
payload = payloads.GetRequestPayload(
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038'
)
stream = utils.BytearrayStream()
@ -366,7 +370,7 @@ class TestGetRequestPayload(testtools.TestCase):
Test that an empty GetRequestPayload struct can be written to a data
stream.
"""
payload = get.GetRequestPayload()
payload = payloads.GetRequestPayload()
stream = utils.BytearrayStream()
payload.write(stream)
@ -379,13 +383,13 @@ class TestGetRequestPayload(testtools.TestCase):
Test that the equality operator returns True when comparing two
GetRequestPayload structs with the same data.
"""
a = get.GetRequestPayload()
b = get.GetRequestPayload()
a = payloads.GetRequestPayload()
b = payloads.GetRequestPayload()
self.assertTrue(a == b)
self.assertTrue(b == a)
a = get.GetRequestPayload(
a = payloads.GetRequestPayload(
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038',
key_format_type=enums.KeyFormatType.RAW,
key_compression_type=enums.KeyCompressionType.
@ -402,7 +406,7 @@ class TestGetRequestPayload(testtools.TestCase):
encoding_option=enums.EncodingOption.NO_ENCODING
)
)
b = get.GetRequestPayload(
b = payloads.GetRequestPayload(
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038',
key_format_type=enums.KeyFormatType.RAW,
key_compression_type=enums.KeyCompressionType.
@ -428,10 +432,10 @@ class TestGetRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
GetRequestPayload structs with different unique identifiers.
"""
a = get.GetRequestPayload(
a = payloads.GetRequestPayload(
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038'
)
b = get.GetRequestPayload(
b = payloads.GetRequestPayload(
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c303f'
)
@ -443,10 +447,10 @@ class TestGetRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
GetRequestPayload structs with different key format types.
"""
a = get.GetRequestPayload(
a = payloads.GetRequestPayload(
key_format_type=enums.KeyFormatType.RAW
)
b = get.GetRequestPayload(
b = payloads.GetRequestPayload(
key_format_type=enums.KeyFormatType.OPAQUE
)
@ -458,11 +462,11 @@ class TestGetRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
GetRequestPayload structs with different key compression types.
"""
a = get.GetRequestPayload(
a = payloads.GetRequestPayload(
key_compression_type=enums.KeyCompressionType.
EC_PUBLIC_KEY_TYPE_UNCOMPRESSED
)
b = get.GetRequestPayload(
b = payloads.GetRequestPayload(
key_compression_type=enums.KeyCompressionType.
EC_PUBLIC_KEY_TYPE_X9_62_HYBRID
)
@ -475,7 +479,7 @@ class TestGetRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
GetRequestPayload structs with different key wrapping specifications.
"""
a = get.GetRequestPayload(
a = payloads.GetRequestPayload(
key_wrapping_specification=objects.KeyWrappingSpecification(
wrapping_method=enums.WrappingMethod.ENCRYPT_THEN_MAC_SIGN,
encryption_key_information=objects.EncryptionKeyInformation(
@ -488,7 +492,7 @@ class TestGetRequestPayload(testtools.TestCase):
encoding_option=enums.EncodingOption.NO_ENCODING
)
)
b = get.GetRequestPayload(
b = payloads.GetRequestPayload(
key_wrapping_specification=objects.KeyWrappingSpecification(
wrapping_method=enums.WrappingMethod.ENCRYPT,
encryption_key_information=objects.EncryptionKeyInformation(
@ -510,7 +514,7 @@ class TestGetRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
GetRequestPayload structs with different types.
"""
a = get.GetRequestPayload()
a = payloads.GetRequestPayload()
b = 'invalid'
self.assertFalse(a == b)
@ -521,13 +525,13 @@ class TestGetRequestPayload(testtools.TestCase):
Test that the inequality operator returns False when comparing two
GetRequestPayload structs with the same data.
"""
a = get.GetRequestPayload()
b = get.GetRequestPayload()
a = payloads.GetRequestPayload()
b = payloads.GetRequestPayload()
self.assertFalse(a != b)
self.assertFalse(b != a)
a = get.GetRequestPayload(
a = payloads.GetRequestPayload(
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038',
key_format_type=enums.KeyFormatType.RAW,
key_compression_type=enums.KeyCompressionType.
@ -544,7 +548,7 @@ class TestGetRequestPayload(testtools.TestCase):
encoding_option=enums.EncodingOption.NO_ENCODING
)
)
b = get.GetRequestPayload(
b = payloads.GetRequestPayload(
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038',
key_format_type=enums.KeyFormatType.RAW,
key_compression_type=enums.KeyCompressionType.
@ -570,10 +574,10 @@ class TestGetRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
GetRequestPayload structs with different unique identifiers.
"""
a = get.GetRequestPayload(
a = payloads.GetRequestPayload(
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038'
)
b = get.GetRequestPayload(
b = payloads.GetRequestPayload(
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c303f'
)
@ -585,10 +589,10 @@ class TestGetRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
GetRequestPayload structs with different key format types.
"""
a = get.GetRequestPayload(
a = payloads.GetRequestPayload(
key_format_type=enums.KeyFormatType.RAW
)
b = get.GetRequestPayload(
b = payloads.GetRequestPayload(
key_format_type=enums.KeyFormatType.OPAQUE
)
@ -600,11 +604,11 @@ class TestGetRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
GetRequestPayload structs with different key compression types.
"""
a = get.GetRequestPayload(
a = payloads.GetRequestPayload(
key_compression_type=enums.KeyCompressionType.
EC_PUBLIC_KEY_TYPE_UNCOMPRESSED
)
b = get.GetRequestPayload(
b = payloads.GetRequestPayload(
key_compression_type=enums.KeyCompressionType.
EC_PUBLIC_KEY_TYPE_X9_62_HYBRID
)
@ -617,7 +621,7 @@ class TestGetRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
GetRequestPayload structs with different key wrapping specifications.
"""
a = get.GetRequestPayload(
a = payloads.GetRequestPayload(
key_wrapping_specification=objects.KeyWrappingSpecification(
wrapping_method=enums.WrappingMethod.ENCRYPT_THEN_MAC_SIGN,
encryption_key_information=objects.EncryptionKeyInformation(
@ -630,7 +634,7 @@ class TestGetRequestPayload(testtools.TestCase):
encoding_option=enums.EncodingOption.NO_ENCODING
)
)
b = get.GetRequestPayload(
b = payloads.GetRequestPayload(
key_wrapping_specification=objects.KeyWrappingSpecification(
wrapping_method=enums.WrappingMethod.ENCRYPT,
encryption_key_information=objects.EncryptionKeyInformation(
@ -652,7 +656,7 @@ class TestGetRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
GetRequestPayload structs with different types.
"""
a = get.GetRequestPayload()
a = payloads.GetRequestPayload()
b = 'invalid'
self.assertTrue(a != b)
@ -662,7 +666,7 @@ class TestGetRequestPayload(testtools.TestCase):
"""
Test that repr can be applied to a GetRequestPayload struct.
"""
payload = get.GetRequestPayload(
payload = payloads.GetRequestPayload(
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038',
key_format_type=enums.KeyFormatType.RAW,
key_compression_type=enums.KeyCompressionType.
@ -717,7 +721,7 @@ class TestGetRequestPayload(testtools.TestCase):
"""
Test that str can be applied to a GetRequestPayload struct.
"""
payload = get.GetRequestPayload(
payload = payloads.GetRequestPayload(
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038',
key_format_type=enums.KeyFormatType.RAW,
key_compression_type=enums.KeyCompressionType.
@ -844,7 +848,7 @@ class TestGetResponsePayload(testtools.TestCase):
Test that a GetRequestPayload struct can be constructed with no
arguments.
"""
payload = get.GetResponsePayload()
payload = payloads.GetResponsePayload()
self.assertEqual(None, payload.object_type)
self.assertEqual(None, payload.unique_identifier)
@ -855,7 +859,7 @@ class TestGetResponsePayload(testtools.TestCase):
Test that a GetRequestPayload struct can be constructed with valid
values.
"""
payload = get.GetResponsePayload(
payload = payloads.GetResponsePayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifier='11111111-3333-5555-7777-999999999999',
secret=secrets.SymmetricKey(
@ -938,11 +942,11 @@ class TestGetResponsePayload(testtools.TestCase):
self.assertRaisesRegexp(
TypeError,
"Object type must be an ObjectType enumeration.",
get.GetResponsePayload,
payloads.GetResponsePayload,
**kwargs
)
args = (get.GetResponsePayload(), 'object_type', 'invalid')
args = (payloads.GetResponsePayload(), 'object_type', 'invalid')
self.assertRaisesRegexp(
TypeError,
"Object type must be an ObjectType enumeration.",
@ -959,11 +963,11 @@ class TestGetResponsePayload(testtools.TestCase):
self.assertRaisesRegexp(
TypeError,
"Unique identifier must be a string.",
get.GetResponsePayload,
payloads.GetResponsePayload,
**kwargs
)
args = (get.GetResponsePayload(), 'unique_identifier', 0)
args = (payloads.GetResponsePayload(), 'unique_identifier', 0)
self.assertRaisesRegexp(
TypeError,
"Unique identifier must be a string.",
@ -982,11 +986,11 @@ class TestGetResponsePayload(testtools.TestCase):
"Secret must be one of the following structs: Certificate, "
"OpaqueObject, PrivateKey, PublicKey, SecretData, SplitKey, "
"SymmetricKey, Template",
get.GetResponsePayload,
payloads.GetResponsePayload,
**kwargs
)
args = (get.GetResponsePayload(), 'secret', 0)
args = (payloads.GetResponsePayload(), 'secret', 0)
self.assertRaisesRegexp(
TypeError,
"Secret must be one of the following structs: Certificate, "
@ -1000,7 +1004,7 @@ class TestGetResponsePayload(testtools.TestCase):
"""
Test that a GetResponsePayload struct can be read from a data stream.
"""
payload = get.GetResponsePayload()
payload = payloads.GetResponsePayload()
self.assertEqual(None, payload.object_type)
self.assertEqual(None, payload.unique_identifier)
@ -1061,7 +1065,7 @@ class TestGetResponsePayload(testtools.TestCase):
Test that a ValueError gets raised when a required GetResponsePayload
field is missing when decoding the struct.
"""
payload = get.GetResponsePayload()
payload = payloads.GetResponsePayload()
args = (self.partial_encoding_missing_object_type, )
self.assertRaisesRegexp(
ValueError,
@ -1075,7 +1079,7 @@ class TestGetResponsePayload(testtools.TestCase):
Test that a ValueError gets raised when a required GetResponsePayload
field is missing when decoding the struct.
"""
payload = get.GetResponsePayload()
payload = payloads.GetResponsePayload()
args = (self.partial_encoding_missing_unique_id, )
self.assertRaisesRegexp(
ValueError,
@ -1089,7 +1093,7 @@ class TestGetResponsePayload(testtools.TestCase):
Test that a ValueError gets raised when a required GetResponsePayload
field is missing when decoding the struct.
"""
payload = get.GetResponsePayload()
payload = payloads.GetResponsePayload()
args = (self.partial_encoding_missing_secret, )
self.assertRaisesRegexp(
ValueError,
@ -1102,7 +1106,7 @@ class TestGetResponsePayload(testtools.TestCase):
"""
Test that a GetResponsePayload struct can be written to a data stream.
"""
payload = get.GetResponsePayload(
payload = payloads.GetResponsePayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038',
secret=secrets.SymmetricKey(
@ -1135,7 +1139,7 @@ class TestGetResponsePayload(testtools.TestCase):
Test that a ValueError gets raised when a required GetResponsePayload
field is missing when encoding the struct.
"""
payload = get.GetResponsePayload()
payload = payloads.GetResponsePayload()
stream = utils.BytearrayStream()
args = (stream, )
self.assertRaisesRegexp(
@ -1150,7 +1154,7 @@ class TestGetResponsePayload(testtools.TestCase):
Test that a ValueError gets raised when a required GetResponsePayload
field is missing when encoding the struct.
"""
payload = get.GetResponsePayload(
payload = payloads.GetResponsePayload(
object_type=enums.ObjectType.SYMMETRIC_KEY
)
stream = utils.BytearrayStream()
@ -1167,7 +1171,7 @@ class TestGetResponsePayload(testtools.TestCase):
Test that a ValueError gets raised when a required GetResponsePayload
field is missing when encoding the struct.
"""
payload = get.GetResponsePayload(
payload = payloads.GetResponsePayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038'
)
@ -1185,8 +1189,8 @@ class TestGetResponsePayload(testtools.TestCase):
Test that the equality operator returns True when comparing two
GetResponsePayload structs with the same data.
"""
a = get.GetResponsePayload()
b = get.GetResponsePayload()
a = payloads.GetResponsePayload()
b = payloads.GetResponsePayload()
self.assertTrue(a == b)
self.assertTrue(b == a)
@ -1212,12 +1216,12 @@ class TestGetResponsePayload(testtools.TestCase):
)
)
a = get.GetResponsePayload(
a = payloads.GetResponsePayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038',
secret=secret
)
b = get.GetResponsePayload(
b = payloads.GetResponsePayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038',
secret=secret
@ -1231,8 +1235,12 @@ class TestGetResponsePayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
GetResponsePayload structs with different object type fields.
"""
a = get.GetResponsePayload(object_type=enums.ObjectType.SYMMETRIC_KEY)
b = get.GetResponsePayload(object_type=enums.ObjectType.OPAQUE_DATA)
a = payloads.GetResponsePayload(
object_type=enums.ObjectType.SYMMETRIC_KEY
)
b = payloads.GetResponsePayload(
object_type=enums.ObjectType.OPAQUE_DATA
)
self.assertFalse(a == b)
self.assertFalse(b == a)
@ -1242,10 +1250,10 @@ class TestGetResponsePayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
GetResponsePayload structs with different unique identifier fields.
"""
a = get.GetResponsePayload(
a = payloads.GetResponsePayload(
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038'
)
b = get.GetResponsePayload(
b = payloads.GetResponsePayload(
unique_identifier='49a1ca88-6bea-4fb2-ffff-7e58802c3038'
)
@ -1259,7 +1267,7 @@ class TestGetResponsePayload(testtools.TestCase):
"""
# TODO (peter-hamilton): Update this test case once SymmetricKeys
# support proper field-based equality.
a = get.GetResponsePayload(
a = payloads.GetResponsePayload(
secret=secrets.SymmetricKey(
key_block=objects.KeyBlock(
key_format_type=misc.KeyFormatType(
@ -1279,7 +1287,7 @@ class TestGetResponsePayload(testtools.TestCase):
)
)
)
b = get.GetResponsePayload(
b = payloads.GetResponsePayload(
secret=secrets.SymmetricKey(
key_block=objects.KeyBlock(
key_format_type=misc.KeyFormatType(
@ -1308,7 +1316,7 @@ class TestGetResponsePayload(testtools.TestCase):
Test that the equality operators returns False when comparing two
GetResponsePayload structs with different types.
"""
a = get.GetResponsePayload()
a = payloads.GetResponsePayload()
b = 'invalid'
self.assertFalse(a == b)
@ -1319,8 +1327,8 @@ class TestGetResponsePayload(testtools.TestCase):
Test that the inequality operator returns False when comparing two
GetResponsePayload structs with the same data.
"""
a = get.GetResponsePayload()
b = get.GetResponsePayload()
a = payloads.GetResponsePayload()
b = payloads.GetResponsePayload()
self.assertFalse(a != b)
self.assertFalse(b != a)
@ -1344,12 +1352,12 @@ class TestGetResponsePayload(testtools.TestCase):
)
)
a = get.GetResponsePayload(
a = payloads.GetResponsePayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038',
secret=secret
)
b = get.GetResponsePayload(
b = payloads.GetResponsePayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038',
secret=secret
@ -1363,8 +1371,12 @@ class TestGetResponsePayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
GetResponsePayload structs with different object type fields.
"""
a = get.GetResponsePayload(object_type=enums.ObjectType.SYMMETRIC_KEY)
b = get.GetResponsePayload(object_type=enums.ObjectType.OPAQUE_DATA)
a = payloads.GetResponsePayload(
object_type=enums.ObjectType.SYMMETRIC_KEY
)
b = payloads.GetResponsePayload(
object_type=enums.ObjectType.OPAQUE_DATA
)
self.assertTrue(a != b)
self.assertTrue(b != a)
@ -1374,10 +1386,10 @@ class TestGetResponsePayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
GetResponsePayload structs with different unique identifier fields.
"""
a = get.GetResponsePayload(
a = payloads.GetResponsePayload(
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038'
)
b = get.GetResponsePayload(
b = payloads.GetResponsePayload(
unique_identifier='49a1ca88-6bea-4fb2-ffff-7e58802c3038'
)
@ -1391,7 +1403,7 @@ class TestGetResponsePayload(testtools.TestCase):
"""
# TODO (peter-hamilton): Update this test case once SymmetricKeys
# support proper field-based equality.
a = get.GetResponsePayload(
a = payloads.GetResponsePayload(
secret=secrets.SymmetricKey(
key_block=objects.KeyBlock(
key_format_type=misc.KeyFormatType(
@ -1411,7 +1423,7 @@ class TestGetResponsePayload(testtools.TestCase):
)
)
)
b = get.GetResponsePayload(
b = payloads.GetResponsePayload(
secret=secrets.SymmetricKey(
key_block=objects.KeyBlock(
key_format_type=misc.KeyFormatType(
@ -1440,7 +1452,7 @@ class TestGetResponsePayload(testtools.TestCase):
Test that the inequality operators returns True when comparing two
GetResponsePayload structs with different types.
"""
a = get.GetResponsePayload()
a = payloads.GetResponsePayload()
b = 'invalid'
self.assertTrue(a != b)
@ -1450,7 +1462,7 @@ class TestGetResponsePayload(testtools.TestCase):
"""
Test that repr can be applied to a GetResponsePayload struct.
"""
payload = get.GetResponsePayload(
payload = payloads.GetResponsePayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038',
secret=secrets.SymmetricKey(
@ -1506,7 +1518,7 @@ class TestGetResponsePayload(testtools.TestCase):
cryptographic_length=attributes.CryptographicLength(168)
)
)
payload = get.GetResponsePayload(
payload = payloads.GetResponsePayload(
object_type=enums.ObjectType.SYMMETRIC_KEY,
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038',
secret=secret

View File

@ -19,7 +19,7 @@ from kmip.core import enums
from kmip.core import primitives
from kmip.core import utils
from kmip.core.messages.payloads import get_attribute_list
from kmip.core.messages import payloads
class TestGetAttributeListRequestPayload(testtools.TestCase):
@ -52,14 +52,14 @@ class TestGetAttributeListRequestPayload(testtools.TestCase):
Test that a GetAttributeList request payload can be constructed with
no arguments.
"""
get_attribute_list.GetAttributeListRequestPayload()
payloads.GetAttributeListRequestPayload()
def test_init_with_args(self):
"""
Test that a GetAttributeList request payload can be constructed with a
valid value.
"""
get_attribute_list.GetAttributeListRequestPayload(
payloads.GetAttributeListRequestPayload(
'test-unique-identifier',
)
@ -68,7 +68,7 @@ class TestGetAttributeListRequestPayload(testtools.TestCase):
Test that the unique_identifier attribute of a GetAttributeList
request payload can be properly set and retrieved.
"""
payload = get_attribute_list.GetAttributeListRequestPayload()
payload = payloads.GetAttributeListRequestPayload()
self.assertIsNone(payload.unique_identifier)
self.assertIsNone(payload._unique_identifier)
@ -89,7 +89,7 @@ class TestGetAttributeListRequestPayload(testtools.TestCase):
Test that a TypeError is raised when an invalid ID is used to set
the unique_identifier attribute of a GetAttributeList request payload.
"""
payload = get_attribute_list.GetAttributeListRequestPayload()
payload = payloads.GetAttributeListRequestPayload()
args = (payload, 'unique_identifier', 0)
self.assertRaisesRegexp(
TypeError,
@ -103,7 +103,7 @@ class TestGetAttributeListRequestPayload(testtools.TestCase):
Test that a GetAttributeList request payload can be read from a data
stream.
"""
payload = get_attribute_list.GetAttributeListRequestPayload()
payload = payloads.GetAttributeListRequestPayload()
self.assertEqual(None, payload._unique_identifier)
@ -123,7 +123,7 @@ class TestGetAttributeListRequestPayload(testtools.TestCase):
Test that a GetAttributeList response payload with no ID or attribute
names can be read from a data stream.
"""
payload = get_attribute_list.GetAttributeListRequestPayload()
payload = payloads.GetAttributeListRequestPayload()
self.assertEqual(None, payload._unique_identifier)
@ -137,7 +137,7 @@ class TestGetAttributeListRequestPayload(testtools.TestCase):
Test that a GetAttributeList request payload can be written to a data
stream.
"""
payload = get_attribute_list.GetAttributeListRequestPayload(
payload = payloads.GetAttributeListRequestPayload(
self.unique_identifier
)
stream = utils.BytearrayStream()
@ -151,7 +151,7 @@ class TestGetAttributeListRequestPayload(testtools.TestCase):
Test that a GetAttributeList request payload with no ID or attribute
names can be written to a data stream.
"""
payload = get_attribute_list.GetAttributeListRequestPayload()
payload = payloads.GetAttributeListRequestPayload()
stream = utils.BytearrayStream()
payload.write(stream)
@ -162,7 +162,7 @@ class TestGetAttributeListRequestPayload(testtools.TestCase):
"""
Test that repr can be applied to a GetAttributeList request payload.
"""
payload = get_attribute_list.GetAttributeListRequestPayload(
payload = payloads.GetAttributeListRequestPayload(
self.unique_identifier
)
unique_identifier = "unique_identifier={0}".format(
@ -179,7 +179,7 @@ class TestGetAttributeListRequestPayload(testtools.TestCase):
Test that repr can be applied to a GetAttributeList request payload
with no ID or attribute names.
"""
payload = get_attribute_list.GetAttributeListRequestPayload(
payload = payloads.GetAttributeListRequestPayload(
None
)
unique_identifier = "unique_identifier={0}".format(
@ -195,7 +195,7 @@ class TestGetAttributeListRequestPayload(testtools.TestCase):
"""
Test that str can be applied to a GetAttributeList request payload.
"""
payload = get_attribute_list.GetAttributeListRequestPayload(
payload = payloads.GetAttributeListRequestPayload(
self.unique_identifier
)
expected = str({
@ -209,7 +209,7 @@ class TestGetAttributeListRequestPayload(testtools.TestCase):
Test that str can be applied to a GetAttributeList request payload
with no ID or attribute names.
"""
payload = get_attribute_list.GetAttributeListRequestPayload(
payload = payloads.GetAttributeListRequestPayload(
None
)
expected = str({
@ -223,10 +223,10 @@ class TestGetAttributeListRequestPayload(testtools.TestCase):
Test that the equality operator returns True when comparing two
GetAttributeList request payloads with the same data.
"""
a = get_attribute_list.GetAttributeListRequestPayload(
a = payloads.GetAttributeListRequestPayload(
self.unique_identifier
)
b = get_attribute_list.GetAttributeListRequestPayload(
b = payloads.GetAttributeListRequestPayload(
self.unique_identifier
)
@ -238,10 +238,10 @@ class TestGetAttributeListRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
GetAttributeList request payloads with different IDs.
"""
a = get_attribute_list.GetAttributeListRequestPayload(
a = payloads.GetAttributeListRequestPayload(
self.unique_identifier
)
b = get_attribute_list.GetAttributeListRequestPayload(
b = payloads.GetAttributeListRequestPayload(
'invalid'
)
@ -254,7 +254,7 @@ class TestGetAttributeListRequestPayload(testtools.TestCase):
GetAttributeList request payload to a non-GetAttributeList request
payload.
"""
a = get_attribute_list.GetAttributeListRequestPayload(
a = payloads.GetAttributeListRequestPayload(
self.unique_identifier
)
b = "invalid"
@ -267,10 +267,10 @@ class TestGetAttributeListRequestPayload(testtools.TestCase):
Test that the inequality operator returns False when comparing
two GetAttributeList request payloads with the same internal data.
"""
a = get_attribute_list.GetAttributeListRequestPayload(
a = payloads.GetAttributeListRequestPayload(
self.unique_identifier
)
b = get_attribute_list.GetAttributeListRequestPayload(
b = payloads.GetAttributeListRequestPayload(
self.unique_identifier
)
@ -282,10 +282,10 @@ class TestGetAttributeListRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
GetAttributeList request payloads with different IDs.
"""
a = get_attribute_list.GetAttributeListRequestPayload(
a = payloads.GetAttributeListRequestPayload(
self.unique_identifier
)
b = get_attribute_list.GetAttributeListRequestPayload(
b = payloads.GetAttributeListRequestPayload(
'invalid'
)
@ -298,7 +298,7 @@ class TestGetAttributeListRequestPayload(testtools.TestCase):
GetAttributeList request payload to a non-GetAttributeList request
payload.
"""
a = get_attribute_list.GetAttributeListRequestPayload(
a = payloads.GetAttributeListRequestPayload(
self.unique_identifier
)
b = "invalid"
@ -398,14 +398,14 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that a GetAttributeList response payload can be constructed with
no arguments.
"""
get_attribute_list.GetAttributeListResponsePayload()
payloads.GetAttributeListResponsePayload()
def test_init_with_args(self):
"""
Test that a GetAttributeList response payload can be constructed with a
valid value.
"""
get_attribute_list.GetAttributeListResponsePayload(
payloads.GetAttributeListResponsePayload(
'test-unique-identifier',
['test-attribute-name-1', 'test-attribute-name-2']
)
@ -415,7 +415,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that the unique_identifier attribute of a GetAttributeList
response payload can be properly set and retrieved.
"""
payload = get_attribute_list.GetAttributeListResponsePayload()
payload = payloads.GetAttributeListResponsePayload()
self.assertIsNone(payload.unique_identifier)
self.assertIsNone(payload._unique_identifier)
@ -437,7 +437,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
the unique_identifier attribute of a GetAttributeList response
payload.
"""
payload = get_attribute_list.GetAttributeListResponsePayload()
payload = payloads.GetAttributeListResponsePayload()
args = (payload, 'unique_identifier', 0)
self.assertRaisesRegexp(
TypeError,
@ -451,7 +451,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that the attribute_names attribute of a GetAttributeList response
payload can be properly set and retrieved.
"""
payload = get_attribute_list.GetAttributeListResponsePayload()
payload = payloads.GetAttributeListResponsePayload()
self.assertEqual(list(), payload.attribute_names)
self.assertEqual(list(), payload._attribute_names)
@ -485,7 +485,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that a TypeError is raised when an invalid list is used to set
the attribute_names attribute of a GetAttributeList response payload.
"""
payload = get_attribute_list.GetAttributeListResponsePayload()
payload = payloads.GetAttributeListResponsePayload()
args = (payload, 'attribute_names', 0)
self.assertRaisesRegexp(
TypeError,
@ -500,7 +500,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
included in the list used to set the attribute_names attribute of a
GetAttributeList response payload.
"""
payload = get_attribute_list.GetAttributeListResponsePayload()
payload = payloads.GetAttributeListResponsePayload()
args = (
payload,
'attribute_names',
@ -519,7 +519,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that duplicate attribute names are silently removed when setting
the attribute_names attribute of a GetAttributeList response payload.
"""
payload = get_attribute_list.GetAttributeListResponsePayload()
payload = payloads.GetAttributeListResponsePayload()
self.assertEqual(list(), payload.attribute_names)
self.assertEqual(list(), payload._attribute_names)
@ -554,7 +554,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that a GetAttributeList response payload can be read from a data
stream.
"""
payload = get_attribute_list.GetAttributeListResponsePayload()
payload = payloads.GetAttributeListResponsePayload()
self.assertEqual(None, payload._unique_identifier)
self.assertEqual(list(), payload._attribute_names)
@ -587,7 +587,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that a GetAttributeList response payload with no ID can be read
from a data stream.
"""
payload = get_attribute_list.GetAttributeListResponsePayload()
payload = payloads.GetAttributeListResponsePayload()
self.assertEqual(None, payload._unique_identifier)
self.assertEqual(list(), payload._attribute_names)
@ -614,7 +614,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that a GetAttributeList response payload with no attribute names
can be read from a data stream.
"""
payload = get_attribute_list.GetAttributeListResponsePayload()
payload = payloads.GetAttributeListResponsePayload()
self.assertEqual(None, payload._unique_identifier)
self.assertEqual(list(), payload._attribute_names)
@ -637,7 +637,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that a GetAttributeList response payload with no ID or attribute
names can be read from a data stream.
"""
payload = get_attribute_list.GetAttributeListResponsePayload()
payload = payloads.GetAttributeListResponsePayload()
self.assertEqual(None, payload._unique_identifier)
self.assertEqual(list(), payload._attribute_names)
@ -654,7 +654,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that a GetAttributeList response payload can be written to a data
stream.
"""
payload = get_attribute_list.GetAttributeListResponsePayload(
payload = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
self.attribute_names
)
@ -669,7 +669,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that a GetAttributeList response payload with no ID can be
written to a data stream.
"""
payload = get_attribute_list.GetAttributeListResponsePayload(
payload = payloads.GetAttributeListResponsePayload(
None,
self.attribute_names
)
@ -690,7 +690,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that a GetAttributeList response payload with no attribute names
can be written to a data stream.
"""
payload = get_attribute_list.GetAttributeListResponsePayload(
payload = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
None
)
@ -705,7 +705,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that a GetAttributeList response payload with no ID or attribute
names can be written to a data stream.
"""
payload = get_attribute_list.GetAttributeListResponsePayload()
payload = payloads.GetAttributeListResponsePayload()
stream = utils.BytearrayStream()
payload.write(stream)
@ -716,7 +716,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
"""
Test that repr can be applied to a GetAttributeList response payload.
"""
payload = get_attribute_list.GetAttributeListResponsePayload(
payload = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
self.attribute_names
)
@ -738,7 +738,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that repr can be applied to a GetAttributeList response payload
with no ID.
"""
payload = get_attribute_list.GetAttributeListResponsePayload(
payload = payloads.GetAttributeListResponsePayload(
None,
self.attribute_names
)
@ -760,7 +760,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that repr can be applied to a GetAttributeList response payload
with no attribute names.
"""
payload = get_attribute_list.GetAttributeListResponsePayload(
payload = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
None
)
@ -782,7 +782,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that repr can be applied to a GetAttributeList response payload
with no ID or attribute names.
"""
payload = get_attribute_list.GetAttributeListResponsePayload(
payload = payloads.GetAttributeListResponsePayload(
None,
None
)
@ -803,7 +803,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
"""
Test that str can be applied to a GetAttributeList response payload.
"""
payload = get_attribute_list.GetAttributeListResponsePayload(
payload = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
self.attribute_names
)
@ -819,7 +819,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that str can be applied to a GetAttributeList response payload
with no ID.
"""
payload = get_attribute_list.GetAttributeListResponsePayload(
payload = payloads.GetAttributeListResponsePayload(
None,
self.attribute_names
)
@ -835,7 +835,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that str can be applied to a GetAttributeList response payload
with no attribute names.
"""
payload = get_attribute_list.GetAttributeListResponsePayload(
payload = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
None
)
@ -851,7 +851,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that str can be applied to a GetAttributeList response payload
with no ID or attribute names.
"""
payload = get_attribute_list.GetAttributeListResponsePayload(
payload = payloads.GetAttributeListResponsePayload(
None,
None
)
@ -867,11 +867,11 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that the equality operator returns True when comparing two
GetAttributeList response payloads with the same data.
"""
a = get_attribute_list.GetAttributeListResponsePayload(
a = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
self.attribute_names
)
b = get_attribute_list.GetAttributeListResponsePayload(
b = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
self.attribute_names
)
@ -885,12 +885,12 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
GetAttributeList response payload with the same attribute_name sets
but with different attribute name orderings.
"""
a = get_attribute_list.GetAttributeListResponsePayload(
a = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
self.attribute_names
)
self.attribute_names.reverse()
b = get_attribute_list.GetAttributeListResponsePayload(
b = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
self.attribute_names
)
@ -903,11 +903,11 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
GetAttributeList response payloads with different IDs.
"""
a = get_attribute_list.GetAttributeListResponsePayload(
a = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
self.attribute_names
)
b = get_attribute_list.GetAttributeListResponsePayload(
b = payloads.GetAttributeListResponsePayload(
'invalid',
self.attribute_names
)
@ -920,11 +920,11 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
GetAttributeList response payloads with different attribute names.
"""
a = get_attribute_list.GetAttributeListResponsePayload(
a = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
self.attribute_names
)
b = get_attribute_list.GetAttributeListResponsePayload(
b = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
None
)
@ -938,7 +938,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
GetAttributeList response payload to a non-GetAttributeList response
payload.
"""
a = get_attribute_list.GetAttributeListResponsePayload(
a = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
self.attribute_names
)
@ -952,11 +952,11 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that the inequality operator returns False when comparing
two GetAttributeList response payloads with the same internal data.
"""
a = get_attribute_list.GetAttributeListResponsePayload(
a = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
self.attribute_names
)
b = get_attribute_list.GetAttributeListResponsePayload(
b = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
self.attribute_names
)
@ -969,11 +969,11 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
GetAttributeList response payloads with different IDs.
"""
a = get_attribute_list.GetAttributeListResponsePayload(
a = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
self.attribute_names
)
b = get_attribute_list.GetAttributeListResponsePayload(
b = payloads.GetAttributeListResponsePayload(
'invalid',
self.attribute_names
)
@ -986,11 +986,11 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
GetAttributeList response payloads with different attribute names.
"""
a = get_attribute_list.GetAttributeListResponsePayload(
a = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
self.attribute_names
)
b = get_attribute_list.GetAttributeListResponsePayload(
b = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
None
)
@ -1004,7 +1004,7 @@ class TestGetAttributeListResponsePayload(testtools.TestCase):
GetAttributeList response payload to a non-GetAttributeList response
payload.
"""
a = get_attribute_list.GetAttributeListResponsePayload(
a = payloads.GetAttributeListResponsePayload(
self.unique_identifier,
self.attribute_names
)

View File

@ -20,7 +20,7 @@ from kmip.core import objects
from kmip.core import primitives
from kmip.core import utils
from kmip.core.messages.payloads import get_attributes
from kmip.core.messages import payloads
class TestGetAttributesRequestPayload(testtools.TestCase):
@ -82,14 +82,14 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that a GetAttributes request payload can be constructed with
no arguments.
"""
get_attributes.GetAttributesRequestPayload()
payloads.GetAttributesRequestPayload()
def test_init_with_args(self):
"""
Test that a GetAttributes request payload can be constructed with a
valid value.
"""
get_attributes.GetAttributesRequestPayload(
payloads.GetAttributesRequestPayload(
'test-unique-identifier',
['test-attribute-name-1', 'test-attribute-name-2']
)
@ -99,7 +99,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that the unique_identifier attribute of a GetAttributes request
payload can be properly set and retrieved.
"""
payload = get_attributes.GetAttributesRequestPayload()
payload = payloads.GetAttributesRequestPayload()
self.assertIsNone(payload.unique_identifier)
self.assertIsNone(payload._unique_identifier)
@ -120,7 +120,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that a TypeError is raised when an invalid ID is used to set
the unique_identifier attribute of a GetAttributes request payload.
"""
payload = get_attributes.GetAttributesRequestPayload()
payload = payloads.GetAttributesRequestPayload()
args = (payload, 'unique_identifier', 0)
self.assertRaisesRegexp(
TypeError,
@ -134,7 +134,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that the attribute_names attribute of a GetAttributes request
payload can be properly set and retrieved.
"""
payload = get_attributes.GetAttributesRequestPayload()
payload = payloads.GetAttributesRequestPayload()
self.assertEqual(list(), payload.attribute_names)
self.assertEqual(list(), payload._attribute_names)
@ -168,7 +168,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that a TypeError is raised when an invalid list is used to set
the attribute_names attribute of a GetAttributes request payload.
"""
payload = get_attributes.GetAttributesRequestPayload()
payload = payloads.GetAttributesRequestPayload()
args = (payload, 'attribute_names', 0)
self.assertRaisesRegexp(
TypeError,
@ -183,7 +183,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
included in the list used to set the attribute_names attribute of a
GetAttributes request payload.
"""
payload = get_attributes.GetAttributesRequestPayload()
payload = payloads.GetAttributesRequestPayload()
args = (
payload,
'attribute_names',
@ -202,7 +202,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that duplicate attribute names are silently removed when setting
the attribute_names attribute of a GetAttributes request payload.
"""
payload = get_attributes.GetAttributesRequestPayload()
payload = payloads.GetAttributesRequestPayload()
self.assertEqual(list(), payload.attribute_names)
self.assertEqual(list(), payload._attribute_names)
@ -237,7 +237,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that a GetAttributes request payload can be read from a data
stream.
"""
payload = get_attributes.GetAttributesRequestPayload()
payload = payloads.GetAttributesRequestPayload()
self.assertEqual(None, payload._unique_identifier)
self.assertEqual(list(), payload._attribute_names)
@ -270,7 +270,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that a GetAttributes request payload with no ID can be read
from a data stream.
"""
payload = get_attributes.GetAttributesRequestPayload()
payload = payloads.GetAttributesRequestPayload()
self.assertEqual(None, payload._unique_identifier)
self.assertEqual(list(), payload._attribute_names)
@ -297,7 +297,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that a GetAttributes request payload with no attribute names
can be read from a data stream.
"""
payload = get_attributes.GetAttributesRequestPayload()
payload = payloads.GetAttributesRequestPayload()
self.assertEqual(None, payload._unique_identifier)
self.assertEqual(list(), payload._attribute_names)
@ -320,7 +320,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that a GetAttributes request payload with no ID or attribute
names can be read from a data stream.
"""
payload = get_attributes.GetAttributesRequestPayload()
payload = payloads.GetAttributesRequestPayload()
self.assertEqual(None, payload._unique_identifier)
self.assertEqual(list(), payload._attribute_names)
@ -337,7 +337,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that a GetAttributes request payload can be written to a data
stream.
"""
payload = get_attributes.GetAttributesRequestPayload(
payload = payloads.GetAttributesRequestPayload(
self.unique_identifier,
self.attribute_names
)
@ -352,7 +352,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that a GetAttributes request payload with no ID can be written
to a data stream.
"""
payload = get_attributes.GetAttributesRequestPayload(
payload = payloads.GetAttributesRequestPayload(
None,
self.attribute_names
)
@ -373,7 +373,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that a GetAttributes request payload with no attribute names
can be written to a data stream.
"""
payload = get_attributes.GetAttributesRequestPayload(
payload = payloads.GetAttributesRequestPayload(
self.unique_identifier,
None
)
@ -388,7 +388,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that a GetAttributes request payload with no ID or attribute
names can be written to a data stream.
"""
payload = get_attributes.GetAttributesRequestPayload()
payload = payloads.GetAttributesRequestPayload()
stream = utils.BytearrayStream()
payload.write(stream)
@ -399,7 +399,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
"""
Test that repr can be applied to a GetAttributes request payload.
"""
payload = get_attributes.GetAttributesRequestPayload(
payload = payloads.GetAttributesRequestPayload(
self.unique_identifier,
self.attribute_names
)
@ -421,7 +421,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that repr can be applied to a GetAttributes request payload with
no ID.
"""
payload = get_attributes.GetAttributesRequestPayload(
payload = payloads.GetAttributesRequestPayload(
None,
self.attribute_names
)
@ -443,7 +443,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that repr can be applied to a GetAttributes request payload with
no attribute names.
"""
payload = get_attributes.GetAttributesRequestPayload(
payload = payloads.GetAttributesRequestPayload(
self.unique_identifier,
None
)
@ -465,7 +465,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that repr can be applied to a GetAttributes request payload with
no ID or attribute names.
"""
payload = get_attributes.GetAttributesRequestPayload(
payload = payloads.GetAttributesRequestPayload(
None,
None
)
@ -486,7 +486,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
"""
Test that str can be applied to a GetAttributes request payload.
"""
payload = get_attributes.GetAttributesRequestPayload(
payload = payloads.GetAttributesRequestPayload(
self.unique_identifier,
self.attribute_names
)
@ -502,7 +502,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that str can be applied to a GetAttributes request payload with
no ID.
"""
payload = get_attributes.GetAttributesRequestPayload(
payload = payloads.GetAttributesRequestPayload(
None,
self.attribute_names
)
@ -518,7 +518,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that str can be applied to a GetAttributes request payload with
no attribute names.
"""
payload = get_attributes.GetAttributesRequestPayload(
payload = payloads.GetAttributesRequestPayload(
self.unique_identifier,
None
)
@ -534,7 +534,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that str can be applied to a GetAttributes request payload with
no ID or attribute names.
"""
payload = get_attributes.GetAttributesRequestPayload(
payload = payloads.GetAttributesRequestPayload(
None,
None
)
@ -550,11 +550,11 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that the equality operator returns True when comparing two
GetAttributes request payloads with the same data.
"""
a = get_attributes.GetAttributesRequestPayload(
a = payloads.GetAttributesRequestPayload(
self.unique_identifier,
self.attribute_names
)
b = get_attributes.GetAttributesRequestPayload(
b = payloads.GetAttributesRequestPayload(
self.unique_identifier,
self.attribute_names
)
@ -568,12 +568,12 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
GetAttributes request payload with the same attribute_name sets
but with different attribute name orderings.
"""
a = get_attributes.GetAttributesRequestPayload(
a = payloads.GetAttributesRequestPayload(
self.unique_identifier,
self.attribute_names
)
self.attribute_names.reverse()
b = get_attributes.GetAttributesRequestPayload(
b = payloads.GetAttributesRequestPayload(
self.unique_identifier,
self.attribute_names
)
@ -586,11 +586,11 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
GetAttributes request payloads with different IDs.
"""
a = get_attributes.GetAttributesRequestPayload(
a = payloads.GetAttributesRequestPayload(
self.unique_identifier,
self.attribute_names
)
b = get_attributes.GetAttributesRequestPayload(
b = payloads.GetAttributesRequestPayload(
'invalid',
self.attribute_names
)
@ -603,11 +603,11 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
GetAttributes request payloads with different attribute names.
"""
a = get_attributes.GetAttributesRequestPayload(
a = payloads.GetAttributesRequestPayload(
self.unique_identifier,
self.attribute_names
)
b = get_attributes.GetAttributesRequestPayload(
b = payloads.GetAttributesRequestPayload(
self.unique_identifier,
None
)
@ -621,7 +621,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
GetAttributes request payload to a non-GetAttributes request
payload.
"""
a = get_attributes.GetAttributesRequestPayload(
a = payloads.GetAttributesRequestPayload(
self.unique_identifier,
self.attribute_names
)
@ -635,11 +635,11 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that the inequality operator returns False when comparing
two GetAttributes request payloads with the same internal data.
"""
a = get_attributes.GetAttributesRequestPayload(
a = payloads.GetAttributesRequestPayload(
self.unique_identifier,
self.attribute_names
)
b = get_attributes.GetAttributesRequestPayload(
b = payloads.GetAttributesRequestPayload(
self.unique_identifier,
self.attribute_names
)
@ -652,11 +652,11 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
GetAttributes request payloads with different IDs.
"""
a = get_attributes.GetAttributesRequestPayload(
a = payloads.GetAttributesRequestPayload(
self.unique_identifier,
self.attribute_names
)
b = get_attributes.GetAttributesRequestPayload(
b = payloads.GetAttributesRequestPayload(
'invalid',
self.attribute_names
)
@ -669,11 +669,11 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
GetAttributes request payloads with different attribute names.
"""
a = get_attributes.GetAttributesRequestPayload(
a = payloads.GetAttributesRequestPayload(
self.unique_identifier,
self.attribute_names
)
b = get_attributes.GetAttributesRequestPayload(
b = payloads.GetAttributesRequestPayload(
self.unique_identifier,
None
)
@ -687,7 +687,7 @@ class TestGetAttributesRequestPayload(testtools.TestCase):
GetAttributes request payload to a non-GetAttributes request
payload.
"""
a = get_attributes.GetAttributesRequestPayload(
a = payloads.GetAttributesRequestPayload(
self.unique_identifier,
self.attribute_names
)
@ -791,14 +791,14 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
"""
Test that a GetAttributes response payload can be constructed.
"""
get_attributes.GetAttributesResponsePayload()
payloads.GetAttributesResponsePayload()
def test_init_with_args(self):
"""
Test that a GetAttributes response payload can be constructed with a
valid value.
"""
get_attributes.GetAttributesResponsePayload(
payloads.GetAttributesResponsePayload(
'test-unique-identifier',
[objects.Attribute(), objects.Attribute()]
)
@ -808,7 +808,7 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that the unique_identifier attribute of a GetAttributes response
payload can be properly set and retrieved.
"""
payload = get_attributes.GetAttributesResponsePayload()
payload = payloads.GetAttributesResponsePayload()
self.assertIsNone(payload.unique_identifier)
self.assertIsNone(payload._unique_identifier)
@ -829,7 +829,7 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that a TypeError is raised when an invalid ID is used to set
the unique_identifier attribute of a GetAttributes response payload.
"""
payload = get_attributes.GetAttributesResponsePayload()
payload = payloads.GetAttributesResponsePayload()
args = (payload, 'unique_identifier', 0)
self.assertRaisesRegexp(
TypeError,
@ -843,7 +843,7 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that the attributes attribute of a GetAttributes response
payload can be properly set and retrieved.
"""
payload = get_attributes.GetAttributesResponsePayload()
payload = payloads.GetAttributesResponsePayload()
self.assertEqual(list(), payload.attributes)
self.assertEqual(list(), payload._attributes)
@ -863,7 +863,7 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that a TypeError is raised when an invalid list is used to set
the attributes attribute of a GetAttributes response payload.
"""
payload = get_attributes.GetAttributesResponsePayload()
payload = payloads.GetAttributesResponsePayload()
args = (payload, 'attributes', 0)
self.assertRaisesRegexp(
TypeError,
@ -878,7 +878,7 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
in the list used to set the attributes attribute of a GetAttributes
response payload.
"""
payload = get_attributes.GetAttributesResponsePayload()
payload = payloads.GetAttributesResponsePayload()
args = (
payload,
'attributes',
@ -897,7 +897,7 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that a GetAttributes response payload can be read from a data
stream.
"""
payload = get_attributes.GetAttributesResponsePayload()
payload = payloads.GetAttributesResponsePayload()
self.assertEqual(None, payload._unique_identifier)
self.assertEqual(list(), payload._attributes)
@ -927,7 +927,7 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that an InvalidKmipEncoding error gets raised when attempting to
read a GetAttributes response encoding with no ID data.
"""
payload = get_attributes.GetAttributesResponsePayload()
payload = payloads.GetAttributesResponsePayload()
self.assertEqual(None, payload._unique_identifier)
self.assertEqual(list(), payload._attributes)
@ -945,7 +945,7 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that a GetAttributes response payload without attribute name
data can be read from a data stream.
"""
payload = get_attributes.GetAttributesResponsePayload()
payload = payloads.GetAttributesResponsePayload()
self.assertEqual(None, payload._unique_identifier)
self.assertEqual(list(), payload._attributes)
@ -968,7 +968,7 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that a GetAttributes response payload can be written to a data
stream.
"""
payload = get_attributes.GetAttributesResponsePayload(
payload = payloads.GetAttributesResponsePayload(
self.unique_identifier,
self.attributes
)
@ -983,7 +983,7 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that a GetAttributes request payload with no ID can be written
to a data stream.
"""
payload = get_attributes.GetAttributesResponsePayload(
payload = payloads.GetAttributesResponsePayload(
None,
self.attributes
)
@ -1002,7 +1002,7 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that a GetAttributes response payload with no attribute name
data can be written to a data stream.
"""
payload = get_attributes.GetAttributesResponsePayload(
payload = payloads.GetAttributesResponsePayload(
self.unique_identifier,
None
)
@ -1016,7 +1016,7 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
"""
Test that repr can be applied to a GetAttributes response payload.
"""
payload = get_attributes.GetAttributesResponsePayload(
payload = payloads.GetAttributesResponsePayload(
self.unique_identifier,
self.attributes
)
@ -1037,7 +1037,7 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
"""
Test that str can be applied to a GetAttributes response payload.
"""
payload = get_attributes.GetAttributesResponsePayload(
payload = payloads.GetAttributesResponsePayload(
self.unique_identifier,
self.attributes
)
@ -1053,11 +1053,11 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that the equality operator returns True when comparing two
GetAttributes response payloads with the same data.
"""
a = get_attributes.GetAttributesResponsePayload(
a = payloads.GetAttributesResponsePayload(
self.unique_identifier,
self.attributes
)
b = get_attributes.GetAttributesResponsePayload(
b = payloads.GetAttributesResponsePayload(
self.unique_identifier,
self.attributes
)
@ -1070,11 +1070,11 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
GetAttributes response payloads with different data.
"""
a = get_attributes.GetAttributesResponsePayload(
a = payloads.GetAttributesResponsePayload(
self.unique_identifier,
self.attributes
)
b = get_attributes.GetAttributesResponsePayload(
b = payloads.GetAttributesResponsePayload(
'invalid',
self.attributes
)
@ -1087,13 +1087,13 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
GetAttributes response payloads with different data.
"""
a = get_attributes.GetAttributesResponsePayload(
a = payloads.GetAttributesResponsePayload(
self.unique_identifier,
self.attributes
)
reversed_attributes = copy.deepcopy(self.attributes)
reversed_attributes.reverse()
b = get_attributes.GetAttributesResponsePayload(
b = payloads.GetAttributesResponsePayload(
self.unique_identifier,
reversed_attributes
)
@ -1106,11 +1106,11 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
GetAttributes response payloads with different data.
"""
a = get_attributes.GetAttributesResponsePayload(
a = payloads.GetAttributesResponsePayload(
self.unique_identifier,
self.attributes
)
b = get_attributes.GetAttributesResponsePayload(
b = payloads.GetAttributesResponsePayload(
self.unique_identifier,
list()
)
@ -1123,11 +1123,11 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that the equality operator returns False when comparing two
GetAttributes response payloads with different data.
"""
a = get_attributes.GetAttributesResponsePayload(
a = payloads.GetAttributesResponsePayload(
self.unique_identifier,
None
)
b = get_attributes.GetAttributesResponsePayload(
b = payloads.GetAttributesResponsePayload(
self.unique_identifier,
self.attributes
)
@ -1141,7 +1141,7 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
GetAttributes response payload to a non-GetAttributes response
payload.
"""
a = get_attributes.GetAttributesResponsePayload(
a = payloads.GetAttributesResponsePayload(
self.unique_identifier,
self.attributes
)
@ -1155,11 +1155,11 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that the inequality operator returns False when comparing
two GetAttributes response payloads with the same internal data.
"""
a = get_attributes.GetAttributesResponsePayload(
a = payloads.GetAttributesResponsePayload(
self.unique_identifier,
self.attributes
)
b = get_attributes.GetAttributesResponsePayload(
b = payloads.GetAttributesResponsePayload(
self.unique_identifier,
self.attributes
)
@ -1172,11 +1172,11 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that the inequality operator returns True when comparing two
GetAttributes request payloads with different data.
"""
a = get_attributes.GetAttributesResponsePayload(
a = payloads.GetAttributesResponsePayload(
self.unique_identifier,
self.attributes
)
b = get_attributes.GetAttributesResponsePayload(
b = payloads.GetAttributesResponsePayload(
'invalid',
self.attributes
)
@ -1189,13 +1189,13 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that the inequality operator returns False when comparing two
GetAttributes response payloads with different data.
"""
a = get_attributes.GetAttributesResponsePayload(
a = payloads.GetAttributesResponsePayload(
self.unique_identifier,
self.attributes
)
reversed_attributes = copy.deepcopy(self.attributes)
reversed_attributes.reverse()
b = get_attributes.GetAttributesResponsePayload(
b = payloads.GetAttributesResponsePayload(
self.unique_identifier,
reversed_attributes
)
@ -1208,11 +1208,11 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that the inequality operator returns False when comparing two
GetAttributes response payloads with different data.
"""
a = get_attributes.GetAttributesResponsePayload(
a = payloads.GetAttributesResponsePayload(
self.unique_identifier,
self.attributes
)
b = get_attributes.GetAttributesResponsePayload(
b = payloads.GetAttributesResponsePayload(
self.unique_identifier,
list()
)
@ -1225,11 +1225,11 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
Test that the inequality operator returns False when comparing two
GetAttributes response payloads with different data.
"""
a = get_attributes.GetAttributesResponsePayload(
a = payloads.GetAttributesResponsePayload(
self.unique_identifier,
None
)
b = get_attributes.GetAttributesResponsePayload(
b = payloads.GetAttributesResponsePayload(
self.unique_identifier,
self.attributes
)
@ -1243,7 +1243,7 @@ class TestGetAttributesResponsePayload(testtools.TestCase):
GetAttributes response payload to a non-GetAttributes response
payload.
"""
a = get_attributes.GetAttributesResponsePayload(
a = payloads.GetAttributesResponsePayload(
self.unique_identifier,
self.attributes
)

View File

@ -20,7 +20,7 @@ from kmip.core import utils
from kmip.core import enums
from kmip.core import exceptions
from kmip.core.messages.payloads import mac
from kmip.core.messages import payloads
class TestMACRequestPayload(TestCase):
@ -55,14 +55,14 @@ class TestMACRequestPayload(TestCase):
super(TestMACRequestPayload, self).tearDown()
def test_init_with_none(self):
mac.MACRequestPayload()
payloads.MACRequestPayload()
def test_init_valid(self):
"""
Test that the payload can be properly constructed and the attributes
cab be properly set and retrieved.
"""
payload = mac.MACRequestPayload(
payload = payloads.MACRequestPayload(
self.unique_identifier,
self.cryptographic_parameters,
self.data)
@ -77,7 +77,7 @@ class TestMACRequestPayload(TestCase):
'data': None}
self.assertRaisesRegexp(
TypeError, "unique identifier must be UniqueIdentifier type",
mac.MACRequestPayload, **kwargs)
payloads.MACRequestPayload, **kwargs)
def test_init_with_invalid_cryptographic_parameters(self):
kwargs = {'unique_identifier': None,
@ -86,7 +86,7 @@ class TestMACRequestPayload(TestCase):
self.assertRaisesRegexp(
TypeError,
"cryptographic parameters must be CryptographicParameters type",
mac.MACRequestPayload, **kwargs)
payloads.MACRequestPayload, **kwargs)
def test_init_with_invalid_data(self):
kwargs = {'unique_identifier': None,
@ -94,11 +94,11 @@ class TestMACRequestPayload(TestCase):
'data': 'invalid'}
self.assertRaises(
TypeError, "data must be Data type",
mac.MACRequestPayload, **kwargs)
payloads.MACRequestPayload, **kwargs)
def test_read_valid(self):
stream = self.encoding_full
payload = mac.MACRequestPayload()
payload = payloads.MACRequestPayload()
payload.read(stream)
self.assertEqual(self.unique_identifier, payload.unique_identifier)
@ -111,7 +111,7 @@ class TestMACRequestPayload(TestCase):
Test that an InvalidKmipEncoding error gets raised when attempting to
read a mac request encoding with no data.
"""
payload = mac.MACRequestPayload()
payload = payloads.MACRequestPayload()
args = (self.encoding_no_data,)
self.assertRaisesRegexp(
exceptions.InvalidKmipEncoding,
@ -124,7 +124,7 @@ class TestMACRequestPayload(TestCase):
expected = self.encoding_full
stream = utils.BytearrayStream()
payload = mac.MACRequestPayload(
payload = payloads.MACRequestPayload(
self.unique_identifier,
self.cryptographic_parameters,
self.data)
@ -138,7 +138,7 @@ class TestMACRequestPayload(TestCase):
write a mac request with no data.
"""
stream = utils.BytearrayStream()
payload = mac.MACRequestPayload(
payload = payloads.MACRequestPayload(
self.unique_identifier,
self.cryptographic_parameters,
None)
@ -189,14 +189,14 @@ class TestMACResponsePayload(TestCase):
super(TestMACResponsePayload, self).tearDown()
def test_init_with_none(self):
mac.MACResponsePayload()
payloads.MACResponsePayload()
def test_init_valid(self):
"""
Test that the payload can be properly constructed and the attributes
can be properly set and retrieved.
"""
payload = mac.MACResponsePayload(
payload = payloads.MACResponsePayload(
self.unique_identifier,
self.mac_data)
self.assertEqual(payload.unique_identifier, self.unique_identifier)
@ -207,18 +207,18 @@ class TestMACResponsePayload(TestCase):
'mac_data': None}
self.assertRaisesRegexp(
TypeError, "unique identifier must be UniqueIdentifier type",
mac.MACResponsePayload, **kwargs)
payloads.MACResponsePayload, **kwargs)
def test_init_with_invalid_mac_data(self):
kwargs = {'unique_identifier': None,
'mac_data': 'invalid'}
self.assertRaises(
TypeError, "data must be MACData type",
mac.MACResponsePayload, **kwargs)
payloads.MACResponsePayload, **kwargs)
def test_read_valid(self):
stream = self.encoding_full
payload = mac.MACResponsePayload()
payload = payloads.MACResponsePayload()
payload.read(stream)
self.assertEqual(self.unique_identifier, payload.unique_identifier)
@ -229,7 +229,7 @@ class TestMACResponsePayload(TestCase):
Test that an InvalidKmipEncoding error gets raised when attempting to
read a mac response encoding with no unique identifier.
"""
payload = mac.MACResponsePayload()
payload = payloads.MACResponsePayload()
args = (self.encoding_no_unique_identifier,)
self.assertRaisesRegexp(
exceptions.InvalidKmipEncoding,
@ -243,7 +243,7 @@ class TestMACResponsePayload(TestCase):
Test that an InvalidKmipEncoding error gets raised when attempting to
read a mac response encoding with no mac data.
"""
payload = mac.MACResponsePayload()
payload = payloads.MACResponsePayload()
args = (self.encoding_no_mac_data,)
self.assertRaisesRegexp(
exceptions.InvalidKmipEncoding,
@ -256,7 +256,7 @@ class TestMACResponsePayload(TestCase):
expected = self.encoding_full
stream = utils.BytearrayStream()
payload = mac.MACResponsePayload(
payload = payloads.MACResponsePayload(
self.unique_identifier,
self.mac_data)
payload.write(stream)
@ -269,7 +269,7 @@ class TestMACResponsePayload(TestCase):
write a mac response with no unique identifier.
"""
stream = utils.BytearrayStream()
payload = mac.MACResponsePayload(
payload = payloads.MACResponsePayload(
None,
self.mac_data)
args = (stream,)
@ -286,7 +286,7 @@ class TestMACResponsePayload(TestCase):
write a mac response with no mac data.
"""
stream = utils.BytearrayStream()
payload = mac.MACResponsePayload(
payload = payloads.MACResponsePayload(
self.unique_identifier,
None)
args = (stream,)

View File

@ -26,7 +26,7 @@ from kmip.core.enums import Operation as OperationEnum
from kmip.core.enums import QueryFunction as QueryFunctionEnum
from kmip.core.messages.contents import Operation
from kmip.core.messages.payloads import query
from kmip.core.messages import payloads
from kmip.core.misc import QueryFunction
from kmip.core.misc import VendorIdentification
@ -82,16 +82,16 @@ class TestQueryRequestPayload(TestCase):
Test that a QueryRequestPayload object can be constructed with no
specified value.
"""
query.QueryRequestPayload()
payloads.QueryRequestPayload()
def test_init_with_args(self):
"""
Test that a QueryRequestPayload object can be constructed with valid
values.
"""
query.QueryRequestPayload(self.query_functions_a)
query.QueryRequestPayload(self.query_functions_b)
query.QueryRequestPayload(self.query_functions_c)
payloads.QueryRequestPayload(self.query_functions_a)
payloads.QueryRequestPayload(self.query_functions_b)
payloads.QueryRequestPayload(self.query_functions_c)
def test_validate_with_invalid_query_functions_list(self):
"""
@ -101,7 +101,7 @@ class TestQueryRequestPayload(TestCase):
kwargs = {'query_functions': 'invalid'}
self.assertRaisesRegexp(
TypeError, "invalid query functions list",
query.QueryRequestPayload, **kwargs)
payloads.QueryRequestPayload, **kwargs)
def test_validate_with_invalid_query_functions_item(self):
"""
@ -111,10 +111,10 @@ class TestQueryRequestPayload(TestCase):
kwargs = {'query_functions': ['invalid']}
self.assertRaisesRegexp(
TypeError, "invalid query function",
query.QueryRequestPayload, **kwargs)
payloads.QueryRequestPayload, **kwargs)
def _test_read(self, stream, query_functions):
payload = query.QueryRequestPayload()
payload = payloads.QueryRequestPayload()
payload.read(stream)
expected = len(query_functions)
observed = len(payload.query_functions)
@ -155,7 +155,7 @@ class TestQueryRequestPayload(TestCase):
def _test_write(self, encoding, query_functions):
stream = utils.BytearrayStream()
payload = query.QueryRequestPayload(query_functions)
payload = payloads.QueryRequestPayload(query_functions)
payload.write(stream)
length_expected = len(encoding)
@ -307,14 +307,14 @@ class TestQueryResponsePayload(TestCase):
Test that a QueryResponsePayload object can be constructed with no
specified value.
"""
query.QueryResponsePayload()
payloads.QueryResponsePayload()
def test_init_with_args(self):
"""
Test that a QueryResponsePayload object can be constructed with valid
values.
"""
query.QueryResponsePayload(
payloads.QueryResponsePayload(
operations=self.operations,
object_types=self.object_types,
vendor_identification=self.vendor_identification,
@ -330,7 +330,7 @@ class TestQueryResponsePayload(TestCase):
kwargs = {'operations': 'invalid'}
self.assertRaisesRegexp(
TypeError, "invalid operations list",
query.QueryResponsePayload, **kwargs)
payloads.QueryResponsePayload, **kwargs)
def test_validate_with_invalid_operations_item(self):
"""
@ -340,7 +340,7 @@ class TestQueryResponsePayload(TestCase):
kwargs = {'operations': ['invalid']}
self.assertRaisesRegexp(
TypeError, "invalid operation",
query.QueryResponsePayload, **kwargs)
payloads.QueryResponsePayload, **kwargs)
def test_validate_with_invalid_object_types_list(self):
"""
@ -350,7 +350,7 @@ class TestQueryResponsePayload(TestCase):
kwargs = {'object_types': 'invalid'}
self.assertRaisesRegexp(
TypeError, "invalid object types list",
query.QueryResponsePayload, **kwargs)
payloads.QueryResponsePayload, **kwargs)
def test_validate_with_invalid_object_types_item(self):
"""
@ -360,7 +360,7 @@ class TestQueryResponsePayload(TestCase):
kwargs = {'object_types': ['invalid']}
self.assertRaisesRegexp(
TypeError, "invalid object type",
query.QueryResponsePayload, **kwargs)
payloads.QueryResponsePayload, **kwargs)
def test_validate_with_invalid_vendor_identification(self):
"""
@ -371,7 +371,7 @@ class TestQueryResponsePayload(TestCase):
kwargs = {'vendor_identification': 'invalid'}
self.assertRaisesRegexp(
TypeError, "invalid vendor identification",
query.QueryResponsePayload, **kwargs)
payloads.QueryResponsePayload, **kwargs)
def test_validate_with_invalid_server_information(self):
"""
@ -382,7 +382,7 @@ class TestQueryResponsePayload(TestCase):
kwargs = {'server_information': 'invalid'}
self.assertRaisesRegexp(
TypeError, "invalid server information",
query.QueryResponsePayload, **kwargs)
payloads.QueryResponsePayload, **kwargs)
def test_validate_with_invalid_application_namespaces_list(self):
"""
@ -393,7 +393,7 @@ class TestQueryResponsePayload(TestCase):
kwargs = {'application_namespaces': 'invalid'}
self.assertRaisesRegexp(
TypeError, "invalid application namespaces list",
query.QueryResponsePayload, **kwargs)
payloads.QueryResponsePayload, **kwargs)
def test_validate_with_invalid_application_namespaces_item(self):
"""
@ -404,7 +404,7 @@ class TestQueryResponsePayload(TestCase):
kwargs = {'application_namespaces': ['invalid']}
self.assertRaisesRegexp(
TypeError, "invalid application namespace",
query.QueryResponsePayload, **kwargs)
payloads.QueryResponsePayload, **kwargs)
def test_validate_with_invalid_extension_information_list(self):
"""
@ -415,7 +415,7 @@ class TestQueryResponsePayload(TestCase):
kwargs = {'extension_information': 'invalid'}
self.assertRaisesRegexp(
TypeError, "invalid extension information list",
query.QueryResponsePayload, **kwargs)
payloads.QueryResponsePayload, **kwargs)
def test_validate_with_invalid_extension_information_item(self):
"""
@ -426,12 +426,12 @@ class TestQueryResponsePayload(TestCase):
kwargs = {'extension_information': ['invalid']}
self.assertRaisesRegexp(
TypeError, "invalid extension information",
query.QueryResponsePayload, **kwargs)
payloads.QueryResponsePayload, **kwargs)
def _test_read(self, stream, operations, object_types,
vendor_identification, server_information,
application_namespaces, extension_information):
payload = query.QueryResponsePayload()
payload = payloads.QueryResponsePayload()
payload.read(stream)
# Test decoding of all operations.
@ -541,7 +541,7 @@ class TestQueryResponsePayload(TestCase):
vendor_identification, server_information,
application_namespaces, extension_information):
stream = utils.BytearrayStream()
payload = query.QueryResponsePayload(
payload = payloads.QueryResponsePayload(
operations, object_types, vendor_identification,
server_information, application_namespaces, extension_information)
payload.write(stream)

View File

@ -20,7 +20,7 @@ from kmip.core import misc
from kmip.core import objects
from kmip.core import utils
from kmip.core.messages.payloads import rekey_key_pair
from kmip.core.messages import payloads
class TestRekeyKeyPairRequestPayload(TestCase):
@ -53,10 +53,10 @@ class TestRekeyKeyPairRequestPayload(TestCase):
super(TestRekeyKeyPairRequestPayload, self).tearDown()
def test_init_with_none(self):
rekey_key_pair.RekeyKeyPairRequestPayload()
payloads.RekeyKeyPairRequestPayload()
def test_init_with_args(self):
rekey_key_pair.RekeyKeyPairRequestPayload(
payloads.RekeyKeyPairRequestPayload(
self.private_key_uuid, self.offset, self.common_template_attribute,
self.private_key_template_attribute,
self.public_key_template_attribute)
@ -68,7 +68,7 @@ class TestRekeyKeyPairRequestPayload(TestCase):
'public_key_template_attribute': None}
self.assertRaisesRegexp(
TypeError, "invalid private key unique identifier",
rekey_key_pair.RekeyKeyPairRequestPayload, **kwargs)
payloads.RekeyKeyPairRequestPayload, **kwargs)
def test_validate_with_invalid_offset(self):
kwargs = {'private_key_uuid': None, 'offset': 'invalid',
@ -77,7 +77,7 @@ class TestRekeyKeyPairRequestPayload(TestCase):
'public_key_template_attribute': None}
self.assertRaisesRegexp(
TypeError, "invalid offset",
rekey_key_pair.RekeyKeyPairRequestPayload, **kwargs)
payloads.RekeyKeyPairRequestPayload, **kwargs)
def test_validate_with_invalid_common_template_attribute(self):
kwargs = {'private_key_uuid': None, 'offset': None,
@ -86,7 +86,7 @@ class TestRekeyKeyPairRequestPayload(TestCase):
'public_key_template_attribute': None}
self.assertRaisesRegexp(
TypeError, "invalid common template attribute",
rekey_key_pair.RekeyKeyPairRequestPayload, **kwargs)
payloads.RekeyKeyPairRequestPayload, **kwargs)
def test_validate_with_invalid_private_key_template_attribute(self):
kwargs = {'private_key_uuid': None, 'offset': None,
@ -95,7 +95,7 @@ class TestRekeyKeyPairRequestPayload(TestCase):
'public_key_template_attribute': None}
self.assertRaisesRegexp(
TypeError, "invalid private key template attribute",
rekey_key_pair.RekeyKeyPairRequestPayload, **kwargs)
payloads.RekeyKeyPairRequestPayload, **kwargs)
def test_validate_with_invalid_public_key_template_attribute(self):
kwargs = {'private_key_uuid': None, 'offset': None,
@ -104,7 +104,7 @@ class TestRekeyKeyPairRequestPayload(TestCase):
'public_key_template_attribute': 'invalid'}
self.assertRaisesRegexp(
TypeError, "invalid public key template attribute",
rekey_key_pair.RekeyKeyPairRequestPayload, **kwargs)
payloads.RekeyKeyPairRequestPayload, **kwargs)
def _test_read(self, stream, payload, private_key_uuid, offset,
common_template_attribute, private_key_template_attribute,
@ -142,13 +142,13 @@ class TestRekeyKeyPairRequestPayload(TestCase):
def test_read_with_none(self):
stream = self.encoding_empty
payload = rekey_key_pair.RekeyKeyPairRequestPayload()
payload = payloads.RekeyKeyPairRequestPayload()
self._test_read(stream, payload, None, None, None, None, None)
def test_read_with_args(self):
stream = self.encoding_full
payload = rekey_key_pair.RekeyKeyPairRequestPayload()
payload = payloads.RekeyKeyPairRequestPayload()
self._test_read(stream, payload, self.private_key_uuid, self.offset,
self.common_template_attribute,
@ -173,13 +173,13 @@ class TestRekeyKeyPairRequestPayload(TestCase):
def test_write_with_none(self):
stream = utils.BytearrayStream()
payload = rekey_key_pair.RekeyKeyPairRequestPayload()
payload = payloads.RekeyKeyPairRequestPayload()
self._test_write(stream, payload, self.encoding_empty)
def test_write_with_args(self):
stream = utils.BytearrayStream()
payload = rekey_key_pair.RekeyKeyPairRequestPayload(
payload = payloads.RekeyKeyPairRequestPayload(
self.private_key_uuid, self.offset, self.common_template_attribute,
self.private_key_template_attribute,
self.public_key_template_attribute)
@ -222,10 +222,10 @@ class TestRekeyKeyPairResponsePayload(TestCase):
super(TestRekeyKeyPairResponsePayload, self).tearDown()
def test_init_with_none(self):
rekey_key_pair.RekeyKeyPairResponsePayload()
payloads.RekeyKeyPairResponsePayload()
def test_init_with_args(self):
rekey_key_pair.RekeyKeyPairResponsePayload(
payloads.RekeyKeyPairResponsePayload(
self.private_key_uuid, self.public_key_uuid,
self.private_key_template_attribute,
self.public_key_template_attribute)
@ -237,7 +237,7 @@ class TestRekeyKeyPairResponsePayload(TestCase):
'public_key_template_attribute': None}
self.assertRaisesRegexp(
TypeError, "invalid private key unique identifier",
rekey_key_pair.RekeyKeyPairResponsePayload, **kwargs)
payloads.RekeyKeyPairResponsePayload, **kwargs)
def test_validate_with_invalid_public_key_unique_identifier(self):
kwargs = {'private_key_uuid': None,
@ -246,7 +246,7 @@ class TestRekeyKeyPairResponsePayload(TestCase):
'public_key_template_attribute': None}
self.assertRaisesRegexp(
TypeError, "invalid public key unique identifier",
rekey_key_pair.RekeyKeyPairResponsePayload, **kwargs)
payloads.RekeyKeyPairResponsePayload, **kwargs)
def test_validate_with_invalid_private_key_template_attribute(self):
kwargs = {'private_key_uuid': None,
@ -255,7 +255,7 @@ class TestRekeyKeyPairResponsePayload(TestCase):
'public_key_template_attribute': None}
self.assertRaisesRegexp(
TypeError, "invalid private key template attribute",
rekey_key_pair.RekeyKeyPairResponsePayload, **kwargs)
payloads.RekeyKeyPairResponsePayload, **kwargs)
def test_validate_with_invalid_public_key_template_attribute(self):
kwargs = {'private_key_uuid': None,
@ -264,7 +264,7 @@ class TestRekeyKeyPairResponsePayload(TestCase):
'public_key_template_attribute': 'invalid'}
self.assertRaisesRegexp(
TypeError, "invalid public key template attribute",
rekey_key_pair.RekeyKeyPairResponsePayload, **kwargs)
payloads.RekeyKeyPairResponsePayload, **kwargs)
def _test_read(self, stream, payload, private_key_uuid, public_key_uuid,
private_key_template_attribute,
@ -297,14 +297,14 @@ class TestRekeyKeyPairResponsePayload(TestCase):
def test_read_with_none(self):
stream = self.encoding_empty
payload = rekey_key_pair.RekeyKeyPairResponsePayload()
payload = payloads.RekeyKeyPairResponsePayload()
self._test_read(stream, payload, self.empty_private_key_uuid,
self.empty_public_key_uuid, None, None)
def test_read_with_args(self):
stream = self.encoding_full
payload = rekey_key_pair.RekeyKeyPairResponsePayload(
payload = payloads.RekeyKeyPairResponsePayload(
self.private_key_uuid, self.public_key_uuid,
self.private_key_template_attribute,
self.public_key_template_attribute)
@ -332,13 +332,13 @@ class TestRekeyKeyPairResponsePayload(TestCase):
def test_write_with_none(self):
stream = utils.BytearrayStream()
payload = rekey_key_pair.RekeyKeyPairResponsePayload()
payload = payloads.RekeyKeyPairResponsePayload()
self._test_write(stream, payload, self.encoding_empty)
def test_write_with_args(self):
stream = utils.BytearrayStream()
payload = rekey_key_pair.RekeyKeyPairResponsePayload(
payload = payloads.RekeyKeyPairResponsePayload(
self.private_key_uuid, self.public_key_uuid,
self.private_key_template_attribute,
self.public_key_template_attribute)

View File

@ -21,7 +21,7 @@ from kmip.core import objects
from kmip.core import primitives
from kmip.core import utils
from kmip.core.messages.payloads import revoke
from kmip.core.messages import payloads
class TestRevokeRequestPayload(TestCase):
@ -55,14 +55,14 @@ class TestRevokeRequestPayload(TestCase):
Test that a RevokeRequestPayload object can be constructed with no
specified value.
"""
revoke.RevokeRequestPayload()
payloads.RevokeRequestPayload()
def test_init_with_args(self):
"""
Test that a RevokeRequestPayload object can be constructed with valid
values.
"""
revoke.RevokeRequestPayload(unique_identifier=self.uuid)
payloads.RevokeRequestPayload(unique_identifier=self.uuid)
def test_validate_with_bad_uuid_type(self):
"""
@ -71,7 +71,7 @@ class TestRevokeRequestPayload(TestCase):
"""
self.assertRaisesRegexp(
TypeError, "invalid unique identifier",
revoke.RevokeRequestPayload, "not-a-uuid")
payloads.RevokeRequestPayload, "not-a-uuid")
def test_validate_with_bad_date_type(self):
"""
@ -81,7 +81,7 @@ class TestRevokeRequestPayload(TestCase):
reason = objects.RevocationReason()
self.assertRaisesRegexp(
TypeError, "invalid compromise time",
revoke.RevokeRequestPayload, self.uuid, reason, "not-a-date")
payloads.RevokeRequestPayload, self.uuid, reason, "not-a-date")
def test_validate_with_bad_reason_type(self):
"""
@ -90,14 +90,14 @@ class TestRevokeRequestPayload(TestCase):
"""
self.assertRaisesRegexp(
TypeError, "invalid revocation reason",
revoke.RevokeRequestPayload, self.uuid, "not-a-reason")
payloads.RevokeRequestPayload, self.uuid, "not-a-reason")
def test_read_with_known_uuid(self):
"""
Test that a RevokeRequestPayload object with known UUID can be read
from a data stream.
"""
payload = revoke.RevokeRequestPayload()
payload = payloads.RevokeRequestPayload()
payload.read(self.encoding_a)
expected = '668eff89-3010-4258-bc0e-8c402309c746'
observed = payload.unique_identifier.value
@ -118,7 +118,7 @@ class TestRevokeRequestPayload(TestCase):
tag=enums.Tags.COMPROMISE_OCCURRENCE_DATE, value=6)
stream = utils.BytearrayStream()
payload = revoke.RevokeRequestPayload(
payload = payloads.RevokeRequestPayload(
unique_identifier=self.uuid,
revocation_reason=reason,
compromise_occurrence_date=date)
@ -165,14 +165,14 @@ class TestRevokeResponsePayload(TestCase):
Test that a RevokeResponsePayload object can be constructed with no
specified value.
"""
revoke.RevokeResponsePayload()
payloads.RevokeResponsePayload()
def test_init_with_args(self):
"""
Test that a RevokeResponsePayload object can be constructed with
valid values.
"""
revoke.RevokeResponsePayload(unique_identifier=self.uuid)
payloads.RevokeResponsePayload(unique_identifier=self.uuid)
def test_validate_with_invalid_uuid(self):
"""
@ -181,14 +181,14 @@ class TestRevokeResponsePayload(TestCase):
"""
self.assertRaisesRegexp(
TypeError, "invalid unique identifier",
revoke.RevokeResponsePayload, "not-a-uuid")
payloads.RevokeResponsePayload, "not-a-uuid")
def test_read_with_known_uuid(self):
"""
Test that a RevokeResponsePayload object with known UUID can be read
from a data stream.
"""
payload = revoke.RevokeResponsePayload()
payload = payloads.RevokeResponsePayload()
payload.read(self.encoding_a)
expected = '668eff89-3010-4258-bc0e-8c402309c746'
observed = payload.unique_identifier.value
@ -204,7 +204,7 @@ class TestRevokeResponsePayload(TestCase):
written to a data stream.
"""
stream = utils.BytearrayStream()
payload = revoke.RevokeResponsePayload(self.uuid)
payload = payloads.RevokeResponsePayload(self.uuid)
payload.write(stream)
length_expected = len(self.encoding_a)

View File

@ -43,12 +43,7 @@ from kmip.core import objects
from kmip.core.messages import contents
from kmip.core.messages import messages
from kmip.core.messages.payloads import create
from kmip.core.messages.payloads import get
from kmip.core.messages.payloads import register
from kmip.core.messages.payloads import locate
from kmip.core.messages.payloads import destroy
from kmip.core.messages.payloads import mac
from kmip.core.messages import payloads
from kmip.core.misc import KeyFormatType
from kmip.core.primitives import TextString
@ -247,8 +242,8 @@ class TestRequestMessage(TestCase):
request_payload = batch_item.request_payload
msg = "Bad request payload type: expected {0}, received {1}"
self.assertIsInstance(request_payload,
create.CreateRequestPayload,
msg.format(create.CreateRequestPayload,
payloads.CreateRequestPayload,
msg.format(payloads.CreateRequestPayload,
type(request_payload)))
object_type = request_payload.object_type
@ -384,8 +379,10 @@ class TestRequestMessage(TestCase):
temp_attr = objects.TemplateAttribute(attributes=[attr_a, attr_b,
attr_c])
req_pl = create.CreateRequestPayload(object_type=object_type,
template_attribute=temp_attr)
req_pl = payloads.CreateRequestPayload(
object_type=object_type,
template_attribute=temp_attr
)
batch_item = messages.RequestBatchItem(operation=operation,
request_payload=req_pl)
req_message = messages.RequestMessage(request_header=request_header,
@ -476,8 +473,8 @@ class TestRequestMessage(TestCase):
request_payload = batch_item.request_payload
msg = "Bad request payload type: expected {0}, received {1}"
self.assertIsInstance(request_payload,
get.GetRequestPayload,
msg.format(get.GetRequestPayload,
payloads.GetRequestPayload,
msg.format(payloads.GetRequestPayload,
type(request_payload)))
# unique_identifier = request_payload.unique_identifier
@ -501,7 +498,7 @@ class TestRequestMessage(TestCase):
operation = contents.Operation(enums.Operation.GET)
# uuid = attr.UniqueIdentifier('49a1ca88-6bea-4fb2-b450-7e58802c3038')
request_payload = get.GetRequestPayload(
request_payload = payloads.GetRequestPayload(
unique_identifier='49a1ca88-6bea-4fb2-b450-7e58802c3038'
)
batch_item = messages.RequestBatchItem(operation=operation,
@ -594,7 +591,7 @@ class TestRequestMessage(TestCase):
request_payload = batch_item.request_payload
msg = "Bad request payload type: expected {0}, received {1}"
exp_type = destroy.DestroyRequestPayload
exp_type = payloads.DestroyRequestPayload
rcv_type = type(request_payload)
self.assertIsInstance(request_payload, exp_type,
msg.format(exp_type, rcv_type))
@ -620,7 +617,9 @@ class TestRequestMessage(TestCase):
operation = contents.Operation(enums.Operation.DESTROY)
uuid = attr.UniqueIdentifier('fb4b5b9c-6188-4c63-8142-fe9c328129fc')
request_payload = destroy.DestroyRequestPayload(unique_identifier=uuid)
request_payload = payloads.DestroyRequestPayload(
unique_identifier=uuid
)
batch_item = messages.RequestBatchItem(operation=operation,
request_payload=request_payload)
request_message = messages.RequestMessage(request_header=req_header,
@ -711,7 +710,7 @@ class TestRequestMessage(TestCase):
request_payload = batch_item.request_payload
msg = "Bad request payload type: expected {0}, received {1}"
exp_type = register.RegisterRequestPayload
exp_type = payloads.RegisterRequestPayload
rcv_type = type(request_payload)
self.assertIsInstance(request_payload, exp_type,
msg.format(exp_type, rcv_type))
@ -817,7 +816,7 @@ class TestRequestMessage(TestCase):
template = Template(attributes=attributes)
request_payload = register.RegisterRequestPayload(
request_payload = payloads.RegisterRequestPayload(
object_type=object_type,
template_attribute=tmpl_attr,
secret=template)
@ -910,7 +909,7 @@ class TestRequestMessage(TestCase):
request_payload = batch_item.request_payload
msg = "Bad request payload type: expected {0}, received {1}"
exp_type = locate.LocateRequestPayload
exp_type = payloads.LocateRequestPayload
rcv_type = type(request_payload)
self.assertIsInstance(request_payload, exp_type,
msg.format(exp_type, rcv_type))
@ -1056,8 +1055,8 @@ class TestRequestMessage(TestCase):
request_payload = batch_item.request_payload
msg = "Bad request payload type: expected {0}, received {1}"
self.assertIsInstance(request_payload,
mac.MACRequestPayload,
msg.format(mac.MACRequestPayload,
payloads.MACRequestPayload,
msg.format(payloads.MACRequestPayload,
type(request_payload)))
unique_identifier = request_payload.unique_identifier
@ -1118,7 +1117,7 @@ class TestRequestMessage(TestCase):
parameters_attribute = attr.CryptographicParameters(
cryptographic_algorithm=enums.CryptographicAlgorithm.HMAC_SHA512
)
request_payload = mac.MACRequestPayload(
request_payload = payloads.MACRequestPayload(
unique_identifier=uuid,
cryptographic_parameters=parameters_attribute,
data=data
@ -1345,7 +1344,7 @@ class TestResponseMessage(TestCase):
result_status.value))
response_payload = batch_item.response_payload
exp_type = create.CreateResponsePayload
exp_type = payloads.CreateResponsePayload
rcv_type = type(response_payload)
self.assertIsInstance(response_payload, exp_type,
self.msg.format('response payload', 'type',
@ -1387,8 +1386,10 @@ class TestResponseMessage(TestCase):
uuid = 'fb4b5b9c-6188-4c63-8142-fe9c328129fc'
uniq_id = attr.UniqueIdentifier(uuid)
resp_pl = create.CreateResponsePayload(object_type=object_type,
unique_identifier=uniq_id)
resp_pl = payloads.CreateResponsePayload(
object_type=object_type,
unique_identifier=uniq_id
)
batch_item = messages.ResponseBatchItem(operation=operation,
result_status=result_status,
response_payload=resp_pl)
@ -1494,7 +1495,7 @@ class TestResponseMessage(TestCase):
result_status.value))
response_payload = batch_item.response_payload
exp_type = get.GetResponsePayload
exp_type = payloads.GetResponsePayload
rcv_type = type(response_payload)
self.assertIsInstance(response_payload, exp_type,
self.msg.format('response payload', 'type',
@ -1610,9 +1611,11 @@ class TestResponseMessage(TestCase):
secret = SymmetricKey(key_block)
resp_pl = get.GetResponsePayload(object_type=object_type.value,
unique_identifier=uniq_id.value,
secret=secret)
resp_pl = payloads.GetResponsePayload(
object_type=object_type.value,
unique_identifier=uniq_id.value,
secret=secret
)
batch_item = messages.ResponseBatchItem(operation=operation,
result_status=result_status,
response_payload=resp_pl)
@ -1726,7 +1729,7 @@ class TestResponseMessage(TestCase):
response_payload = batch_item.response_payload
msg = "Bad response payload type: expected {0}, received {1}"
exp_type = destroy.DestroyResponsePayload
exp_type = payloads.DestroyResponsePayload
rcv_type = type(response_payload)
self.assertIsInstance(response_payload, exp_type,
msg.format(exp_type, rcv_type))
@ -1757,7 +1760,7 @@ class TestResponseMessage(TestCase):
result_status = contents.ResultStatus(enums.ResultStatus.SUCCESS)
uuid = attr.UniqueIdentifier('fb4b5b9c-6188-4c63-8142-fe9c328129fc')
resp_pl = destroy.DestroyResponsePayload(unique_identifier=uuid)
resp_pl = payloads.DestroyResponsePayload(unique_identifier=uuid)
batch_item = messages.ResponseBatchItem(operation=operation,
result_status=result_status,
response_payload=resp_pl)
@ -1869,7 +1872,7 @@ class TestResponseMessage(TestCase):
response_payload = batch_item.response_payload
msg = "Bad response payload type: expected {0}, received {1}"
exp_type = register.RegisterResponsePayload
exp_type = payloads.RegisterResponsePayload
rcv_type = type(response_payload)
self.assertIsInstance(response_payload, exp_type,
msg.format(exp_type, rcv_type))
@ -1900,7 +1903,7 @@ class TestResponseMessage(TestCase):
result_status = contents.ResultStatus(enums.ResultStatus.SUCCESS)
uuid = attr.UniqueIdentifier('5c9b81ef-4ee5-42cd-ba2d-c002fdd0c7b3')
resp_pl = register.RegisterResponsePayload(unique_identifier=uuid)
resp_pl = payloads.RegisterResponsePayload(unique_identifier=uuid)
batch_item = messages.ResponseBatchItem(operation=operation,
result_status=result_status,
response_payload=resp_pl)
@ -1933,7 +1936,7 @@ class TestResponseMessage(TestCase):
result_status = contents.ResultStatus(enums.ResultStatus.SUCCESS)
uuid = attr.UniqueIdentifier('49a1ca88-6bea-4fb2-b450-7e58802c3038')
resp_pl = locate.LocateResponsePayload(unique_identifiers=[uuid])
resp_pl = payloads.LocateResponsePayload(unique_identifiers=[uuid])
batch_item = messages.ResponseBatchItem(operation=operation,
result_status=result_status,
@ -2041,7 +2044,7 @@ class TestResponseMessage(TestCase):
result_status.value))
response_payload = batch_item.response_payload
exp_type = mac.MACResponsePayload
exp_type = payloads.MACResponsePayload
rcv_type = type(response_payload)
self.assertIsInstance(response_payload, exp_type,
self.msg.format('response payload', 'type',
@ -2103,8 +2106,10 @@ class TestResponseMessage(TestCase):
b'\xff\x7c')
mac_data = objects.MACData(value)
resp_pl = mac.MACResponsePayload(unique_identifier=uniq_id,
mac_data=mac_data)
resp_pl = payloads.MACResponsePayload(
unique_identifier=uniq_id,
mac_data=mac_data
)
batch_item = messages.ResponseBatchItem(operation=operation,
result_status=result_status,
response_payload=resp_pl)

File diff suppressed because it is too large Load Diff

View File

@ -42,21 +42,7 @@ from kmip.core.messages.contents import ResultStatus
from kmip.core.messages.contents import ResultReason
from kmip.core.messages.contents import ResultMessage
from kmip.core.messages.contents import ProtocolVersion
from kmip.core.messages.payloads.create_key_pair import \
CreateKeyPairRequestPayload, CreateKeyPairResponsePayload
from kmip.core.messages.payloads import decrypt
from kmip.core.messages.payloads import derive_key
from kmip.core.messages.payloads.discover_versions import \
DiscoverVersionsRequestPayload, DiscoverVersionsResponsePayload
from kmip.core.messages.payloads import encrypt
from kmip.core.messages.payloads import get_attributes
from kmip.core.messages.payloads import get_attribute_list
from kmip.core.messages.payloads.query import \
QueryRequestPayload, QueryResponsePayload
from kmip.core.messages.payloads.rekey_key_pair import \
RekeyKeyPairRequestPayload, RekeyKeyPairResponsePayload
from kmip.core.messages.payloads import sign
from kmip.core.messages.payloads import signature_verify
from kmip.core.messages import payloads
from kmip.core.misc import Offset
from kmip.core.misc import QueryFunction
@ -236,8 +222,12 @@ class TestKMIPClient(TestCase):
payload = batch_item.request_payload
msg = base.format(CreateKeyPairRequestPayload, payload)
self.assertIsInstance(payload, CreateKeyPairRequestPayload, msg)
msg = base.format(payloads.CreateKeyPairRequestPayload, payload)
self.assertIsInstance(
payload,
payloads.CreateKeyPairRequestPayload,
msg
)
common_observed = payload.common_template_attribute
private_observed = payload.private_key_template_attribute
@ -285,8 +275,12 @@ class TestKMIPClient(TestCase):
payload = batch_item.request_payload
msg = base.format(RekeyKeyPairRequestPayload, payload)
self.assertIsInstance(payload, RekeyKeyPairRequestPayload, msg)
msg = base.format(payloads.RekeyKeyPairRequestPayload, payload)
self.assertIsInstance(
payload,
payloads.RekeyKeyPairRequestPayload,
msg
)
private_key_uuid_observed = payload.private_key_uuid
offset_observed = payload.offset
@ -342,8 +336,8 @@ class TestKMIPClient(TestCase):
if query_functions is None:
query_functions = list()
msg = base.format(QueryRequestPayload, payload)
self.assertIsInstance(payload, QueryRequestPayload, msg)
msg = base.format(payloads.QueryRequestPayload, payload)
self.assertIsInstance(payload, payloads.QueryRequestPayload, msg)
query_functions_observed = payload.query_functions
self.assertEqual(query_functions, query_functions_observed)
@ -378,8 +372,12 @@ class TestKMIPClient(TestCase):
if protocol_versions is None:
protocol_versions = list()
msg = base.format(DiscoverVersionsRequestPayload, payload)
self.assertIsInstance(payload, DiscoverVersionsRequestPayload, msg)
msg = base.format(payloads.DiscoverVersionsRequestPayload, payload)
self.assertIsInstance(
payload,
payloads.DiscoverVersionsRequestPayload,
msg
)
observed = payload.protocol_versions
@ -413,7 +411,7 @@ class TestKMIPClient(TestCase):
)
self.assertIsInstance(
batch_item.request_payload,
get_attributes.GetAttributesRequestPayload
payloads.GetAttributesRequestPayload
)
self.assertEqual(uuid, batch_item.request_payload.unique_identifier)
self.assertEqual(
@ -431,13 +429,13 @@ class TestKMIPClient(TestCase):
OperationEnum.GET_ATTRIBUTE_LIST, batch_item.operation.value)
self.assertIsInstance(
batch_item.request_payload,
get_attribute_list.GetAttributeListRequestPayload)
payloads.GetAttributeListRequestPayload)
self.assertEqual(uid, batch_item.request_payload.unique_identifier)
def test_process_batch_items(self):
batch_item = ResponseBatchItem(
operation=Operation(OperationEnum.CREATE_KEY_PAIR),
response_payload=CreateKeyPairResponsePayload())
response_payload=payloads.CreateKeyPairResponsePayload())
response = ResponseMessage(batch_items=[batch_item, batch_item])
results = self.client._process_batch_items(response)
@ -522,7 +520,7 @@ class TestKMIPClient(TestCase):
def test_process_create_key_pair_batch_item(self):
batch_item = ResponseBatchItem(
operation=Operation(OperationEnum.CREATE_KEY_PAIR),
response_payload=CreateKeyPairResponsePayload())
response_payload=payloads.CreateKeyPairResponsePayload())
result = self.client._process_create_key_pair_batch_item(batch_item)
msg = "expected {0}, received {1}".format(CreateKeyPairResult, result)
@ -531,7 +529,7 @@ class TestKMIPClient(TestCase):
def test_process_rekey_key_pair_batch_item(self):
batch_item = ResponseBatchItem(
operation=Operation(OperationEnum.REKEY_KEY_PAIR),
response_payload=RekeyKeyPairResponsePayload())
response_payload=payloads.RekeyKeyPairResponsePayload())
result = self.client._process_rekey_key_pair_batch_item(batch_item)
msg = "expected {0}, received {1}".format(RekeyKeyPairResult, result)
@ -546,7 +544,7 @@ class TestKMIPClient(TestCase):
application_namespaces,
extension_information):
payload = QueryResponsePayload(
payload = payloads.QueryResponsePayload(
operations,
object_types,
vendor_identification,
@ -598,7 +596,7 @@ class TestKMIPClient(TestCase):
def _test_process_discover_versions_batch_item(self, protocol_versions):
batch_item = ResponseBatchItem(
operation=Operation(OperationEnum.DISCOVER_VERSIONS),
response_payload=DiscoverVersionsResponsePayload(
response_payload=payloads.DiscoverVersionsResponsePayload(
protocol_versions))
result = self.client._process_discover_versions_batch_item(batch_item)
@ -624,7 +622,7 @@ class TestKMIPClient(TestCase):
def test_process_get_attributes_batch_item(self):
uuid = '00000000-1111-2222-3333-444444444444'
attributes = []
payload = get_attributes.GetAttributesResponsePayload(
payload = payloads.GetAttributesResponsePayload(
unique_identifier=uuid,
attributes=attributes
)
@ -641,7 +639,7 @@ class TestKMIPClient(TestCase):
def test_process_get_attribute_list_batch_item(self):
uid = '00000000-1111-2222-3333-444444444444'
names = ['Cryptographic Algorithm', 'Cryptographic Length']
payload = get_attribute_list.GetAttributeListResponsePayload(
payload = payloads.GetAttributeListResponsePayload(
unique_identifier=uid, attribute_names=names)
batch_item = ResponseBatchItem(
operation=Operation(OperationEnum.GET_ATTRIBUTE_LIST),
@ -735,7 +733,7 @@ class TestKMIPClient(TestCase):
"""
Test that the client can derive a key.
"""
payload = derive_key.DeriveKeyResponsePayload(
payload = payloads.DeriveKeyResponsePayload(
unique_identifier='1',
)
batch_item = ResponseBatchItem(
@ -793,7 +791,7 @@ class TestKMIPClient(TestCase):
"""
Test that the client can encrypt data.
"""
payload = encrypt.EncryptResponsePayload(
payload = payloads.EncryptResponsePayload(
unique_identifier='1',
data=(
b'\x6B\x77\xB4\xD6\x30\x06\xDE\xE6'
@ -856,7 +854,7 @@ class TestKMIPClient(TestCase):
"""
Test that the client can decrypt data.
"""
payload = decrypt.DecryptResponsePayload(
payload = payloads.DecryptResponsePayload(
unique_identifier='1',
data=(
b'\x37\x36\x35\x34\x33\x32\x31\x20'
@ -918,7 +916,7 @@ class TestKMIPClient(TestCase):
"""
Test that the client can verify a signature.
"""
payload = signature_verify.SignatureVerifyResponsePayload(
payload = payloads.SignatureVerifyResponsePayload(
unique_identifier='1',
validity_indicator=enums.ValidityIndicator.INVALID
)
@ -972,7 +970,7 @@ class TestKMIPClient(TestCase):
"""
Test that the client can sign data
"""
payload = sign.SignResponsePayload(
payload = payloads.SignResponsePayload(
unique_identifier='1',
signature_data=b'aaaaaaaaaaaaaaaa'
)