From 664c9964c8ca147bfa4841f2055442ce6f4b490f Mon Sep 17 00:00:00 2001 From: Peter Hamilton Date: Tue, 17 Apr 2018 00:32:20 -0400 Subject: [PATCH] Add Check support to the payload factories This change adds Check payload support to the payload factories. Payload factory unit tests have been updated to account for the change. --- kmip/core/factories/payloads/request.py | 3 +++ kmip/core/factories/payloads/response.py | 3 +++ kmip/tests/unit/core/factories/payloads/test_request.py | 3 ++- kmip/tests/unit/core/factories/payloads/test_response.py | 3 ++- 4 files changed, 10 insertions(+), 2 deletions(-) diff --git a/kmip/core/factories/payloads/request.py b/kmip/core/factories/payloads/request.py index e2b18b2..4d0c9f4 100644 --- a/kmip/core/factories/payloads/request.py +++ b/kmip/core/factories/payloads/request.py @@ -40,6 +40,9 @@ class RequestPayloadFactory(PayloadFactory): def _create_locate_payload(self): return payloads.LocateRequestPayload() + def _create_check_payload(self): + return payloads.CheckRequestPayload() + def _create_get_payload(self): return payloads.GetRequestPayload() diff --git a/kmip/core/factories/payloads/response.py b/kmip/core/factories/payloads/response.py index 3b20df5..3b9c5a5 100644 --- a/kmip/core/factories/payloads/response.py +++ b/kmip/core/factories/payloads/response.py @@ -40,6 +40,9 @@ class ResponsePayloadFactory(PayloadFactory): def _create_locate_payload(self): return payloads.LocateResponsePayload() + def _create_check_payload(self): + return payloads.CheckResponsePayload() + def _create_get_payload(self): return payloads.GetResponsePayload() diff --git a/kmip/tests/unit/core/factories/payloads/test_request.py b/kmip/tests/unit/core/factories/payloads/test_request.py index 2ef763c..035d204 100644 --- a/kmip/tests/unit/core/factories/payloads/test_request.py +++ b/kmip/tests/unit/core/factories/payloads/test_request.py @@ -77,7 +77,8 @@ class TestRequestPayloadFactory(testtools.TestCase): self._test_payload_type(payload, payloads.LocateRequestPayload) def test_create_check_payload(self): - self._test_not_implemented(self.factory.create, enums.Operation.CHECK) + payload = self.factory.create(enums.Operation.CHECK) + self._test_payload_type(payload, payloads.CheckRequestPayload) def test_create_get_payload(self): payload = self.factory.create(enums.Operation.GET) diff --git a/kmip/tests/unit/core/factories/payloads/test_response.py b/kmip/tests/unit/core/factories/payloads/test_response.py index 071b509..46af787 100644 --- a/kmip/tests/unit/core/factories/payloads/test_response.py +++ b/kmip/tests/unit/core/factories/payloads/test_response.py @@ -77,7 +77,8 @@ class TestResponsePayloadFactory(testtools.TestCase): self._test_payload_type(payload, payloads.LocateResponsePayload) def test_create_check_payload(self): - self._test_not_implemented(self.factory.create, enums.Operation.CHECK) + payload = self.factory.create(enums.Operation.CHECK) + self._test_payload_type(payload, payloads.CheckResponsePayload) def test_create_get_payload(self): payload = self.factory.create(enums.Operation.GET)