From b14df719747c984eece5410a66ba6012e71ec33c Mon Sep 17 00:00:00 2001
From: Koen <koen@pop-os.localdomain>
Date: Wed, 13 Sep 2023 09:29:12 +0200
Subject: [PATCH] More bundle verification.

---
 .../core/GenerateTestProfileTests.kt          | 21 ++++++++++++-------
 1 file changed, 14 insertions(+), 7 deletions(-)

diff --git a/app/src/androidTest/java/com/futo/polycentric/core/GenerateTestProfileTests.kt b/app/src/androidTest/java/com/futo/polycentric/core/GenerateTestProfileTests.kt
index c780f09..40c786a 100644
--- a/app/src/androidTest/java/com/futo/polycentric/core/GenerateTestProfileTests.kt
+++ b/app/src/androidTest/java/com/futo/polycentric/core/GenerateTestProfileTests.kt
@@ -54,13 +54,20 @@ class GenerateTestProfileTests {
         }
 
         //Verify resolve claim
-        val resolveClaimAnyField = ApiMethods.getResolveClaim(TestConstants.SERVER, s2p1.system.toProto(), 2, "UCR7KMD7jkSefYYWgSwNPEBA")
-        assertEquals(1, resolveClaimAnyField.matchesCount)
-        assertEquals(s1p1.system, SignedEvent.fromProto(resolveClaimAnyField.getMatches(0).claim).event.system)
-
-        val resolveClaimSpecificField = ApiMethods.getResolveClaim(TestConstants.SERVER, s2p1.system.toProto(), 2, 1, "UCR7KMD7jkSefYYWgSwNPEBA")
-        assertEquals(1, resolveClaimSpecificField.matchesCount)
-        assertEquals(s1p1.system, SignedEvent.fromProto(resolveClaimSpecificField.getMatches(0).claim).event.system)
+        for (pair in listOf(
+            Pair(ClaimType.YOUTUBE.value, Pair(1, "UCR7KMD7jkSefYYWgSwNPEBA")),
+            Pair(ClaimType.YOUTUBE.value, Pair(0, "@koen-futo")),
+            Pair(ClaimType.RUMBLE.value, Pair(0, "koenfuto")),
+            Pair(ClaimType.RUMBLE.value, Pair(1, "c-3366838"))
+        )) {
+            val resolveClaimAnyField = ApiMethods.getResolveClaim(TestConstants.SERVER, s2p1.system.toProto(), 2, "UCR7KMD7jkSefYYWgSwNPEBA")
+            assertEquals(1, resolveClaimAnyField.matchesCount)
+            assertEquals(s1p1.system, SignedEvent.fromProto(resolveClaimAnyField.getMatches(0).claim).event.system)
+
+            val resolveClaimSpecificField = ApiMethods.getResolveClaim(TestConstants.SERVER, s2p1.system.toProto(), 2, 1, "UCR7KMD7jkSefYYWgSwNPEBA")
+            assertEquals(1, resolveClaimSpecificField.matchesCount)
+            assertEquals(s1p1.system, SignedEvent.fromProto(resolveClaimSpecificField.getMatches(0).claim).event.system)
+        }
 
         Log.i("EXPORT_BUNDLE s1p1", createExportBundle(s1p1))
         Log.i("EXPORT_BUNDLE s2p1", createExportBundle(s2p1))
-- 
GitLab