B64url: No padding

Author: bjoern-otto

aas_test_engines/data_types.py

@@ -407,3 +407,7 @@ class DataTypeDefXsd(Enum):
     DataTypeDefXsd.base64Binary: is_base64_binary,
     DataTypeDefXsd.hexBinary: is_hex_binary,
 }
+
+
+def base64_urlsafe(s: str) -> str:
+    return base64.urlsafe_b64encode(s.encode()).decode().strip("=")

aas_test_engines/reflect.py

@@ -13,7 +13,7 @@
 from dataclasses import is_dataclass, fields
 import re
 import inspect
-import base64
+from aas_test_engines.data_types import base64_urlsafe
 
 
 def abstract(cls):
@@ -88,7 +88,7 @@ def __eq__(self, other: "StringFormattedValue") -> bool:
 
     def __str__(self) -> str:
         if self.base64:
-            return base64.urlsafe_b64encode(self.raw_value.encode()).decode()
+            return base64_urlsafe(self.raw_value)
         else:
             return self.raw_value
 

aas_test_engines/test_cases/v3_0/interfaces/aas.py

@@ -9,7 +9,6 @@
     ApiTestSuite,
     invoke,
     PagedResult,
-    Base64String,
     PaginationTests,
 )
 from aas_test_engines.test_cases.v3_0.model import (

aas_test_engines/test_cases/v3_0/interfaces/shared.py

@@ -19,7 +19,7 @@
 import base64
 import json
 import requests
-from fences.core.util import ConfusionMatrix
+from aas_test_engines.data_types import base64_urlsafe
 
 # Util
 
@@ -136,7 +136,8 @@ class AssetId:
     value: str
 
     def __str__(self):
-        return base64.b64encode(json.dumps({"name": self.name, "value": self.value}).encode()).decode()
+        data = {"name": self.name, "value": self.value}
+        return base64_urlsafe(json.dumps(data))
 
 
 class ApiTestSuite:

test/test_data_types.py

@@ -0,0 +1,27 @@
+from unittest import TestCase
+from aas_test_engines import data_types
+from base64 import urlsafe_b64decode
+
+
+def b64urlsafe_decode(s: str) -> str:
+    # Append missing padding
+    padding = 4 - len(s) % 4
+    if padding != 4:
+        s += "=" * padding
+    return urlsafe_b64decode(s).decode()
+
+
+class Base64UrlTest(TestCase):
+
+    def test_all(self):
+        for value in [
+            "",
+            " ",
+            "=",
+            "ab",
+            "abc",
+            "#" * 80,
+        ]:
+            encoded = data_types.base64_urlsafe(value)
+            self.assertNotIn("=", encoded)
+            self.assertEqual(b64urlsafe_decode(encoded), value)