fix test_colliding_short_id

lbryio · Nov 9, 2021 · 552d9a7 · 552d9a7
1 parent c3c1dc4
commit 552d9a7
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tests/integration/takeovers/test_resolve_command.py b/tests/integration/takeovers/test_resolve_command.py
@@ -153,11 +153,17 @@ async def test_colliding_short_id(self):
         await self.assertResolvesToClaimId(
             f'@abc#{colliding_claim_ids[0][:1]}', first_claim
         )
+        collision_depth = 0
+        for c1, c2 in zip(colliding_claim_ids[0], colliding_claim_ids[1]):
+            if c1 == c2:
+                collision_depth += 1
+            else:
+                break
         await self.assertResolvesToClaimId(f'@abc#{colliding_claim_ids[0][:2]}', colliding_claim_ids[0])
         await self.assertResolvesToClaimId(f'@abc#{colliding_claim_ids[0][:7]}', colliding_claim_ids[0])
         await self.assertResolvesToClaimId(f'@abc#{colliding_claim_ids[0][:17]}', colliding_claim_ids[0])
         await self.assertResolvesToClaimId(f'@abc#{colliding_claim_ids[0]}', colliding_claim_ids[0])
-        await self.assertResolvesToClaimId(f'@abc#{colliding_claim_ids[1][:3]}', colliding_claim_ids[1])
+        await self.assertResolvesToClaimId(f'@abc#{colliding_claim_ids[1][:collision_depth + 1]}', colliding_claim_ids[1])
         await self.assertResolvesToClaimId(f'@abc#{colliding_claim_ids[1][:7]}', colliding_claim_ids[1])
         await self.assertResolvesToClaimId(f'@abc#{colliding_claim_ids[1][:17]}', colliding_claim_ids[1])
         await self.assertResolvesToClaimId(f'@abc#{colliding_claim_ids[1]}', colliding_claim_ids[1])