79 lines
2.8 KiB
Python
79 lines
2.8 KiB
Python
|
from lbry.testcase import UnitDBTestCase
|
||
|
|
|
||
|
|
|
||
|
|
class TestClientDBSync(UnitDBTestCase):
|
||
|
|
|
||
|
|
async def asyncSetUp(self):
|
||
|
|
await super().asyncSetUp()
|
||
|
|
await self.add(self.coinbase())
|
||
|
|
|
||
|
|
async def test_process_inputs(self):
|
||
|
|
await self.add(self.tx())
|
||
|
|
await self.add(self.tx())
|
||
|
|
txo1, txo2a, txo2b, txo3a, txo3b = self.outputs
|
||
|
|
self.assertEqual([
|
||
|
|
(txo1.id, None),
|
||
|
|
(txo2b.id, None),
|
||
|
|
], await self.get_txis())
|
||
|
|
self.assertEqual([
|
||
|
|
(txo1.id, False),
|
||
|
|
(txo2a.id, False),
|
||
|
|
(txo2b.id, False),
|
||
|
|
(txo3a.id, False),
|
||
|
|
(txo3b.id, False),
|
||
|
|
], await self.get_txos())
|
||
|
|
await self.db.process_all_things_after_sync()
|
||
|
|
self.assertEqual([
|
||
|
|
(txo1.id, txo1.get_address(self.ledger)),
|
||
|
|
(txo2b.id, txo2b.get_address(self.ledger)),
|
||
|
|
], await self.get_txis())
|
||
|
|
self.assertEqual([
|
||
|
|
(txo1.id, True),
|
||
|
|
(txo2a.id, False),
|
||
|
|
(txo2b.id, True),
|
||
|
|
(txo3a.id, False),
|
||
|
|
(txo3b.id, False),
|
||
|
|
], await self.get_txos())
|
||
|
|
|
||
|
|
async def test_process_claims(self):
|
||
|
|
claim1 = await self.add(self.create_claim())
|
||
|
|
await self.db.process_all_things_after_sync()
|
||
|
|
self.assertEqual([claim1.claim_id], await self.get_claims())
|
||
|
|
|
||
|
|
claim2 = await self.add(self.create_claim())
|
||
|
|
self.assertEqual([claim1.claim_id], await self.get_claims())
|
||
|
|
await self.db.process_all_things_after_sync()
|
||
|
|
self.assertEqual([claim1.claim_id, claim2.claim_id], await self.get_claims())
|
||
|
|
|
||
|
|
claim3 = await self.add(self.create_claim())
|
||
|
|
claim4 = await self.add(self.create_claim())
|
||
|
|
await self.db.process_all_things_after_sync()
|
||
|
|
self.assertEqual([
|
||
|
|
claim1.claim_id,
|
||
|
|
claim2.claim_id,
|
||
|
|
claim3.claim_id,
|
||
|
|
claim4.claim_id,
|
||
|
|
], await self.get_claims())
|
||
|
|
|
||
|
|
await self.add(self.abandon_claim(claim4))
|
||
|
|
await self.db.process_all_things_after_sync()
|
||
|
|
self.assertEqual([
|
||
|
|
claim1.claim_id, claim2.claim_id, claim3.claim_id
|
||
|
|
], await self.get_claims())
|
||
|
|
|
||
|
|
# create and abandon in same block
|
||
|
|
claim5 = await self.add(self.create_claim())
|
||
|
|
await self.add(self.abandon_claim(claim5))
|
||
|
|
await self.db.process_all_things_after_sync()
|
||
|
|
self.assertEqual([
|
||
|
|
claim1.claim_id, claim2.claim_id, claim3.claim_id
|
||
|
|
], await self.get_claims())
|
||
|
|
|
||
|
|
# create and abandon in different blocks but with bulk sync
|
||
|
|
claim6 = await self.add(self.create_claim())
|
||
|
|
await self.add(self.abandon_claim(claim6))
|
||
|
|
await self.db.process_all_things_after_sync()
|
||
|
|
self.assertEqual([
|
||
|
|
claim1.claim_id, claim2.claim_id, claim3.claim_id
|
||
|
|
], await self.get_claims())
|