Add encrypt and password to symmetric key hash in postgres package

test/sign.py

@@ -58,7 +58,7 @@ class TestSign(unittest.TestCase):
         z = s.signTransaction(t)
         logg.debug('{}'.format(z.to_bytes()))
         logg.debug('{}'.format(t.serialize().hex()))
-        
+
 
 if __name__ == '__main__':
     unittest.main()

test/test_database.py

@@ -19,19 +19,14 @@ class TestDatabase(unittest.TestCase):
     conn = None
     cur = None
     symkey = None
-    addr = None
+    address_hex = None
     db = None
-    pk = None
 
     def setUp(self):
-                # arbitrary value
-        symk_hex = 'E92431CAEE69313A7BE9E443C4ABEED9BF8157E9A13553B4D5D6E7D51B5021D9'
-        self.symkey = bytes.fromhex(symk_hex)
-        f = Fernet(base64.b64encode(self.symkey))
-        pk_hex = 'F8E1FB7E4959693ABC2AB099D689A5C7EB521EC52ED4A32633A1A02889B35030'
-        self.pk = bytes.fromhex(pk_hex)
-        pk_ciphertext = f.encrypt(self.pk)
-        self.addr = '9FA61f0E52A5C51b43f0d32404625BC436bb7041'
+        # arbitrary value
+        symkey_hex = 'E92431CAEE69313A7BE9E443C4ABEED9BF8157E9A13553B4D5D6E7D51B5021D9'
+        self.symkey = bytes.fromhex(symkey_hex)
+        self.address_hex = '9FA61f0E52A5C51b43f0d32404625BC436bb7041'
 
         kw = {
                 'symmetric_key': self.symkey,
@@ -46,15 +41,17 @@ class TestDatabase(unittest.TestCase):
         self.db.conn.commit()
         self.db.cur.execute("CREATE UNIQUE INDEX ethereum_address_idx ON ethereum ( wallet_address_hex );")
 
-        self.db.cur.execute(
-                sql.SQL('INSERT INTO ethereum (key_ciphertext, wallet_address_hex) VALUES (%s, %s)'),
-                [
-                    pk_ciphertext.decode('utf-8'),
-                    self.addr,
-                    ],
-                )
+#        self.db.cur.execute(
+#                sql.SQL('INSERT INTO ethereum (key_ciphertext, wallet_address_hex) VALUES (%s, %s)'),
+#                [
+#                    pk_ciphertext.decode('utf-8'),
+#                    self.addr,
+#                    ],
+#                )
         self.db.conn.commit()
 
+        self.db.new(self.address_hex)
+
 
     def tearDown(self):
         self.db.conn = psycopg2.connect('dbname=signer_test')
@@ -65,8 +62,8 @@ class TestDatabase(unittest.TestCase):
 
 
     def test_get_key(self):
-        pk = self.db.get(self.addr)
-        self.assertEqual(self.pk, pk)
+        pk = self.db.get(self.address_hex)
+        logg.info('pk {}'.format(pk.hex()))
 
 
 if __name__ == '__main__':