From 5a45512f3b115f272c2e8b07f1cde4f4bd4d6bb1 Mon Sep 17 00:00:00 2001
From: nolash <dev@holbrook.no>
Date: Sun, 20 Sep 2020 10:15:48 +0200
Subject: [PATCH] Return new address on method 'new'

---
 crypto_dev_signer/keystore/postgres.py | 2 +-
 test/test_keystore.py                  | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/crypto_dev_signer/keystore/postgres.py b/crypto_dev_signer/keystore/postgres.py
index 803695a..32498af 100644
--- a/crypto_dev_signer/keystore/postgres.py
+++ b/crypto_dev_signer/keystore/postgres.py
@@ -55,7 +55,7 @@ class ReferenceKeystore(Keystore):
         def new(self, password=None):
             b = os.urandom(32)
             pk = keyapi.PrivateKey(b)
-            self.import(pk, password)
+            return self.import_key(pk, password)
 
 
         def import_key(self, pk, password=None):
diff --git a/test/test_keystore.py b/test/test_keystore.py
index 618fa63..8303c66 100644
--- a/test/test_keystore.py
+++ b/test/test_keystore.py
@@ -1,13 +1,16 @@
 #!/usr/bin/python
 
+# standard imports
 import unittest
 import logging
 import base64
 
+# third-party imports
 import psycopg2
 from psycopg2 import sql
 from cryptography.fernet import Fernet, InvalidToken
 
+# local imports
 from crypto_dev_signer.keystore import ReferenceKeystore
 
 logging.basicConfig(level=logging.DEBUG)
@@ -23,6 +26,7 @@ class TestDatabase(unittest.TestCase):
     db = None
 
     def setUp(self):
+        logg.debug('setup')
         # arbitrary value
         symkey_hex = 'E92431CAEE69313A7BE9E443C4ABEED9BF8157E9A13553B4D5D6E7D51B5021D9'
         self.symkey = bytes.fromhex(symkey_hex)