Merge pull request #425 from OpenKMIP/feat/add-rekey-to-factory

Add Rekey support to the payload factories
This commit is contained in:
Peter Hamilton 2018-04-17 00:25:35 -04:00 committed by GitHub
commit 72280f8ddc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 10 additions and 2 deletions

View File

@ -31,6 +31,9 @@ class RequestPayloadFactory(PayloadFactory):
def _create_derive_key_payload(self): def _create_derive_key_payload(self):
return payloads.DeriveKeyRequestPayload() return payloads.DeriveKeyRequestPayload()
def _create_rekey_payload(self):
return payloads.RekeyRequestPayload()
def _create_rekey_key_pair_payload(self): def _create_rekey_key_pair_payload(self):
return payloads.RekeyKeyPairRequestPayload() return payloads.RekeyKeyPairRequestPayload()

View File

@ -31,6 +31,9 @@ class ResponsePayloadFactory(PayloadFactory):
def _create_derive_key_payload(self): def _create_derive_key_payload(self):
return payloads.DeriveKeyResponsePayload() return payloads.DeriveKeyResponsePayload()
def _create_rekey_payload(self):
return payloads.RekeyResponsePayload()
def _create_rekey_key_pair_payload(self): def _create_rekey_key_pair_payload(self):
return payloads.RekeyKeyPairResponsePayload() return payloads.RekeyKeyPairResponsePayload()

View File

@ -53,7 +53,8 @@ class TestRequestPayloadFactory(testtools.TestCase):
self._test_payload_type(payload, payloads.RegisterRequestPayload) self._test_payload_type(payload, payloads.RegisterRequestPayload)
def test_create_rekey_payload(self): def test_create_rekey_payload(self):
self._test_not_implemented(self.factory.create, enums.Operation.REKEY) payload = self.factory.create(enums.Operation.REKEY)
self._test_payload_type(payload, payloads.RekeyRequestPayload)
def test_create_derive_key_payload(self): def test_create_derive_key_payload(self):
payload = self.factory.create(enums.Operation.DERIVE_KEY) payload = self.factory.create(enums.Operation.DERIVE_KEY)

View File

@ -53,7 +53,8 @@ class TestResponsePayloadFactory(testtools.TestCase):
self._test_payload_type(payload, payloads.RegisterResponsePayload) self._test_payload_type(payload, payloads.RegisterResponsePayload)
def test_create_rekey_payload(self): def test_create_rekey_payload(self):
self._test_not_implemented(self.factory.create, enums.Operation.REKEY) payload = self.factory.create(enums.Operation.REKEY)
self._test_payload_type(payload, payloads.RekeyResponsePayload)
def test_create_derive_key_payload(self): def test_create_derive_key_payload(self):
payload = self.factory.create(enums.Operation.DERIVE_KEY) payload = self.factory.create(enums.Operation.DERIVE_KEY)