Merge pull request #262 from OpenKMIP/bug/fix-pie-client-get

Fixing stringent uid value checks in the ProxyKmipClient
This commit is contained in:
Peter Hamilton 2017-04-12 18:31:38 -04:00 committed by GitHub
commit 19560b92b7
2 changed files with 9 additions and 7 deletions

View File

@ -391,7 +391,7 @@ class ProxyKmipClient(api.KmipClient):
message = result.result_message.value message = result.result_message.value
raise exceptions.KmipOperationFailure(status, reason, message) raise exceptions.KmipOperationFailure(status, reason, message)
def get(self, uid): def get(self, uid=None):
""" """
Get a managed object from a KMIP appliance. Get a managed object from a KMIP appliance.
@ -407,6 +407,7 @@ class ProxyKmipClient(api.KmipClient):
TypeError: if the input argument is invalid TypeError: if the input argument is invalid
""" """
# Check input # Check input
if uid is not None:
if not isinstance(uid, six.string_types): if not isinstance(uid, six.string_types):
raise TypeError("uid must be a string") raise TypeError("uid must be a string")
@ -541,7 +542,7 @@ class ProxyKmipClient(api.KmipClient):
message = result.result_message.value message = result.result_message.value
raise exceptions.KmipOperationFailure(status, reason, message) raise exceptions.KmipOperationFailure(status, reason, message)
def destroy(self, uid): def destroy(self, uid=None):
""" """
Destroy a managed object stored by a KMIP appliance. Destroy a managed object stored by a KMIP appliance.
@ -557,6 +558,7 @@ class ProxyKmipClient(api.KmipClient):
TypeError: if the input argument is invalid TypeError: if the input argument is invalid
""" """
# Check input # Check input
if uid is not None:
if not isinstance(uid, six.string_types): if not isinstance(uid, six.string_types):
raise TypeError("uid must be a string") raise TypeError("uid must be a string")

View File

@ -357,7 +357,7 @@ class KMIPProxy(KMIP):
revocation_message=message, revocation_message=message,
credential=credential) credential=credential)
def destroy(self, uuid, credential=None): def destroy(self, uuid=None, credential=None):
return self._destroy(unique_identifier=uuid, return self._destroy(unique_identifier=uuid,
credential=credential) credential=credential)