claim_search by --claim_type=repost support

This commit is contained in:
Lex Berezhny 2020-02-12 11:51:35 -05:00
parent 6b745c53dc
commit 605fc8ecd8
3 changed files with 5 additions and 1 deletions

View file

@ -1,6 +1,7 @@
CLAIM_TYPES = { CLAIM_TYPES = {
'stream': 1, 'stream': 1,
'channel': 2, 'channel': 2,
'repost': 3
} }
STREAM_TYPES = { STREAM_TYPES = {

View file

@ -360,6 +360,7 @@ class SQLDB:
if isinstance(fee.amount, Decimal): if isinstance(fee.amount, Decimal):
claim_record['fee_amount'] = int(fee.amount*1000) claim_record['fee_amount'] = int(fee.amount*1000)
elif claim.is_repost: elif claim.is_repost:
claim_record['claim_type'] = CLAIM_TYPES['repost']
claim_record['reposted_claim_hash'] = claim.repost.reference.claim_hash claim_record['reposted_claim_hash'] = claim.repost.reference.claim_hash
elif claim.is_channel: elif claim.is_channel:
claim_record['claim_type'] = CLAIM_TYPES['channel'] claim_record['claim_type'] = CLAIM_TYPES['channel']

View file

@ -309,12 +309,14 @@ class ClaimSearchCommand(ClaimTestCase):
octet = await self.stream_create() octet = await self.stream_create()
video = await self.stream_create('chrome', file_path=self.video_file_name) video = await self.stream_create('chrome', file_path=self.video_file_name)
image = await self.stream_create('blank-image', data=self.image_data, suffix='.png') image = await self.stream_create('blank-image', data=self.image_data, suffix='.png')
repost = await self.stream_repost(self.get_claim_id(image))
channel = await self.channel_create() channel = await self.channel_create()
unknown = self.sout(tx) unknown = self.sout(tx)
# claim_type # claim_type
await self.assertFindsClaims([image, video, octet, unknown], claim_type='stream') await self.assertFindsClaims([image, video, octet, unknown], claim_type='stream')
await self.assertFindsClaims([channel], claim_type='channel') await self.assertFindsClaims([channel], claim_type='channel')
await self.assertFindsClaims([repost], claim_type='repost')
# stream_type # stream_type
await self.assertFindsClaims([octet, unknown], stream_types=['binary']) await self.assertFindsClaims([octet, unknown], stream_types=['binary'])
@ -322,7 +324,7 @@ class ClaimSearchCommand(ClaimTestCase):
await self.assertFindsClaims([image], stream_types=['image']) await self.assertFindsClaims([image], stream_types=['image'])
await self.assertFindsClaims([image, video], stream_types=['video', 'image']) await self.assertFindsClaims([image, video], stream_types=['video', 'image'])
# stream_type # media_type
await self.assertFindsClaims([octet, unknown], media_types=['application/octet-stream']) await self.assertFindsClaims([octet, unknown], media_types=['application/octet-stream'])
await self.assertFindsClaims([video], media_types=['video/mp4']) await self.assertFindsClaims([video], media_types=['video/mp4'])
await self.assertFindsClaims([image], media_types=['image/png']) await self.assertFindsClaims([image], media_types=['image/png'])