chore: Add validation if token was generated

pprzetacznik · Dec 31, 2023 · 44abf2f · 44abf2f
1 parent fd577a3
commit 44abf2f
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 4 deletions.
diff --git a/ksef_utils/server.py b/ksef_utils/server.py
@@ -147,7 +147,7 @@ def generate_token(self, session_token):
                         "roleDescription": "write invoices",
                     },
                 ],
-                "description": "token_to_grant_access",
+                "description": "0_ksef-utils_test_token",
             }
         }
         response = requests.post(
@@ -161,6 +161,17 @@ def generate_token(self, session_token):
         )
         return response
 
+    def get_token_status(self, session_token, reference_number):
+        response = requests.get(
+            f"{self.config.URL}/api/online/Credentials/Status/{reference_number}",
+            headers={
+                "Content-Type": "application/json",
+                "SessionToken": session_token,
+                "Accept": "application/json",
+            },
+        )
+        return response
+
     def get_invoice_status(self, reference_number: str, session_token: str):
         headers = {
             "accept": "application/json",
@@ -304,6 +315,21 @@ def generate_token(self):
         response = self.server.generate_token(self.init_token)
         return response.json()
 
+    def wait_until_token(self, element_reference_number):
+        response = self.server.get_token_status(
+            self.init_token, element_reference_number
+        )
+        processing_code = 302
+        while processing_code != 200:
+            response_status = self.server.get_token_status(
+                self.init_token, element_reference_number
+            )
+            processing_code = response_status.json().get("processingCode")
+            print(response)
+            if processing_code != 200:
+                sleep(1)
+        return response.json()
+
 
 class KSEFUtils:
     @staticmethod

diff --git a/tests/test_e2e.py b/tests/test_e2e.py
@@ -25,11 +25,12 @@ def when_generate_token(config, service, testing_context):
     print(response)
     print(response.get("authorisationToken"))
     testing_context["authorisationToken"] = response.get("authorisationToken")
+    service.wait_until_token(response.get("elementReferenceNumber"))
 
 
 @then("sign in using token")
 def then_sign_in_token(config, service, testing_context):
     print(dumps(testing_context, indent=4))
-    # config.KSEF_TOKEN = testing_context.get("authorisationToken")
-    # response = service.init_session()
-    # print(response)
+    config.KSEF_TOKEN = testing_context.get("authorisationToken")
+    response = service.init_session()
+    print(response)