def test_kdf_params():
salt = rando.newHash()
auth = rando.newHash()
kdf = crypto.KDFParams(salt, auth)
b = kdf.serialize()
reKDF = crypto.KDFParams.unblob(b.b)
assert kdf.kdfFunc == reKDF.kdfFunc
assert kdf.hashName == reKDF.hashName
assert kdf.salt == reKDF.salt
assert kdf.auth == reKDF.auth
assert kdf.iterations == reKDF.iterations
def test_tx_from_hex(self, prepareLogger):
tx = msgtx.MsgTx(
cachedHash=None,
serType=0,
version=123,
txIn=[
msgtx.TxIn(
previousOutPoint=msgtx.OutPoint(txHash=newHash(),
idx=5,
tree=1),
sequence=msgtx.MaxTxInSequenceNum,
valueIn=500,
blockHeight=111,
blockIndex=12,
signatureScript=newHash(),
),
],
txOut=[
msgtx.TxOut(value=321, pkScript=newHash(), version=0),
msgtx.TxOut(value=654, pkScript=newHash(), version=0),
msgtx.TxOut(value=987, pkScript=newHash(), version=0),
],
lockTime=int(time.time()),
expiry=int(time.time()) + 86400,
)
b = tx.serialize()
reTx = msgtx.MsgTx.unblob(b)
assert tx.serType == reTx.serType
assert tx.version == reTx.version
assert tx.lockTime == reTx.lockTime
assert tx.expiry == reTx.expiry
for i, txIn in enumerate(tx.txIn):
reTxIn = reTx.txIn[i]
assert txIn.previousOutPoint.hash == reTxIn.previousOutPoint.hash
assert txIn.previousOutPoint.index == reTxIn.previousOutPoint.index
assert txIn.previousOutPoint.tree == reTxIn.previousOutPoint.tree
assert txIn.sequence == reTxIn.sequence
assert txIn.valueIn == reTxIn.valueIn
assert txIn.blockHeight == reTxIn.blockHeight
assert txIn.blockIndex == reTxIn.blockIndex
assert txIn.signatureScript == reTxIn.signatureScript
for i, txOut in enumerate(tx.txOut):
reTxOut = reTx.txOut[i]
assert txOut.value == reTxOut.value
assert txOut.version == reTxOut.version
assert txOut.pkScript == reTxOut.pkScript
def utxosource(amt, filter):
nextVal = 10
total = 0
utxos = []
while total < amt:
atoms = int(nextVal * 1e8)
privKey = Curve.generateKey()
pkHash = crypto.hash160(
privKey.pub.serializeCompressed().b)
addr = addrlib.AddressPubKeyHash(pkHash, testnet)
addrs.append(addr)
addrString = addr.string()
keys[addrString] = privKey
pkScript = txscript.makePayToAddrScript(
addrString, testnet)
txHash = rando.newHash()
txid = reversed(txHash).hex()
utxos.append(
account.UTXO(
address=addrString,
txHash=txHash,
vout=0,
ts=int(time.time()),
scriptPubKey=pkScript,
satoshis=atoms,
))
tx = msgtx.MsgTx.new()
tx.addTxOut(msgtx.TxOut(value=atoms, pkScript=pkScript))
txs[txid] = tx
total += atoms
nextVal *= 2
return utxos, True
from decred.util.encode import ByteArray
# if __name__ == "__main__":
# for i in range(256):
# b = ByteArray("0f17") + ByteArray(3) + newHash() + newHash()
# enc = b58encode(b.bytes()).decode()
# print(i, enc)
# addr = addrlib.AddressPubKeyHash(newHash()[:20], nets.mainnet)
# print("addr", addr.string(), addr.netID.hex())
DummyHash = ByteArray(
"0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef")
redeemScript = ByteArray(opcode.OP_SHA256)
redeemScript += txscript.addData(newHash())
redeemScript += opcode.OP_EQUALVERIFY
redeemScript += txscript.addData(signingKey.pub.serializeCompressed())
redeemScript += opcode.OP_CHECKSIG
txHex = (
"0100000001aac1eeaecbba3ee756053d7177ef76dec3e2709be1d605711022a4e72d9"
"2acda0000000000ffffffff01cad5f5050000000000001976a914523368819aa5628c"
"cc178de8686e6153fb7ec67588ac00000000000000000100e1f505000000000000000"
"000000000b1483045022100be3f496646f8f99a182e9ffa97a8da050b0bdf679c0a3c"
"2abbbe9874e77e48af02200e2033a5301b351be66478bedf0e65cfb07a104e3dbe3cd"
"d6316f2d6aafe93de0120ba7816bf8f01cfea414140de5dae2223b00361a396177a9c"
"b410ff61f20015ad46c0204f8b42c22dd3729b519ba6f68d2da7cc5b2d606d05daed5"
"ad5128cc03e6c635888210227726129c88a525ecb55d2ff5390a447894f6b810d977e"
"25f91ea3b5e2d66d6eac")
def test_main(self, monkeypatch):
"""
Test account functionality.
"""
# Set up globals for test.
monkeypatch.setattr(account, "DefaultGapLimit", 2)
db = KeyValueDatabase(":memory:").child("tmp")
acct = self.newAccount(db)
acct.unlock(self.cryptoKey)
acct.generateGapAddresses()
for _ in range(20):
acct.nextExternalAddress()
satoshis = int(round(5 * 1e8))
txHash = rando.newHash()
txid = reversed(txHash).hex()
vout = 2
address = acct.nextExternalAddress()
utxo = account.UTXO(
address=address,
txHash=txHash,
vout=vout,
scriptPubKey=ByteArray(0),
satoshis=satoshis,
maturity=1,
)
# A helper function to get the current utxo count.
utxocount = lambda: len(list(acct.utxoscan()))
# Add the utxo
acct.addUTXO(utxo)
# Check the count and balance categories.
assert utxocount() == 1
assert acct.calcBalance(1).total == satoshis
assert acct.calcBalance(1).available == satoshis
assert acct.calcBalance(0).available == 0
# Helper functions.
assert acct.caresAboutTxid(txid)
assert not acct.caresAboutTxid("")
acct.addUTXO(utxo)
assert utxocount() == 1
acct.spendUTXO(utxo)
assert utxocount() == 0
b = acct.serialize()
reAcct = account.Account.unblob(b.b)
assert acct.pubKeyEncrypted == reAcct.pubKeyEncrypted
assert acct.privKeyEncrypted == reAcct.privKeyEncrypted
assert acct.name == reAcct.name
assert acct.coinID == reAcct.coinID
assert acct.netID == reAcct.netID
assert acct.gapLimit == reAcct.gapLimit
assert acct.cursorExt == reAcct.cursorExt
assert acct.cursorInt == reAcct.cursorInt
assert acct.gapLimit == reAcct.gapLimit
# Create a faux blockchain for the account.
class Blockchain:
txidsForAddr = lambda addr: []
UTXOs = lambda addrs: []
tipHeight = 5
acct.blockchain = Blockchain
class Signals:
b = None
@classmethod
def balance(cls, b):
cls.b = b
acct.signals = Signals
# Add a txid for this first address
txid = rando.newHash().hex()
zerothAddr = acct.externalAddresses[0]
acct.addTxid(zerothAddr, txid)
# Add a voting service provider
vspKey = rando.newKey().hex()
ticketAddr = "ticketAddr"
pi = PurchaseInfo("addr", 1.0, ByteArray(b"scripthashscript"),
ticketAddr, 1, 0, 2)
vsp = VotingServiceProvider("https://myvsp.com", vspKey,
nets.mainnet.Name, pi)
with pytest.raises(DecredError):
acct.setPool(None)
acct.setPool(vsp)
acct.setNewPool(vsp)
# Add UTXOs
utxos = [account.UTXO.parse(u) for u in self.dcrdataUTXOs]
acct.resolveUTXOs(utxos)
ticketStats = acct.ticketStats()
assert ticketStats.count == 0
assert ticketStats.value == 0
# Check that the addresses txid and the vsp load from the database.
acct.txs.clear()
acct.stakePools.clear()
acct.utxos.clear()
assert acct.stakePool() is None
acct.load(db, Blockchain, Signals)
assert zerothAddr in acct.txs
assert len(acct.txs[zerothAddr]) == 1
assert acct.txs[zerothAddr][0] == txid
assert len(acct.stakePools) == 1
assert acct.stakePool().apiKey == vspKey
assert acct.calcBalance().available == self.utxoTotal
assert len(acct.getUTXOs(self.utxoTotal)[0]) == 2
assert len(acct.getUTXOs(self.utxoTotal, lambda u: False)[0]) == 0
assert acct.lastSeen(acct.externalAddresses, -1) == 0
branch, idx = acct.branchAndIndex(zerothAddr)
assert branch == account.EXTERNAL_BRANCH
assert idx == 0
branch, idx = acct.branchAndIndex(acct.internalAddresses[0])
assert branch == account.INTERNAL_BRANCH
assert idx == 0
checkKey = acct.privKey.child(account.EXTERNAL_BRANCH).child(0).key
with pytest.raises(DecredError):
acct.privKeyForAddress("")
assert acct.privKeyForAddress(zerothAddr).key == checkKey
ticketAddrs = acct.addTicketAddresses([])
assert len(ticketAddrs) == 1
assert ticketAddrs[0] == ticketAddr
assert acct.hasPool()
# Exercise setNode
acct.setNode(1)
assert acct.node == 1
# Add a coinbase transaction output to the account.
coinbase = newCoinbaseTx()
cbUTXO = account.UTXO("addr",
ByteArray(b"id"),
1,
height=5,
satoshis=int(1e8))
coinbase.addTxOut(msgtx.TxOut(int(1e8)))
coinbaseID = coinbase.id()
cbUTXO.txid = coinbaseID
acct.addUTXO(cbUTXO)
acct.addMempoolTx(coinbase)
assert coinbaseID in acct.mempool
# Send a block signal and have the transaction confirmed.
sig = {"message": {"block": {"Tx": [{"TxID": coinbaseID}]}}}
acct.blockchain.tipHeight = 5
acct.blockchain.tx = lambda *a: coinbase
acct.blockSignal(sig)
assert coinbaseID not in acct.mempool
assert cbUTXO.key() in acct.utxos
maturity = 5 + nets.mainnet.CoinbaseMaturity
assert acct.utxos[cbUTXO.key()].maturity == maturity
assert acct.calcBalance(
maturity).available == self.utxoTotal + cbUTXO.satoshis
assert acct.calcBalance(0).available == self.utxoTotal
acct.spendUTXOs([cbUTXO])
assert acct.calcBalance(maturity).available == self.utxoTotal
# make a ticket and a vote
ticket = account.UTXO.parse(dcrdataUTXO)
utxo.setTicketInfo(dcrdataTinfo)
utxo.scriptPubKey = ticketScript
utxo.parseScriptClass()
acct.addUTXO(ticket)
expVal = acct.calcBalance().available - ticket.satoshis
voteTx = msgtx.MsgTx.new()
voteTx.addTxIn(
msgtx.TxIn(
msgtx.OutPoint(reversed(ByteArray(ticket.txid)), ticket.vout,
0)))
acct.blockchain.tx = lambda *a: voteTx
acct.addressSignal(ticketAddr, "somehash")
assert Signals.b.available == expVal
# Detect a new utxo for the account.
newVal = int(5e8)
expVal = acct.calcBalance().available + newVal
addr = acct.externalAddresses[1]
a = addrlib.decodeAddress(addr, nets.mainnet)
script = txscript.payToAddrScript(a)
newTx = msgtx.MsgTx.new()
op = msgtx.TxOut(newVal, script)
newTx.addTxOut(op)
acct.blockchain.tx = lambda *a: newTx
utxo = account.UTXO(addr, ByteArray(b"txid"), 0, satoshis=newVal)
acct.blockchain.txVout = lambda *a: utxo
acct.addressSignal(addr, "somehash")
assert Signals.b.available == expVal
# test syncing. add a single output for external address #2
acct.stakePool().getPurchaseInfo = lambda: None
acct.stakePool().authorize = lambda a: True
newVal = int(3e8)
addr = acct.externalAddresses[2]
a = addrlib.decodeAddress(addr, nets.mainnet)
script = txscript.payToAddrScript(a)
newTx = msgtx.MsgTx.new()
op = msgtx.TxOut(newVal, script)
newTx.addTxOut(op)
utxo = account.UTXO(addr, ByteArray(b"txid"), 0, satoshis=newVal)
def t4a(*a):
acct.blockchain.txidsForAddr = lambda *a: []
return [newTx.id()]
acct.blockchain.txidsForAddr = t4a
def utxos4a(*a):
acct.blockchain.UTXOs = lambda *a: []
return [utxo]
acct.blockchain.UTXOs = utxos4a
acct.blockchain.subscribeAddresses = lambda addrs, receiver: None
acct.blockchain.subscribeBlocks = lambda a: None
acct.sync()
assert Signals.b.available == newVal
# spend the utxo by sending it. Reusing newTx for convenience.
changeVal = int(5e7)
addr = acct.internalAddresses[0]
change = account.UTXO(addr,
ByteArray(b"newtxid"),
0,
satoshis=changeVal)
acct.blockchain.sendToAddress = lambda *a: (newTx, [utxo], [change])
acct.sendToAddress(1, "recipient")
assert Signals.b.available == changeVal
# purchase some tickets. Reusing newTx for convenience again.
ticket = account.UTXO.parse(dcrdataUTXO)
ticket.setTicketInfo(dcrdataTinfo)
ticket.scriptPubKey = ticketScript
ticket.parseScriptClass()
acct.blockchain.purchaseTickets = lambda *a: ([newTx, []], [],
[ticket])
acct.signals.spentTickets = lambda: True
acct.purchaseTickets(1, 1)
assert Signals.b.total == changeVal + ticket.satoshis
# revoke the ticket
ticketTx = msgtx.MsgTx.new()
op = msgtx.TxOut(ticket.satoshis, ticket.scriptPubKey)
ticketTx.addTxOut(op)
ticket.tinfo.status = "missed"
redeemHash = addrlib.AddressScriptHash(
txscript.extractStakeScriptHash(ticketScript, opcode.OP_SSTX),
nets.mainnet,
)
acct.stakePool().purchaseInfo.ticketAddress = redeemHash.string()
revoked = False
def rev(*a):
nonlocal revoked
revoked = True
acct.blockchain.tx = lambda *a: ticketTx
acct.blockchain.revokeTicket = rev
acct.revokeTickets()
assert revoked