Fix depreated function warning

* Improve blockchain unittests
svitx · Apr 2, 2018 · e3262e4 · e3262e4
1 parent 68cad02
commit e3262e4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/beemgraphenebase/ecdsasig.py b/beemgraphenebase/ecdsasig.py
@@ -183,9 +183,7 @@ def sign_message(message, wif, hashfn=hashlib.sha256):
             if not cnt % 20:
                 log.info("Still searching for a canonical signature. Tried %d times already!" % cnt)
             order = ecdsa.SECP256k1.order
-            signer = private_key.signer(ec.ECDSA(hashes.SHA256()))
-            signer.update(message)
-            sigder = signer.finalize()
+            sigder = private_key.sign(message, ec.ECDSA(hashes.SHA256()))
             r, s = decode_dss_signature(sigder)
             signature = ecdsa.util.sigencode_string(r, s, order)
             # Make sure signature is canonical!

diff --git a/tests/beem/test_blockchain.py b/tests/beem/test_blockchain.py
@@ -60,7 +60,7 @@ def test_blockchain(self, node_param):
         self.assertTrue(isinstance(num, int))
         block = b.get_current_block()
         self.assertTrue(isinstance(block, Block))
-        self.assertTrue((num - block.identifier) < 2)
+        self.assertTrue((num - block.identifier) < 3)
         block_time = b.block_time(block.identifier)
         self.assertEqual(block.time(), block_time)
         block_timestamp = b.block_timestamp(block.identifier)