Add password to test

2020-08-05 19:51:22 +02:00
parent 36ffad1202
commit a845aecda1
2 changed files with 8 additions and 2 deletions
--- a/src/keystore/postgres.py
+++ b/src/keystore/postgres.py
@@ -15,6 +15,11 @@ logging.basicConfig(level=logging.DEBUG)
 logg = logging.getLogger(__file__)


+
+def to_bytes(x):
+    return x.encode('utf-8')
+
+    
 class ReferenceDatabase:


@@ -52,7 +57,8 @@ class ReferenceDatabase:
            h = sha3.keccak_256()
            h.update(self.symmetric_key)
            if password != None:
-                h.update(password)
+                password_bytes = to_bytes(password)
+                h.update(password_bytes)
            g = h.digest()
            return Fernet(base64.b64encode(g))