LBRYCommunity/lbry-sdk
1
0
Fork 1
Code Issues Pull requests Projects Releases 282 Packages Wiki Activity

updating wallet tests

Browse source
This commit is contained in:
Lex Berezhny 2018-06-14 01:32:11 -04:00 committed by Jack Robison
parent 2668ceac0e
commit a4e9f3100f
No known key found for this signature in database
GPG key ID: DF25C68FE0239BB2
2 changed files with 18 additions and 39 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
lbrynet/database/storage.py
View file

@ -11,7 +11,7 @@ from lbryschema.decode import smart_decode
from lbrynet import conf
from lbrynet.cryptstream.CryptBlob import CryptBlobInfo
from lbrynet.dht.constants import dataExpireTimeout
from lbrynet.wallet.database import SQLiteWalletStorage
from lbrynet.wallet.database import WalletDatabase
from torba.constants import COIN
log = logging.getLogger(__name__)
@ -95,7 +95,7 @@ class SqliteConnection(adbapi.ConnectionPool):
cls.reactor = reactor
class SQLiteStorage(SQLiteWalletStorage):
class SQLiteStorage(WalletDatabase):
CREATE_TABLES_QUERY = """
pragma foreign_keys=on;
@ -166,7 +166,7 @@ class SQLiteStorage(SQLiteWalletStorage):
timestamp integer,
primary key (sd_hash, reflector_address)
);
""" + SQLiteWalletStorage.CREATE_TABLES_QUERY
""" + WalletDatabase.CREATE_TABLES_QUERY
def __init__(self, db_dir, reactor=None):
if not reactor:

51
lbrynet/tests/integration/wallet/test_transactions.py
View file

@ -2,30 +2,8 @@ import asyncio
from binascii import hexlify, unhexlify
from orchstr8.testcase import IntegrationTestCase
from lbryschema.claim import ClaimDict
class BasicTransactionTests(IntegrationTestCase):
VERBOSE = True
async def test_sending_and_recieving(self):
self.assertEqual(await self.get_balance(), 0.0)
address = self.account.get_least_used_receiving_address()
sendtxid = await self.blockchain.send_to_address(address.decode(), 5.5)
await self.blockchain.generate(1)
await self.on_transaction(sendtxid)
self.assertAlmostEqual(await self.get_balance(), 5.5, places=2)
lbrycrd_address = await self.blockchain.get_raw_change_address()
tx = self.manager.send_amount_to_address(5, lbrycrd_address)
await self.broadcast(tx)
await self.on_transaction(tx.id.decode())
await self.blockchain.generate(1)
self.assertAlmostEqual(await self.get_balance(), 0.5, places=2)
from torba.constants import COIN
from lbrynet.wallet.manager import LbryWalletManager
example_claim_dict = {
@ -58,28 +36,29 @@ example_claim_dict = {
class ClaimTransactionTests(IntegrationTestCase):
VERBOSE = True
WALLET_MANAGER = LbryWalletManager
async def test_creating_updating_and_abandoning_claim(self):
address = self.account.get_least_used_receiving_address()
sendtxid = await self.lbrycrd.send_to_address(address.decode(), 9.0)
await self.lbrycrd.generate(1)
await self.on_transaction(sendtxid)
await self.account.ensure_address_gap().asFuture(asyncio.get_event_loop())
self.assertAlmostEqual(self.manager.get_balance(), 9, places=2)
address = await self.account.receiving.get_or_create_usable_address().asFuture(asyncio.get_event_loop())
sendtxid = await self.blockchain.send_to_address(address.decode(), 10)
await self.on_transaction(sendtxid) #mempool
await self.blockchain.generate(1)
await self.on_transaction(sendtxid) #confirmed
self.assertEqual(round(await self.get_balance(self.account)/COIN, 1), 10.0)
claim = ClaimDict.load_dict(example_claim_dict)
tx = self.manager.claim_name(b'foo', 5, hexlify(claim.serialized))
tx = self.manager.claim_name(b'foo', 1*COIN, hexlify(claim.serialized))
await self.broadcast(tx)
await self.on_transaction(tx.id.decode())
await self.lbrycrd.generate(1)
await asyncio.sleep(2)
await self.on_transaction(tx.hex_id.decode()) #mempool
await self.blockchain.generate(1)
await self.on_transaction(tx.hex_id.decode()) #confirmed
self.assertAlmostEqual(self.manager.get_balance(), 9, places=2)
await asyncio.sleep(2)
response = await self.resolve('lbry://foo')
print(response)