starred/chia-blockchain
1
0
Fork 0
mirror of https://github.com/Chia-Network/chia-blockchain.git synced 2026-09-27 17:46:28 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add test.

Browse Source
This commit is contained in:
Florin Chirica
2022-02-22 20:21:28 +02:00
parent ed67d6bf15
commit 57fbcc6554
1 changed files with 19 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/core/data_layer/test_data_store.py
+19
View File
@@ -991,6 +991,25 @@ async def test_kv_diff(data_store: DataStore, tree_id: bytes32) -> None:
assert diffs == expected_diff
@pytest.mark.asyncio
async def test_kv_diff_2(data_store: DataStore, tree_id: bytes32) -> None:
node_hash = await data_store.insert(
key=b"000",
value=b"000",
tree_id=tree_id,
reference_node_hash=None,
side=None,
)
empty_hash = bytes32([0] * 32)
invalid_hash = bytes32([0] * 31 + [1])
diff_1 = await data_store.get_kv_diff(tree_id, empty_hash, node_hash)
assert diff_1 == set([DiffData(OperationType.INSERT, b"000", b"000")])
diff_2 = await data_store.get_kv_diff(tree_id, node_hash, empty_hash)
assert diff_2 == set([DiffData(OperationType.DELETE, b"000", b"000")])
diff_3 = await data_store.get_kv_diff(tree_id, invalid_hash, node_hash)
assert diff_3 == set()
@pytest.mark.asyncio
async def test_rollback_to_generation(data_store: DataStore, tree_id: bytes32) -> None:
await add_0123_example(data_store, tree_id)