fix up the unit tests

xmtp · Oct 25, 2023 · 0d85fe6 · 0d85fe6
1 parent d1f33d1
commit 0d85fe6
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 14 deletions.
diff --git a/...va/org/xmtp/android/library/ClientTest.kt → ...va/org/xmtp/android/library/ClientTest.kt b/...va/org/xmtp/android/library/ClientTest.kt → ...va/org/xmtp/android/library/ClientTest.kt
@@ -1,9 +1,15 @@
 package org.xmtp.android.library
 
+import androidx.test.ext.junit.runners.AndroidJUnit4
 import org.junit.Assert.assertEquals
 import org.junit.Test
+import org.junit.runner.RunWith
 import org.xmtp.android.library.messages.PrivateKeyBuilder
+import org.xmtp.android.library.messages.PrivateKeyBundleV1Builder
+import org.xmtp.android.library.messages.generate
+import org.xmtp.proto.message.contents.PrivateKeyOuterClass
 
+@RunWith(AndroidJUnit4::class)
 class ClientTest {
     @Test
     fun testTakesAWallet() {
@@ -20,6 +26,20 @@ class ClientTest {
         assert(preKey?.publicKey?.hasSignature() ?: false)
     }
 
+    @Test
+    fun testSerialization() {
+        val wallet = PrivateKeyBuilder()
+        val v1 =
+            PrivateKeyOuterClass.PrivateKeyBundleV1.newBuilder().build().generate(wallet = wallet)
+        val encodedData = PrivateKeyBundleV1Builder.encodeData(v1)
+        val v1Copy = PrivateKeyBundleV1Builder.fromEncodedData(encodedData)
+        val client = Client().buildFrom(v1Copy)
+        assertEquals(
+            wallet.address,
+            client.address
+        )
+    }
+
     @Test
     fun testCanBeCreatedWithBundle() {
         val fakeWallet = PrivateKeyBuilder()

diff --git a/library/src/test/java/org/xmtp/android/library/PrivateKeyBundleTest.kt b/library/src/test/java/org/xmtp/android/library/PrivateKeyBundleTest.kt
@@ -26,20 +26,6 @@ class PrivateKeyBundleTest {
         )
     }
 
-    @Test
-    fun testSerialization() {
-        val wallet = PrivateKeyBuilder()
-        val v1 =
-            PrivateKeyOuterClass.PrivateKeyBundleV1.newBuilder().build().generate(wallet = wallet)
-        val encodedData = PrivateKeyBundleV1Builder.encodeData(v1)
-        val v1Copy = PrivateKeyBundleV1Builder.fromEncodedData(encodedData)
-        val client = Client().buildFrom(v1Copy)
-        assertEquals(
-            wallet.address,
-            client.address
-        )
-    }
-
     @Test
     fun testKeyBundlesAreSigned() {
         val wallet = PrivateKeyBuilder()