chore: condense setup code in MemberAPI tests (#267)

digidem · Sep 18, 2023 · 09d6ad0 · 09d6ad0
1 parent 479f96f
commit 09d6ad0
Showing 1 changed file with 24 additions and 18 deletions.
diff --git a/tests/member-api.js b/tests/member-api.js
@@ -9,11 +9,9 @@ import { replicate } from './helpers/rpc.js'
 test('invite() sends expected project-related details', async (t) => {
   t.plan(4)
 
-  const projectKey = KeyManager.generateProjectKeypair().publicKey
-  const encryptionKeys = { auth: randomBytes(32) }
-  const projectInfo = { name: 'mapeo' }
+  const { projectKey, encryptionKeys, rpc: r1 } = setup()
 
-  const r1 = new MapeoRPC()
+  const projectInfo = { name: 'mapeo' }
   const r2 = new MapeoRPC()
 
   const memberApi = new MemberApi({
@@ -54,7 +52,8 @@ test('invite() sends expected project-related details', async (t) => {
 test('invite() assigns role to invited device after invite accepted', async (t) => {
   t.plan(4)
 
-  const r1 = new MapeoRPC()
+  const { projectKey, encryptionKeys, rpc: r1 } = setup()
+
   const r2 = new MapeoRPC()
 
   const expectedRoleId = randomBytes(8).toString('hex')
@@ -71,8 +70,8 @@ test('invite() assigns role to invited device after invite accepted', async (t)
 
   const memberApi = new MemberApi({
     capabilities,
-    encryptionKeys: { auth: randomBytes(32) },
-    projectKey: KeyManager.generateProjectKeypair().publicKey,
+    encryptionKeys,
+    projectKey,
     rpc: r1,
     dataTypes: {
       project: {
@@ -112,7 +111,8 @@ test('invite() does not assign role to invited device if invite is not accepted'
     t.test(decision, (t) => {
       t.plan(1)
 
-      const r1 = new MapeoRPC()
+      const { projectKey, encryptionKeys, rpc: r1 } = setup()
+
       const r2 = new MapeoRPC()
 
       const capabilities = {
@@ -126,8 +126,8 @@ test('invite() does not assign role to invited device if invite is not accepted'
 
       const memberApi = new MemberApi({
         capabilities,
-        encryptionKeys: { auth: randomBytes(32) },
-        projectKey: KeyManager.generateProjectKeypair().publicKey,
+        encryptionKeys,
+        projectKey,
         rpc: r1,
         dataTypes: {
           project: {
@@ -159,10 +159,7 @@ test('invite() does not assign role to invited device if invite is not accepted'
 })
 
 test('getById() works', async (t) => {
-  const projectKey = KeyManager.generateProjectKeypair().publicKey
-  const encryptionKeys = { auth: randomBytes(32) }
-
-  const rpc = new MapeoRPC()
+  const { projectKey, encryptionKeys, rpc } = setup()
 
   const deviceId = randomBytes(32).toString('hex')
 
@@ -204,10 +201,7 @@ test('getById() works', async (t) => {
 })
 
 test('getMany() works', async (t) => {
-  const projectKey = KeyManager.generateProjectKeypair().publicKey
-  const encryptionKeys = { auth: randomBytes(32) }
-
-  const rpc = new MapeoRPC()
+  const { projectKey, encryptionKeys, rpc } = setup()
 
   const deviceInfoRecords = []
 
@@ -250,6 +244,18 @@ test('getMany() works', async (t) => {
   }
 })
 
+function setup() {
+  const projectKey = KeyManager.generateProjectKeypair().publicKey
+  const encryptionKeys = { auth: randomBytes(32) }
+  const rpc = new MapeoRPC()
+
+  return {
+    projectKey,
+    encryptionKeys,
+    rpc,
+  }
+}
+
 /**
  * @param {Object} opts
  * @param {string} [opts.deviceId]