fix: use a invite singleton

Author: abishekk92

Anchor.toml

@@ -6,10 +6,12 @@ slugger = "SAbD2TPKyTd54oahjz6UEBzweXvojsRWbGB2t21gDnB"
 [programs.devnet]
 wordcel = "D9JJgeRf2rKq5LNMHLBMb92g4ZpeMgCyvZkd7QKwSCzg"
 invite = "6G5x4Es2YZYB5e4QkFJN88TrfLABkYEQpkUH5Gob9Cut"
+slugger = "SAbD2TPKyTd54oahjz6UEBzweXvojsRWbGB2t21gDnB"
 
 [programs.mainnet]
 wordcel = "EXzAYHZ8xS6QJ6xGRsdKZXixoQBLsuMbmwJozm85jHp"
 invite = "Fc4q6ttyDHr11HjMHRvanG9SskeR24Q62egdwsUUMHLf"
+slugger = "SAbD2TPKyTd54oahjz6UEBzweXvojsRWbGB2t21gDnB"
 
 [registry]
 url = "https://anchor.projectserum.com"

tests/slugger.spec.ts

@@ -4,7 +4,7 @@ import {Wordcel} from '../target/types/wordcel';
 import {Slugger} from '../target/types/slugger';
 import {expect} from 'chai';
 import {PublicKey} from '@solana/web3.js';
-import {getInviteAccount, invitationProgram} from "./utils/invite";
+import {getInviteSingleton, invitationProgram} from "./utils/invite";
 import {airdrop} from './utils';
 import randombytes from 'randombytes';
 import {createHash} from 'crypto';
@@ -37,15 +37,7 @@ describe('Slugger', async () => {
     // Prepare test user.
     before(async () => {
         await airdrop(user);
-        inviteAccount = await getInviteAccount(user);
-        // Initialize invite
-        await invitationProgram.methods.initialize()
-            .accounts({
-                inviteAccount: inviteAccount,
-                authority: user,
-                payer: user,
-                systemProgram: SystemProgram.programId
-            }).rpc();
+        inviteAccount = await getInviteSingleton(user);
 
         // Set up a profile
         const profileHash = randombytes(32);

tests/utils/invite.ts

@@ -7,6 +7,8 @@ const {SystemProgram} = anchor.web3;
 const provider = anchor.getProvider();
 const invitationPrefix = Buffer.from("invite");
 
+let inviteMap = new Map<string, PublicKey>();
+
 export const invitationProgram = anchor.workspace.Invite as Program<Invite>;
 
 export async function getInviteAccount(key: PublicKey) {
@@ -33,3 +35,20 @@ export async function sendInvite(from_user: Keypair, to: PublicKey, feePayer: Pu
     await provider.sendAndConfirm(tx);
     return [inviteAccount, toInviteAccount];
 }
+
+export async function getInviteSingleton(user: PublicKey) {
+    const key = user.toString()
+    if(inviteMap.has(key)) {
+        return inviteMap.get(key)
+    }
+    const inviteAccount = await getInviteAccount(user);
+    await invitationProgram.methods.initialize()
+        .accounts({
+            inviteAccount: inviteAccount,
+            authority: user,
+            payer: user,
+            systemProgram: SystemProgram.programId
+        }).rpc();
+    inviteMap.set(key, inviteAccount);
+    return inviteAccount;
+}