fix claim integration tests

This commit is contained in:
Lex Berezhny 2019-07-20 13:30:41 -04:00
parent e4f802e5f4
commit b89f04b1f1

View file

@ -89,6 +89,7 @@ class ClaimSearchCommand(ClaimTestCase):
await self.assertFindsClaims([claim], **kwargs) await self.assertFindsClaims([claim], **kwargs)
async def assertFindsClaims(self, claims, **kwargs): async def assertFindsClaims(self, claims, **kwargs):
kwargs.setdefault('order_by', ['height', '^name'])
results = await self.claim_search(**kwargs) results = await self.claim_search(**kwargs)
self.assertEqual(len(claims), len(results)) self.assertEqual(len(claims), len(results))
for claim, result in zip(claims, results): for claim, result in zip(claims, results):
@ -165,15 +166,15 @@ class ClaimSearchCommand(ClaimTestCase):
await self.create_channel() await self.create_channel()
await self.create_lots_of_streams() await self.create_lots_of_streams()
page = await self.claim_search(page_size=20, channel='@abc') page = await self.claim_search(page_size=20, channel='@abc', order_by=['height', '^name'])
page_claim_ids = [item['name'] for item in page] page_claim_ids = [item['name'] for item in page]
self.assertEqual(page_claim_ids, self.streams) self.assertEqual(page_claim_ids, self.streams)
page = await self.claim_search(page_size=6, channel='@abc') page = await self.claim_search(page_size=6, channel='@abc', order_by=['height', '^name'])
page_claim_ids = [item['name'] for item in page] page_claim_ids = [item['name'] for item in page]
self.assertEqual(page_claim_ids, self.streams[:6]) self.assertEqual(page_claim_ids, self.streams[:6])
page = await self.claim_search(page=2, page_size=6, channel='@abc') page = await self.claim_search(page=2, page_size=6, channel='@abc', order_by=['height', '^name'])
page_claim_ids = [item['name'] for item in page] page_claim_ids = [item['name'] for item in page]
self.assertEqual(page_claim_ids, self.streams[6:]) self.assertEqual(page_claim_ids, self.streams[6:])