Fix: Create from key bundle

example/src/tests.ts

@@ -99,51 +99,102 @@
 })
 
 test('can make a MLS V3 client', async () => {
   const client = await Client.createRandom({
     env: 'local',
     appVersion: 'Testing/0.0.0',
-    enableAlphaMls: true
+    enableAlphaMls: true,
+  })
+
+  return true
+})
+
+test('can make a MLS V3 client from bundle', async () => {
+  const client = await Client.createRandom({
+    env: 'local',
+    appVersion: 'Testing/0.0.0',
+    enableAlphaMls: true,
   })
 
+  const anotherClient = await Client.createRandom({
+    env: 'local',
+    appVersion: 'Testing/0.0.0',
+    enableAlphaMls: true,
+  })
+
+  const group1 = await client.conversations.newGroup([anotherClient.address])
+
+  if (group1.clientAddress !== client.address) {
+    throw new Error(
+      `clients dont match ${client.address} and ${group1.clientAddress}`
+    )
+  }
+  const bundle = await client.exportKeyBundle()
+
+  const client2 = await Client.createFromKeyBundle(bundle, {
+    env: 'local',
+    appVersion: 'Testing/0.0.0',
+    enableAlphaMls: true,
+  })
+
+  if (client.address !== client2.address) {
+    throw new Error(
+      `clients dont match ${client2.address} and ${client.address}`
+    )
+  }
+
+  const randomClient = await Client.createRandom({
+    env: 'local',
+    appVersion: 'Testing/0.0.0',
+    enableAlphaMls: true,
+  })
+
+  const group = await client2.conversations.newGroup([randomClient.address])
+
+  if (group.clientAddress !== client2.address) {
+    throw new Error(
+      `clients dont match ${client2.address} and ${group.clientAddress}`
+    )
+  }
+
   return true
 })
 
 test('production MLS V3 client creation throws error', async () => {
   try {
     const client = await Client.createRandom({
       env: 'production',
       appVersion: 'Testing/0.0.0',
       enableAlphaMls: true
     })
   } catch (error: any) {
     return error.message.endsWith("Environment must be \"local\" or \"dev\" to enable alpha MLS")
   }
   throw new Error('should throw error on MLS V3 client create when environment is not local')
 })
 
 test('can message in a group', async () => {
   // Create three MLS enabled Clients
   const aliceClient = await Client.createRandom({
     env: 'local',
     enableAlphaMls: true
   })
   const bobClient = await Client.createRandom({
     env: 'local',
     enableAlphaMls: true
   })
   const camClient = await Client.createRandom({
     env: 'local',
     enableAlphaMls: true
   })
 
   // Alice's num groups start at 0
   let aliceGroups = await aliceClient.conversations.listGroups()
   if (aliceGroups.length != 0) {
     throw new Error('num groups should be 0')
   }
 
   // Alice creates a group
   const aliceGroup = await aliceClient.conversations.newGroup(
     [bobClient.address, camClient.address]
   )
 

src/index.ts

@@ -71,12 +71,14 @@ export async function createRandom(
 export async function createFromKeyBundle(
   keyBundle: string,
   environment: 'local' | 'dev' | 'production',
-  appVersion?: string | undefined
+  appVersion?: string | undefined,
+  enableAlphaMls?: boolean | undefined
 ): Promise<string> {
   return await XMTPModule.createFromKeyBundle(
     keyBundle,
     environment,
-    appVersion
+    appVersion,
+    enableAlphaMls
   )
 }
 

src/lib/Client.ts

@@ -186,7 +186,8 @@ export class Client<ContentTypes> {
     const address = await XMTPModule.createFromKeyBundle(
       keyBundle,
       options.env,
-      options.appVersion
+      options.appVersion,
+      Boolean(options.enableAlphaMls)
     )
     return new Client(address, opts?.codecs || [])
   }