From 27bcc3c685c0954b80ee03a20533ea3c89c3a785 Mon Sep 17 00:00:00 2001
From: shortcutme <tamas@zeronet.io>
Date: Sat, 16 Mar 2019 00:59:11 +0100
Subject: [PATCH] Test sign and verify with all crypto lib avaliable

---
 src/Test/TestCryptBitcoin.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/Test/TestCryptBitcoin.py b/src/Test/TestCryptBitcoin.py
index a6009679..54195c93 100644
--- a/src/Test/TestCryptBitcoin.py
+++ b/src/Test/TestCryptBitcoin.py
@@ -22,15 +22,15 @@ class TestCryptBitcoin:
         sign_bad = CryptBitcoin.signOld("hello", privatekey_bad)
         assert not CryptBitcoin.verify("hello", address, sign_bad)
 
-    def testSign(self):
+    def testSign(self, crypt_bitcoin_lib):
         privatekey = "5K9S6dVpufGnroRgFrT6wsKiz2mJRYsC73eWDmajaHserAp3F1C"
         privatekey_bad = "5Jbm9rrusXyApAoM8YoM4Rja337zMMoBUMRJ1uijiguU2aZRnwC"
 
         # Get address by privatekey
-        address = CryptBitcoin.privatekeyToAddress(privatekey)
+        address = crypt_bitcoin_lib.privatekeyToAddress(privatekey)
         assert address == "1MpDMxFeDUkiHohxx9tbGLeEGEuR4ZNsJz"
 
-        address_bad = CryptBitcoin.privatekeyToAddress(privatekey_bad)
+        address_bad = crypt_bitcoin_lib.privatekeyToAddress(privatekey_bad)
         assert address_bad != "1MpDMxFeDUkiHohxx9tbGLeEGEuR4ZNsJz"
 
         # Text signing
@@ -38,12 +38,12 @@ class TestCryptBitcoin:
             pad = pad_len * "!"
             sign = CryptBitcoin.sign("hello" + pad, privatekey)
 
-            assert CryptBitcoin.verify("hello" + pad, address, sign)
-            assert not CryptBitcoin.verify("not hello" + pad, address, sign)
+            assert crypt_bitcoin_lib.verify(data, address, sign)
+            assert not crypt_bitcoin_lib.verify("invalid" + data, address, sign)
 
         # Signed by bad privatekey
-        sign_bad = CryptBitcoin.sign("hello", privatekey_bad)
-        assert not CryptBitcoin.verify("hello", address, sign_bad)
+        sign_bad = crypt_bitcoin_lib.sign("hello", privatekey_bad)
+        assert not crypt_bitcoin_lib.verify("hello", address, sign_bad)
 
     def testNewPrivatekey(self):
         assert CryptBitcoin.newPrivatekey() != CryptBitcoin.newPrivatekey()