Chia-Network/chia-blockchain
1
0
Fork 0
mirror of https://github.com/Chia-Network/chia-blockchain.git synced 2024-09-21 00:24:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

test higher fee per cost

Browse Source
This commit is contained in:
Yostra 2020-01-30 16:56:24 -08:00
parent a1f5cbaab0
commit bb09b90c28
1 changed files with 42 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
tests/test_mempool.py
View File

@ -4,6 +4,7 @@ from src.server.outbound_message import OutboundMessage
from src.protocols import peer_protocol
from src.wallet.wallets.standard_wallet.wallet import Wallet
from tests.setup_nodes import setup_two_nodes, test_constants, bt
from tests.wallet_tools import WalletTool
@pytest.fixture(scope="module")
@ -21,7 +22,7 @@ class TestMempool:
@pytest.mark.asyncio
async def test_basic_mempool(self, two_nodes):
num_blocks = 3
wallet_a = Wallet()
wallet_a = WalletTool()
coinbase_puzzlehash = wallet_a.get_new_puzzlehash()
wallet_receiver = Wallet()
receiver_puzzlehash = wallet_receiver.get_new_puzzlehash()
@ -31,8 +32,7 @@ class TestMempool:
block = blocks[1]
async for _ in full_node_1.block(peer_protocol.Block(block)):
wallet_a.notify([block.body.coinbase], [])
spend_bundle = wallet_a.generate_signed_transaction(1000, receiver_puzzlehash)
spend_bundle = wallet_a.generate_signed_transaction(1000, receiver_puzzlehash, [block.body.coinbase])
tx: peer_protocol.Transaction = peer_protocol.Transaction(spend_bundle)
async for _ in full_node_1.transaction(tx):
outbound: OutboundMessage = _
@ -42,9 +42,9 @@ class TestMempool:
@pytest.mark.asyncio
async def test_double_spend(self, two_nodes):
num_blocks = 3
wallet_a = Wallet()
wallet_a = WalletTool()
coinbase_puzzlehash = wallet_a.get_new_puzzlehash()
wallet_receiver = Wallet()
wallet_receiver = WalletTool()
receiver_puzzlehash = wallet_receiver.get_new_puzzlehash()
blocks = bt.get_consecutive_blocks(test_constants, num_blocks, [], 10, b"", coinbase_puzzlehash)
@ -54,15 +54,15 @@ class TestMempool:
async for _ in full_node_1.block(peer_protocol.Block(block)):
pass
wallet_a.notify([block.body.coinbase], [])
spend_bundle1 = wallet_a.generate_signed_transaction(1000, receiver_puzzlehash)
spend_bundle1 = wallet_a.generate_signed_transaction(1000, receiver_puzzlehash, [block.body.coinbase])
tx1: peer_protocol.Transaction = peer_protocol.Transaction(spend_bundle1)
async for _ in full_node_1.transaction(tx1):
outbound: OutboundMessage = _
# Maybe transaction means that it's accepted in mempool
assert outbound.message.function == "maybe_transaction"
spend_bundle2 = wallet_a.generate_signed_transaction(2000, receiver_puzzlehash)
other_receiver = WalletTool()
spend_bundle2 = wallet_a.generate_signed_transaction(1000, other_receiver.get_new_puzzlehash(), [block.body.coinbase])
tx2: peer_protocol.Transaction = peer_protocol.Transaction(spend_bundle2)
async for _ in full_node_1.transaction(tx2):
pass
@ -73,3 +73,37 @@ class TestMempool:
assert sb1 == spend_bundle1
assert sb2 is None
@pytest.mark.asyncio
async def test_double_spend_with_higher_fee(self, two_nodes):
num_blocks = 3
wallet_a = WalletTool()
coinbase_puzzlehash = wallet_a.get_new_puzzlehash()
wallet_receiver = WalletTool()
receiver_puzzlehash = wallet_receiver.get_new_puzzlehash()
blocks = bt.get_consecutive_blocks(test_constants, num_blocks, [], 10, b"", coinbase_puzzlehash)
full_node_1, full_node_2, server_1, server_2 = two_nodes
block = blocks[1]
async for _ in full_node_1.block(peer_protocol.Block(block)):
pass
spend_bundle1 = wallet_a.generate_signed_transaction(1000, receiver_puzzlehash, [block.body.coinbase])
tx1: peer_protocol.Transaction = peer_protocol.Transaction(spend_bundle1)
async for _ in full_node_1.transaction(tx1):
outbound: OutboundMessage = _
# Maybe transaction means that it's accepted in mempool
assert outbound.message.function == "maybe_transaction"
spend_bundle2 = wallet_a.generate_signed_transaction(1000, receiver_puzzlehash, [block.body.coinbase], 1)
tx2: peer_protocol.Transaction = peer_protocol.Transaction(spend_bundle2)
async for _ in full_node_1.transaction(tx2):
pass
sb1 = await full_node_1.mempool.get_spendbundle(spend_bundle1.name())
sb2 = await full_node_1.mempool.get_spendbundle(spend_bundle2.name())
assert sb1 is None
assert sb2 == spend_bundle2