fix up the test helper

xmtp · Jan 23, 2024 · 14b3870 · 14b3870
1 parent 9bbee43
commit 14b3870
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/library/src/test/java/org/xmtp/android/library/TestHelpers.kt b/library/src/test/java/org/xmtp/android/library/TestHelpers.kt
@@ -42,18 +42,23 @@ class FakeWallet : SigningKey {
         }
     }
 
-    override val address: String
-        get() = privateKey.walletAddress
-
     override suspend fun sign(data: ByteArray): Signature {
         val signature = privateKeyBuilder.sign(data)
         return signature
     }
 
+    override fun sign(text: String): ByteArray {
+        return privateKeyBuilder.sign(text)
+    }
+
     override suspend fun signLegacy(message: String): Signature {
         val signature = privateKeyBuilder.signLegacy(message)
         return signature
     }
+
+    override fun getAddress(): String {
+        return privateKey.walletAddress
+    }
 }
 
 class FakeStreamHolder {
@@ -170,6 +175,7 @@ class FakeApiClient : ApiClient {
                 MessageApiOuterClass.SortDirection.SORT_DIRECTION_ASCENDING -> {
                     result = result.reversed().toMutableList()
                 }
+
                 else -> Unit
             }
         }