feat: Сборка окончания связки аккаунтов

#3
KrapivinAndrey · May 4, 2022 · 2ee70c4 · 2ee70c4
1 parent 61e7f8a
commit 2ee70c4
Show file tree

Hide file tree

Showing 2 changed files with 67 additions and 20 deletions.
diff --git a/alicefluentcheck/AliceRequest.py b/alicefluentcheck/AliceRequest.py
@@ -25,6 +25,7 @@ def __init__(
         self._state_application = {}
         self._interfaces = {}
         self._access_token = None
+        self._account_linking_complete_event = False
         if has_screen:
             self._interfaces["screen"] = {}
         if has_payments:
@@ -80,6 +81,10 @@ def add_to_state_application(self, name: str, value):
     def from_scene(self, scene: str):
         self.add_to_state_session("scene", scene)
 
+    @chained
+    def account_linking_complete(self):
+        self._account_linking_complete_event = True
+
     def build(self):
         def meta():
             return {
@@ -104,25 +109,39 @@ def session(new=False):
 
             return temp
 
-        req = {
-            "meta": meta(),
-            "session": session(self._new_session),
-            "request": {
-                "command": self._command,
-                "original_utterance": self._original_utterance,
-                "nlu": {
-                    "tokens": self._nlu_tokens,
-                    "entities": self._entities,
-                    "intents": self._intents,
+        if self._account_linking_complete_event:
+            req = {
+                "meta": meta(),
+                "session": session(self._new_session),
+                "account_linking_complete_event": {},
+                "version": "1.0",
+                "state": {
+                    "session": self._state_sessions,
+                    "user": self._state_user,
+                    "applications": self._state_application,
                 },
-                "markup": {"dangerous_context": False},
-                "type": "SimpleUtterance",
-            },
-            "version": "1.0",
-            "state": {
-                "session": self._state_sessions,
-                "user": self._state_user,
-                "applications": self._state_application,
-            },
-        }
+            }
+
+        else:
+            req = {
+                "meta": meta(),
+                "session": session(self._new_session),
+                "request": {
+                    "command": self._command,
+                    "original_utterance": self._original_utterance,
+                    "nlu": {
+                        "tokens": self._nlu_tokens,
+                        "entities": self._entities,
+                        "intents": self._intents,
+                    },
+                    "markup": {"dangerous_context": False},
+                    "type": "SimpleUtterance",
+                },
+                "version": "1.0",
+                "state": {
+                    "session": self._state_sessions,
+                    "user": self._state_user,
+                    "applications": self._state_application,
+                },
+            }
         return req
diff --git a/tests/test_request.py b/tests/test_request.py
@@ -78,6 +78,34 @@ def test_create_token(self):
         }
         assert test == control
 
+    def test_acount_link(self):
+        test = AliceRequest().account_linking_complete().access_token("000").build()
+        control = {
+            "meta": {
+                "locale": "ru-RU",
+                "timezone": "UTC",
+                "client_id": "AliceMock",
+                "interfaces": {},
+            },
+            "session": {
+                "message_id": 3,
+                "session_id": "d825cbef-e7d6-4af9-9810-3ff3f358ac16",
+                "skill_id": "3308dc06-b901-4f7e-8882-beb1b84c0753",
+                "user": {"user_id": "000", "access_token": "000"},
+                "application": {"application_id": "000"},
+                "user_id": "000",
+                "new": False,
+            },
+            "account_linking_complete_event": {},
+            "version": "1.0",
+            "state": {
+                "session": {},
+                "user": {},
+                "applications": {},
+            },
+        }
+        assert test == control
+
     def test_add_entity(self):
         fio = AliceEntity().fio("Фамилия", "Имя", "Отчество")
         test = AliceRequest().command("Фамилия Имя Отчество").add_entity(fio).build()