Improve cluster_id generation

testcontainers · Jun 21, 2024 · 791d6e4 · 791d6e4
1 parent aa9a1ad
commit 791d6e4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/modules/kafka/testcontainers/kafka/__init__.py b/modules/kafka/testcontainers/kafka/__init__.py
@@ -85,9 +85,9 @@ def with_cluster_id(self, cluster_id: str) -> Self:
     def _random_uuid(cls):
         uuid_value = uuid.uuid4()
         uuid_bytes = uuid_value.bytes
-        base64_encoded_uuid = base64.b64encode(uuid_bytes)
+        base64_encoded_uuid = base64.urlsafe_b64encode(uuid_bytes)
 
-        return base64_encoded_uuid.decode()
+        return base64_encoded_uuid.decode().rstrip("=")
 
     def configure(self):
         if self.kraft_enabled: