Chia-Network/chia-blockchain
1
0
Fork 0
mirror of https://github.com/Chia-Network/chia-blockchain.git synced 2024-11-29 05:18:11 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix tests

Browse Source
This commit is contained in:
Mariano Sorgente 2020-03-21 01:24:48 +09:00
parent 87e5946f41
commit 2a388d3545
No known key found for this signature in database
GPG Key ID: 0F866338C369278C
9 changed files with 212 additions and 144 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
.gitignore vendored
View File

@ -66,4 +66,7 @@ chia-blockchain.tar.gz
src/wallet/electron/node_modules/
# Bip158 build dir
lib/bip158/build/
lib/bip158/build/
# Node modules
**/node_modules

4
CONTRIBUTING.md
View File

@ -12,7 +12,7 @@ If you want to learn more about this project, read the [wiki](https://github.com
We would be pleased to accept code contributions to this project.
As we are in the alpha stage, the main priority is getting a robust blockchain up and running, with as many of the mainnet features as possible.
You can visit our [Trello project board](https://trello.com/b/ZuNx7sET) to get a sense of what is in the backlog.
Generally things to the left are in progess or done. Some things go through "Coming up soon" but some will come directly out of other columns.
Generally things to the left are in progess or done. Some things go through "Coming up soon" but some will come directly out of other columns.
Usually the things closer to the top of each column are the ones that will be worked on soonest.
If you are interested in cryptography, math, or just like hacking in python, there are many interesting problems to work on.
Contact any of the team members on keybase: https://keybase.io/team/chia_network.public, which we use as the main communication method and you can comment on any Trello card.
@ -22,7 +22,7 @@ The first time the tests are run, BlockTools will create and persist many plots.
proofs of space during testing. The next time tests are run, this won't be necessary.
```bash
black src tests && flake8 src && mypy src tests
black src tests scripts && flake8 src scripts --exclude src/wallet/electron/node_modules && mypy src scripts tests
py.test tests -s -v
```
Black is used as an automatic style formatter to make things easier, and flake8 helps ensure consistent style.

2
src/consensus/constants.py
View File

File diff suppressed because one or more lines are too long

8
tests/full_node/test_full_node.py
View File

@ -424,12 +424,15 @@ class TestFullNodeProtocol:
blocks_new = bt.get_consecutive_blocks(
test_constants,
1,
2,
blocks_list,
10,
reward_puzzlehash=coinbase_puzzlehash,
seed=b"another seed 3",
)
# Add one block
[_ async for _ in full_node_1.respond_block(fnp.RespondBlock(blocks_new[-2]))]
unf_block = FullBlock(
blocks_new[-1].proof_of_space,
None,
@ -446,7 +449,6 @@ class TestFullNodeProtocol:
assert res[0].message.data == fnp.RejectUnfinishedBlockRequest(
unf_block.header_hash
)
# Have unfinished block
[x async for x in full_node_1.respond_unfinished_block(unf_block_req)]
res = [x async for x in full_node_1.request_unfinished_block(req)]
@ -735,7 +737,7 @@ class TestFullNodeProtocol:
msgs = [
_ async for _ in full_node_1.respond_block(fnp.RespondBlock(blocks_new[-5]))
]
assert len(msgs) == 5
assert len(msgs) == 5 or len(msgs) == 6
# Updates blockchain tips
tip_hashes_again = set(
[t.header_hash for t in full_node_1.blockchain.get_current_tips()]

137
tests/full_node/test_full_sync.py
View File

@ -105,140 +105,3 @@ class TestFullSync:
await asyncio.sleep(0.1)
raise Exception("Took too long to process blocks")
class TestWalletSync:
@pytest.fixture(scope="function")
async def wallet_node(self):
async for _ in setup_node_and_wallet():
yield _
@pytest.fixture(scope="function")
async def wallet_node_starting_height(self):
async for _ in setup_node_and_wallet(dic={"starting_height": 100}):
yield _
@pytest.mark.asyncio
async def test_basic_sync_wallet(self, wallet_node):
num_blocks = 25
blocks = bt.get_consecutive_blocks(test_constants, num_blocks, [])
full_node_1, wallet_node, server_1, server_2 = wallet_node
for i in range(1, len(blocks)):
async for _ in full_node_1.respond_block(
full_node_protocol.RespondBlock(blocks[i])
):
pass
await server_2.start_client(
PeerInfo(server_1._host, uint16(server_1._port)), None
)
start = time.time()
found = False
while time.time() - start < 60:
# The second node should eventually catch up to the first one, and have the
# same tip at height num_blocks - 1.
if (
wallet_node.wallet_state_manager.block_records[
wallet_node.wallet_state_manager.lca
].height
>= num_blocks - 6
):
found = True
break
await asyncio.sleep(0.1)
if not found:
raise Exception(
f"Took too long to process blocks, stopped at: {time.time() - start}"
)
# Tests a reorg with the wallet
start = time.time()
found = False
blocks_reorg = bt.get_consecutive_blocks(test_constants, 45, blocks[:-5])
for i in range(1, len(blocks_reorg)):
async for msg in full_node_1.respond_block(
full_node_protocol.RespondBlock(blocks_reorg[i])
):
server_1.push_message(msg)
start = time.time()
while time.time() - start < 100:
if (
wallet_node.wallet_state_manager.block_records[
wallet_node.wallet_state_manager.lca
].height
== 63
):
found = True
break
await asyncio.sleep(0.1)
if not found:
raise Exception(
f"Took too long to process blocks, stopped at: {time.time() - start}"
)
@pytest.mark.asyncio
async def test_fast_sync_wallet(self, wallet_node_starting_height):
num_blocks = 50
blocks = bt.get_consecutive_blocks(test_constants, num_blocks, [])
full_node_1, wallet_node, server_1, server_2 = wallet_node_starting_height
for i in range(1, len(blocks)):
async for _ in full_node_1.respond_block(
full_node_protocol.RespondBlock(blocks[i])
):
pass
await server_2.start_client(
PeerInfo(server_1._host, uint16(server_1._port)), None
)
start = time.time()
found = False
while time.time() - start < 60:
if (
wallet_node.wallet_state_manager.block_records[
wallet_node.wallet_state_manager.lca
].height
>= num_blocks - 6
):
found = True
break
await asyncio.sleep(0.1)
if not found:
raise Exception(
f"Took too long to process blocks, stopped at: {time.time() - start}"
)
@pytest.mark.asyncio
async def test_short_sync_wallet(self, wallet_node):
num_blocks = 8
blocks = bt.get_consecutive_blocks(test_constants, num_blocks, [], 10)
full_node_1, wallet_node, server_1, server_2 = wallet_node
for i in range(1, len(blocks)):
async for _ in full_node_1.respond_block(
full_node_protocol.RespondBlock(blocks[i])
):
pass
await server_2.start_client(
PeerInfo(server_1._host, uint16(server_1._port)), None
)
start = time.time()
while time.time() - start < 60:
# The second node should eventually catch up to the first one, and have the
# same tip at height num_blocks - 1.
if (
wallet_node.wallet_state_manager.block_records[
wallet_node.wallet_state_manager.lca
].height
== 6
):
return
await asyncio.sleep(0.1)
raise Exception(
f"Took too long to process blocks, stopped at: {time.time() - start}"
)

1
tests/rpc/test_rpc.py
View File

@ -68,6 +68,7 @@ class TestRpc:
assert len(await client.get_connections()) == 0
await client.open_connection(server_2._host, server_2._port)
await asyncio.sleep(2)
connections = await client.get_connections()
assert len(connections) == 1

0
tests/test_wallet.py → tests/wallet/test_wallet.py
View File

47
tests/wallet/test_wallet_store.py Normal file
View File

@ -0,0 +1,47 @@
# import asyncio
# from secrets import token_bytes
# from pathlib import Path
# from typing import Any, Dict
# import sqlite3
# import random
# import pytest
# from src.full_node.store import FullNodeStore
# from src.types.full_block import FullBlock
# from src.types.sized_bytes import bytes32
# from src.util.ints import uint32, uint64
# from tests.block_tools import BlockTools
# bt = BlockTools()
# test_constants: Dict[str, Any] = {
# "DIFFICULTY_STARTING": 5,
# "DISCRIMINANT_SIZE_BITS": 16,
# "BLOCK_TIME_TARGET": 10,
# "MIN_BLOCK_TIME": 2,
# "DIFFICULTY_EPOCH": 12, # The number of blocks per epoch
# "DIFFICULTY_DELAY": 3, # EPOCH / WARP_FACTOR
# }
# test_constants["GENESIS_BLOCK"] = bytes(
# bt.create_genesis_block(test_constants, bytes([0] * 32), b"0")
# )
# @pytest.fixture(scope="module")
# def event_loop():
# loop = asyncio.get_event_loop()
# yield loop
# class TestWalletStore:
# @pytest.mark.asyncio
# async def test_store(self):
# blocks = bt.get_consecutive_blocks(test_constants, 9, [], 9, b"0")
# db_filename = Path("blockchain_wallet_store_test.db")
# if db_filename.exists():
# db_filename.unlink()
# db = await FullNodeStore.create(db_filename)
# try:
# await db._clear_database()

152
tests/wallet/test_wallet_sync.py Normal file
View File

@ -0,0 +1,152 @@
import asyncio
import time
import pytest
from src.types.peer_info import PeerInfo
from src.protocols import full_node_protocol
from src.util.ints import uint16
from tests.setup_nodes import setup_two_nodes, setup_node_and_wallet, test_constants, bt
@pytest.fixture(scope="module")
def event_loop():
loop = asyncio.get_event_loop()
yield loop
class TestWalletSync:
@pytest.fixture(scope="function")
async def wallet_node(self):
async for _ in setup_node_and_wallet():
yield _
@pytest.fixture(scope="function")
async def wallet_node_starting_height(self):
async for _ in setup_node_and_wallet(dic={"starting_height": 100}):
yield _
@pytest.mark.asyncio
async def test_basic_sync_wallet(self, wallet_node):
num_blocks = 25
blocks = bt.get_consecutive_blocks(test_constants, num_blocks, [])
full_node_1, wallet_node, server_1, server_2 = wallet_node
for i in range(1, len(blocks)):
async for _ in full_node_1.respond_block(
full_node_protocol.RespondBlock(blocks[i])
):
pass
await server_2.start_client(
PeerInfo(server_1._host, uint16(server_1._port)), None
)
start = time.time()
found = False
while time.time() - start < 60:
# The second node should eventually catch up to the first one, and have the
# same tip at height num_blocks - 1.
if (
wallet_node.wallet_state_manager.block_records[
wallet_node.wallet_state_manager.lca
].height
>= num_blocks - 6
):
found = True
break
await asyncio.sleep(0.1)
if not found:
raise Exception(
f"Took too long to process blocks, stopped at: {time.time() - start}"
)
# Tests a reorg with the wallet
start = time.time()
found = False
blocks_reorg = bt.get_consecutive_blocks(test_constants, 45, blocks[:-5])
for i in range(1, len(blocks_reorg)):
async for msg in full_node_1.respond_block(
full_node_protocol.RespondBlock(blocks_reorg[i])
):
server_1.push_message(msg)
start = time.time()
while time.time() - start < 100:
if (
wallet_node.wallet_state_manager.block_records[
wallet_node.wallet_state_manager.lca
].height
== 63
):
found = True
break
await asyncio.sleep(0.1)
if not found:
raise Exception(
f"Took too long to process blocks, stopped at: {time.time() - start}"
)
@pytest.mark.asyncio
async def test_fast_sync_wallet(self, wallet_node_starting_height):
num_blocks = 50
blocks = bt.get_consecutive_blocks(test_constants, num_blocks, [])
full_node_1, wallet_node, server_1, server_2 = wallet_node_starting_height
for i in range(1, len(blocks)):
async for _ in full_node_1.respond_block(
full_node_protocol.RespondBlock(blocks[i])
):
pass
await server_2.start_client(
PeerInfo(server_1._host, uint16(server_1._port)), None
)
start = time.time()
found = False
while time.time() - start < 60:
if (
wallet_node.wallet_state_manager.block_records[
wallet_node.wallet_state_manager.lca
].height
>= num_blocks - 6
):
found = True
break
await asyncio.sleep(0.1)
if not found:
raise Exception(
f"Took too long to process blocks, stopped at: {time.time() - start}"
)
@pytest.mark.asyncio
async def test_short_sync_wallet(self, wallet_node):
num_blocks = 8
blocks = bt.get_consecutive_blocks(test_constants, num_blocks, [], 10)
full_node_1, wallet_node, server_1, server_2 = wallet_node
for i in range(1, len(blocks)):
async for _ in full_node_1.respond_block(
full_node_protocol.RespondBlock(blocks[i])
):
pass
await server_2.start_client(
PeerInfo(server_1._host, uint16(server_1._port)), None
)
start = time.time()
while time.time() - start < 60:
# The second node should eventually catch up to the first one, and have the
# same tip at height num_blocks - 1.
if (
wallet_node.wallet_state_manager.block_records[
wallet_node.wallet_state_manager.lca
].height
== 6
):
return
await asyncio.sleep(0.1)
raise Exception(
f"Took too long to process blocks, stopped at: {time.time() - start}"
)