Skip to content

Commit

Permalink
Added missing mock method to test_client (#40)
Browse files Browse the repository at this point in the history
  • Loading branch information
Andrew Lum committed Feb 23, 2017
1 parent 61ebc3f commit 0812bcf
Showing 1 changed file with 10 additions and 3 deletions.
13 changes: 10 additions & 3 deletions src/test/python/test_client.py
Expand Up @@ -110,6 +110,9 @@ def test_create_secret(mocker, client, api_client, private_key):
mocker.patch('covata.delta.crypto.generate_initialisation_vector',
return_value=iv)

mocker.patch('covata.delta.crypto.encrypt',
return_value=bytes('encrypted secret'.encode('utf-8')))

mocker.patch('covata.delta.crypto.encrypt_key_with_public_key',
return_value=encrypted_key)

Expand All @@ -126,7 +129,8 @@ def test_create_secret(mocker, client, api_client, private_key):
initialisationVector=iv,
symmetricKey=encrypted_key)))

secret = client.create_secret(created_by_id, "this is my secret")
secret = client.create_secret(created_by_id,
"this is my secret".encode('utf-8'))

assert secret.parent == client
assert secret.id == expected_id
Expand Down Expand Up @@ -161,6 +165,9 @@ def test_create_secret_via_identity(mocker, client, api_client, private_key):
mocker.patch('covata.delta.crypto.generate_initialisation_vector',
return_value=iv)

mocker.patch('covata.delta.crypto.encrypt',
return_value=bytes('encrypted secret'.encode('utf-8')))

mocker.patch('covata.delta.crypto.encrypt_key_with_public_key',
return_value=encrypted_key)

Expand All @@ -179,12 +186,12 @@ def test_create_secret_via_identity(mocker, client, api_client, private_key):

identity = client.get_identity(created_by_id)

secret = identity.create_secret("this is my secret")
secret = identity.create_secret("this is my secret".encode('utf-8'))

assert secret.parent == client
assert secret.id == expected_id
assert secret.created == "12345"
assert secret.rsa_key_owner == created_by_id
assert secret.created_by == created_by_id
assert secret.encryption_details.initialisation_vector == iv
assert secret.encryption_details.symmetric_key == encrypted_key
assert secret.encryption_details.symmetric_key == encrypted_key

0 comments on commit 0812bcf

Please sign in to comment.