test_file_commands integration tests fixed

This commit is contained in:
Lex Berezhny 2019-05-05 15:35:41 -04:00
parent 7ca01511f4
commit 048aa07e80
3 changed files with 40 additions and 10 deletions

View file

@ -3474,10 +3474,11 @@ class Daemon(metaclass=JSONRPCServerType):
async def resolve(self, urls): async def resolve(self, urls):
results = await self.ledger.resolve(urls) results = await self.ledger.resolve(urls)
#if 'error' not in results: if results:
# await self.storage.save_claims_for_resolve([ claims = self.stream_manager._convert_to_old_resolve_output(results)
# value for value in results.values() if isinstance(value, Output) await self.storage.save_claims_for_resolve([
# ]) value for value in claims.values() if 'error' not in value
])
return results return results
def _old_get_temp_claim_info(self, tx, txo, address, claim_dict, name, bid): def _old_get_temp_claim_info(self, tx, txo, address, claim_dict, name, bid):

View file

@ -13,7 +13,8 @@ from lbrynet.stream.descriptor import StreamDescriptor
from lbrynet.stream.managed_stream import ManagedStream from lbrynet.stream.managed_stream import ManagedStream
from lbrynet.schema.claim import Claim from lbrynet.schema.claim import Claim
from lbrynet.schema.url import URL from lbrynet.schema.url import URL
from lbrynet.extras.daemon.storage import lbc_to_dewies from lbrynet.extras.daemon.storage import lbc_to_dewies, dewies_to_lbc
from lbrynet.wallet.transaction import Output
if typing.TYPE_CHECKING: if typing.TYPE_CHECKING:
from lbrynet.conf import Config from lbrynet.conf import Config
from lbrynet.blob.blob_manager import BlobManager from lbrynet.blob.blob_manager import BlobManager
@ -300,6 +301,33 @@ class StreamManager:
return None, existing_for_claim_id[0] return None, existing_for_claim_id[0]
return None, None return None, None
def _convert_to_old_resolve_output(self, resolves):
result = {}
for url, txo in resolves.items():
if isinstance(txo, Output):
tx_height = txo.tx_ref.height
best_height = self.wallet.ledger.headers.height
result[url] = {
'name': txo.claim_name,
'value': txo.claim,
'protobuf': binascii.hexlify(txo.claim.to_bytes()),
'claim_id': txo.claim_id,
'txid': txo.tx_ref.id,
'nout': txo.position,
'amount': dewies_to_lbc(txo.amount),
'effective_amount': txo.meta.get('effective_amount', 0),
'height': tx_height,
'confirmations': (best_height+1) - tx_height if tx_height > 0 else tx_height,
'claim_sequence': -1,
'address': txo.get_address(self.wallet.ledger),
'valid_at_height': txo.meta.get('activation_height', None),
'timestamp': self.wallet.ledger.headers[tx_height]['timestamp'],
'supports': []
}
else:
result[url] = txo
return result
@cache_concurrent @cache_concurrent
async def download_stream_from_uri(self, uri, exchange_rate_manager: 'ExchangeRateManager', async def download_stream_from_uri(self, uri, exchange_rate_manager: 'ExchangeRateManager',
timeout: typing.Optional[float] = None, timeout: typing.Optional[float] = None,
@ -327,7 +355,9 @@ class StreamManager:
if not URL.parse(uri).has_stream: if not URL.parse(uri).has_stream:
raise ResolveError("cannot download a channel claim, specify a /path") raise ResolveError("cannot download a channel claim, specify a /path")
try: try:
resolved_result = await asyncio.wait_for(self.wallet.ledger.resolve(uri), resolve_timeout) resolved_result = self._convert_to_old_resolve_output(
await asyncio.wait_for(self.wallet.ledger.resolve([uri]), resolve_timeout)
)
except asyncio.TimeoutError: except asyncio.TimeoutError:
raise ResolveTimeout(uri) raise ResolveTimeout(uri)
await self.storage.save_claims_for_resolve([ await self.storage.save_claims_for_resolve([

View file

@ -77,12 +77,11 @@ class FileCommands(CommandTestCase):
async def test_file_list_updated_metadata_on_resolve(self): async def test_file_list_updated_metadata_on_resolve(self):
await self.stream_create('foo', '0.01') await self.stream_create('foo', '0.01')
claim = await self.daemon.resolve('lbry://foo') txo = (await self.daemon.resolve(['lbry://foo']))['lbry://foo']
claim = claim['lbry://foo']['claim']['protobuf'].decode() claim = txo.claim
await self.daemon.jsonrpc_file_delete(claim_name='foo') await self.daemon.jsonrpc_file_delete(claim_name='foo')
txid = await self.blockchain_claim_name('bar', claim, '0.01') txid = await self.blockchain_claim_name('bar', hexlify(claim.to_bytes()).decode(), '0.01')
await self.daemon.jsonrpc_get('lbry://bar') await self.daemon.jsonrpc_get('lbry://bar')
claim = Claim.from_bytes(unhexlify(claim))
claim.stream.description = "fix typos, fix the world" claim.stream.description = "fix typos, fix the world"
await self.blockchain_update_name(txid, hexlify(claim.to_bytes()).decode(), '0.01') await self.blockchain_update_name(txid, hexlify(claim.to_bytes()).decode(), '0.01')
await self.daemon.jsonrpc_resolve('lbry://bar') await self.daemon.jsonrpc_resolve('lbry://bar')