SDK-1260: Add ThirdPartyAttributeExtension

Author: emmas-yoti

yoti_python_sdk/dynamic_sharing_service/extension/third_party_attribute_extension.py

@@ -0,0 +1,23 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+import copy
+
+
+class ThirdPartyAttributeExtension(object):
+    THIRDPARTY_ATTRIBUTE = "THIRD_PARTY_ATTRIBUTE"
+
+    def __init__(self):
+        self.__extension = {}
+        self.__extension["type"] = self.THIRDPARTY_ATTRIBUTE
+        self.__extension["content"] = {"expiry_date": None, "definitions": []}
+
+    def with_expiry_date(self, expiry_date):
+        self.__extension["content"]["expiry_date"] = expiry_date.isoformat()
+        return self
+
+    def with_definitions(self, *names):
+        self.__extension["content"]["definitions"].extend([{"name": s} for s in names])
+        return self
+
+    def build(self):
+        return copy.deepcopy(self.__extension)

yoti_python_sdk/tests/dynamic_sharing_service/extension/test_third_party_attribute_extension.py

@@ -0,0 +1,64 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from datetime import datetime
+from yoti_python_sdk.dynamic_sharing_service.extension.third_party_attribute_extension import (
+    ThirdPartyAttributeExtension,
+)
+
+
+def test_should_create_extension():
+    DEFINITION = "some_value"
+    expiry_date = datetime(2019, 10, 30, 12, 10, 9, int(458e3))
+
+    extension = (
+        ThirdPartyAttributeExtension()
+        .with_expiry_date(expiry_date)
+        .with_definitions(DEFINITION)
+        .build()
+    )
+
+    assert extension["type"] == ThirdPartyAttributeExtension.THIRDPARTY_ATTRIBUTE
+    assert extension["content"]["expiry_date"] == "2019-10-30T12:10:09.458000"
+    assert extension["content"]["definitions"][0]["name"] == DEFINITION
+
+
+def test_with_definition_should_add_to_list():
+    DEFINITION1 = "some_attribute"
+    DEFINITION2 = "some_other_attribute"
+
+    expiry_date = datetime(2019, 10, 30, 12, 10, 9, int(458e3))
+
+    extension = (
+        ThirdPartyAttributeExtension()
+        .with_expiry_date(expiry_date)
+        .with_definitions(DEFINITION1)
+        .with_definitions(DEFINITION2)
+        .build()
+    )
+
+    assert extension["type"] == ThirdPartyAttributeExtension.THIRDPARTY_ATTRIBUTE
+    assert extension["content"]["expiry_date"] == "2019-10-30T12:10:09.458000"
+
+    assert extension["content"]["definitions"][0]["name"] == DEFINITION1
+    assert extension["content"]["definitions"][1]["name"] == DEFINITION2
+
+
+def test_with_definition_should_add_multiple():
+    DEFINITION1 = "some_attribute"
+    DEFINITION2 = "some_other_attribute"
+
+    expiry_date = datetime(2019, 10, 30, 12, 10, 9, int(458e3))
+
+    extension = (
+        ThirdPartyAttributeExtension()
+        .with_expiry_date(expiry_date)
+        .with_definitions(DEFINITION1, DEFINITION2)
+        .build()
+    )
+
+    assert extension["type"] == ThirdPartyAttributeExtension.THIRDPARTY_ATTRIBUTE
+    assert extension["content"]["expiry_date"] == "2019-10-30T12:10:09.458000"
+
+    assert extension["content"]["definitions"][0]["name"] == DEFINITION1
+    assert extension["content"]["definitions"][1]["name"] == DEFINITION2