mirror of
https://github.com/OpenKMIP/PyKMIP.git
synced 2025-04-08 19:25:06 +02:00
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:
parent
7f6f4d004f
commit
82a7b8a741
@ -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()
|
||||
|
@ -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()
|
||||
|
@ -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"
|
||||
]
|
||||
|
@ -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))
|
||||
|
@ -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
|
||||
)
|
||||
|
@ -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(
|
||||
|
@ -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):
|
||||
|
@ -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)
|
||||
|
@ -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)
|
||||
|
@ -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'
|
||||
)
|
||||
|
@ -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
|
||||
)
|
||||
|
@ -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
|
||||
|
||||
|
@ -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'
|
||||
|
@ -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
|
||||
|
@ -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
|
||||
)
|
||||
|
@ -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
|
||||
)
|
||||
|
@ -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,)
|
||||
|
@ -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)
|
||||
|
@ -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)
|
||||
|
@ -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)
|
||||
|
@ -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
@ -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'
|
||||
)
|
||||
|
Loading…
x
Reference in New Issue
Block a user