mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-11-28 14:59:22 +01:00
[tests] move generate_wif_key to wallet_util.py
generate_wif_key is a wallet utility function. Move it from the EC key module to the wallet util module.
This commit is contained in:
@@ -11,12 +11,13 @@ import os
|
|||||||
|
|
||||||
from test_framework.authproxy import JSONRPCException
|
from test_framework.authproxy import JSONRPCException
|
||||||
from test_framework.descriptors import descsum_create, drop_origins
|
from test_framework.descriptors import descsum_create, drop_origins
|
||||||
from test_framework.key import ECPubKey, ECKey, bytes_to_wif
|
from test_framework.key import ECPubKey, ECKey
|
||||||
from test_framework.test_framework import BitcoinTestFramework
|
from test_framework.test_framework import BitcoinTestFramework
|
||||||
from test_framework.util import (
|
from test_framework.util import (
|
||||||
assert_raises_rpc_error,
|
assert_raises_rpc_error,
|
||||||
assert_equal,
|
assert_equal,
|
||||||
)
|
)
|
||||||
|
from test_framework.wallet_util import bytes_to_wif
|
||||||
|
|
||||||
class RpcCreateMultiSigTest(BitcoinTestFramework):
|
class RpcCreateMultiSigTest(BitcoinTestFramework):
|
||||||
def set_test_params(self):
|
def set_test_params(self):
|
||||||
|
|||||||
@@ -8,8 +8,6 @@ keys, and is trivially vulnerable to side channel attacks. Do not use for
|
|||||||
anything but tests."""
|
anything but tests."""
|
||||||
import random
|
import random
|
||||||
|
|
||||||
from .address import byte_to_base58
|
|
||||||
|
|
||||||
def modinv(a, n):
|
def modinv(a, n):
|
||||||
"""Compute the modular inverse of a modulo n
|
"""Compute the modular inverse of a modulo n
|
||||||
|
|
||||||
@@ -386,14 +384,3 @@ class ECKey():
|
|||||||
rb = r.to_bytes((r.bit_length() + 8) // 8, 'big')
|
rb = r.to_bytes((r.bit_length() + 8) // 8, 'big')
|
||||||
sb = s.to_bytes((s.bit_length() + 8) // 8, 'big')
|
sb = s.to_bytes((s.bit_length() + 8) // 8, 'big')
|
||||||
return b'\x30' + bytes([4 + len(rb) + len(sb), 2, len(rb)]) + rb + bytes([2, len(sb)]) + sb
|
return b'\x30' + bytes([4 + len(rb) + len(sb), 2, len(rb)]) + rb + bytes([2, len(sb)]) + sb
|
||||||
|
|
||||||
def bytes_to_wif(b, compressed=True):
|
|
||||||
if compressed:
|
|
||||||
b += b'\x01'
|
|
||||||
return byte_to_base58(b, 239)
|
|
||||||
|
|
||||||
def generate_wif_key():
|
|
||||||
# Makes a WIF privkey for imports
|
|
||||||
k = ECKey()
|
|
||||||
k.generate()
|
|
||||||
return bytes_to_wif(k.get_bytes(), k.is_compressed)
|
|
||||||
|
|||||||
@@ -6,6 +6,7 @@
|
|||||||
from collections import namedtuple
|
from collections import namedtuple
|
||||||
|
|
||||||
from test_framework.address import (
|
from test_framework.address import (
|
||||||
|
byte_to_base58,
|
||||||
key_to_p2pkh,
|
key_to_p2pkh,
|
||||||
key_to_p2sh_p2wpkh,
|
key_to_p2sh_p2wpkh,
|
||||||
key_to_p2wpkh,
|
key_to_p2wpkh,
|
||||||
@@ -13,10 +14,7 @@ from test_framework.address import (
|
|||||||
script_to_p2sh_p2wsh,
|
script_to_p2sh_p2wsh,
|
||||||
script_to_p2wsh,
|
script_to_p2wsh,
|
||||||
)
|
)
|
||||||
from test_framework.key import (
|
from test_framework.key import ECKey
|
||||||
bytes_to_wif,
|
|
||||||
ECKey,
|
|
||||||
)
|
|
||||||
from test_framework.script import (
|
from test_framework.script import (
|
||||||
CScript,
|
CScript,
|
||||||
OP_0,
|
OP_0,
|
||||||
@@ -120,3 +118,14 @@ def test_address(node, address, **kwargs):
|
|||||||
raise AssertionError("key {} unexpectedly returned in getaddressinfo.".format(key))
|
raise AssertionError("key {} unexpectedly returned in getaddressinfo.".format(key))
|
||||||
elif addr_info[key] != value:
|
elif addr_info[key] != value:
|
||||||
raise AssertionError("key {} value {} did not match expected value {}".format(key, addr_info[key], value))
|
raise AssertionError("key {} value {} did not match expected value {}".format(key, addr_info[key], value))
|
||||||
|
|
||||||
|
def bytes_to_wif(b, compressed=True):
|
||||||
|
if compressed:
|
||||||
|
b += b'\x01'
|
||||||
|
return byte_to_base58(b, 239)
|
||||||
|
|
||||||
|
def generate_wif_key():
|
||||||
|
# Makes a WIF privkey for imports
|
||||||
|
k = ECKey()
|
||||||
|
k.generate()
|
||||||
|
return bytes_to_wif(k.get_bytes(), k.is_compressed)
|
||||||
|
|||||||
Reference in New Issue
Block a user