Add tests for missing profile

Author: Vasile Zaremba

yoti_python_sdk/activity_details.py

@@ -3,7 +3,7 @@
 
 
 class ActivityDetails:
-    def __init__(self, receipt, decrypted_profile = None):
+    def __init__(self, receipt, decrypted_profile=None):
         self.decrypted_profile = decrypted_profile
         self.user_profile = {}
 

yoti_python_sdk/tests/fixtures/response_empty_profile.txt

@@ -1,17 +1,36 @@
 from uuid import UUID
 
 
-def mocked_requests_get(*args, **kwargs):
-    class MockResponse:
-        def __init__(self, status_code, text):
-            self.status_code = status_code
-            self.text = text
+class MockResponse:
+    def __init__(self, status_code, text):
+        self.status_code = status_code
+        self.text = text
+
 
+def mocked_requests_get(*args, **kwargs):
     with open('yoti_python_sdk/tests/fixtures/response.txt', 'r') as f:
         response = f.read()
     return MockResponse(status_code=200, text=response)
 
 
+def mocked_requests_get_null_profile(*args, **kwargs):
+    with open('yoti_python_sdk/tests/fixtures/response_null_profile.txt', 'r') as f:
+        response = f.read()
+    return MockResponse(status_code=200, text=response)
+
+
+def mocked_requests_get_empty_profile(*args, **kwargs):
+    with open('yoti_python_sdk/tests/fixtures/response_empty_profile.txt', 'r') as f:
+        response = f.read()
+    return MockResponse(status_code=200, text=response)
+
+
+def mocked_requests_get_missing_profile(*args, **kwargs):
+    with open('yoti_python_sdk/tests/fixtures/response_missing_profile.txt', 'r') as f:
+        response = f.read()
+    return MockResponse(status_code=200, text=response)
+
+
 def mocked_timestamp():
     return 1476441361.2395663
 

yoti_python_sdk/tests/fixtures/response_missing_profile.txt

@@ -18,6 +18,9 @@
 from yoti_python_sdk.tests.conftest import YOTI_CLIENT_SDK_ID, PEM_FILE_PATH
 from yoti_python_sdk.tests.mocks import (
     mocked_requests_get,
+    mocked_requests_get_null_profile,
+    mocked_requests_get_empty_profile,
+    mocked_requests_get_missing_profile,
     mocked_timestamp,
     mocked_uuid4
 )
@@ -105,6 +108,46 @@ def test_requesting_activity_details_with_correct_data(
 
     mock_get.assert_called_once_with(url=expected_url, headers=expected_headers)
     assert isinstance(activity_details, ActivityDetails)
+    assert activity_details.user_id == "ijH4kkqMKTG0FSNUgQIvd2Z3Nx1j8f5RjVQMyoKOvO/hkv43Ik+t6d6mGfP2tdrN"
     selfie = activity_details.user_profile.get('selfie')
     assert isinstance(selfie, basestring)
     assert selfie.startswith('data:image/jpeg;base64')
+
+
+@mock.patch('requests.get', side_effect=mocked_requests_get_null_profile)
+@mock.patch('time.time', side_effect=mocked_timestamp)
+@mock.patch('uuid.uuid4', side_effect=mocked_uuid4)
+def test_requesting_activity_details_with_null_profile(
+        mock_uuid4, mock_time, mock_get, client, expected_url,
+        expected_headers, encrypted_request_token):
+    activity_details = client.get_activity_details(encrypted_request_token)
+
+    mock_get.assert_called_once_with(url=expected_url, headers=expected_headers)
+    assert activity_details.user_id == "ijH4kkqMKTG0FSNUgQIvd2Z3Nx1j8f5RjVQMyoKOvO/hkv43Ik+t6d6mGfP2tdrN"
+    assert isinstance(activity_details, ActivityDetails)
+
+
+@mock.patch('requests.get', side_effect=mocked_requests_get_empty_profile)
+@mock.patch('time.time', side_effect=mocked_timestamp)
+@mock.patch('uuid.uuid4', side_effect=mocked_uuid4)
+def test_requesting_activity_details_with_empty_profile(
+        mock_uuid4, mock_time, mock_get, client, expected_url,
+        expected_headers, encrypted_request_token):
+    activity_details = client.get_activity_details(encrypted_request_token)
+
+    mock_get.assert_called_once_with(url=expected_url, headers=expected_headers)
+    assert activity_details.user_id == "ijH4kkqMKTG0FSNUgQIvd2Z3Nx1j8f5RjVQMyoKOvO/hkv43Ik+t6d6mGfP2tdrN"
+    assert isinstance(activity_details, ActivityDetails)
+
+
+@mock.patch('requests.get', side_effect=mocked_requests_get_missing_profile)
+@mock.patch('time.time', side_effect=mocked_timestamp)
+@mock.patch('uuid.uuid4', side_effect=mocked_uuid4)
+def test_requesting_activity_details_with_missing_profile(
+        mock_uuid4, mock_time, mock_get, client, expected_url,
+        expected_headers, encrypted_request_token):
+    activity_details = client.get_activity_details(encrypted_request_token)
+
+    mock_get.assert_called_once_with(url=expected_url, headers=expected_headers)
+    assert activity_details.user_id == "ijH4kkqMKTG0FSNUgQIvd2Z3Nx1j8f5RjVQMyoKOvO/hkv43Ik+t6d6mGfP2tdrN"
+    assert isinstance(activity_details, ActivityDetails)

yoti_python_sdk/tests/fixtures/response_null_profile.txt

@@ -32,4 +32,3 @@ def test_given_proper_encrypted_token__decrypting_should_yield_decrypted_token(
 ])
 def test_strip_pkcs5_padding(with_padding, stripped):
     assert Crypto.strip_pkcs5_padding(with_padding) == stripped
-