fix: reset account during tests

ieow · ieow · commit fcc7b412d7b6 · 2025-11-27T17:43:35.000+08:00
diff --git a/tests/ed25519.spec.ts b/tests/ed25519.spec.ts
@@ -6,7 +6,7 @@ import { UX_MODE_TYPE } from "@toruslabs/customauth";
 import { tssLib } from "@toruslabs/tss-frost-lib";
 import BN from "bn.js";
 
-import { AsyncStorage, COREKIT_STATUS, ed25519, MemoryStorage, TssShareType, WEB3AUTH_NETWORK, WEB3AUTH_NETWORK_TYPE, Web3AuthMPCCoreKit } from "../src";
+import { AsyncStorage, COREKIT_STATUS, ed25519, MemoryStorage,  WEB3AUTH_NETWORK, WEB3AUTH_NETWORK_TYPE, Web3AuthMPCCoreKit } from "../src";
 import { bufferToElliptic, criticalResetAccount, mockLogin, mockLogin2 } from "./setup";
 
 type TestVariable = {
@@ -64,7 +64,7 @@ variable.forEach((testVariable) => {
       verifierId: parsedToken.email,
       idToken,
     });
-    await criticalResetAccount(resetInstance);
+    await criticalResetAccount(resetInstance, manualSync);
     await new AsyncStorage(resetInstance._storageKey, storageInstance).resetStore();
   }
 
diff --git a/tests/factors.spec.ts b/tests/factors.spec.ts
@@ -75,7 +75,7 @@ export const FactorManipulationTest = async (testVariable: FactorTestVariable) =
 
   async function beforeTest() {
     const resetInstance = await newInstance();
-    await criticalResetAccount(resetInstance);
+    await criticalResetAccount(resetInstance, testVariable.manualSync);
     await resetInstance.logout();
     await new AsyncStorage(resetInstance._storageKey, testVariable.storage).resetStore();
   }
diff --git a/tests/gating.spec.ts b/tests/gating.spec.ts
@@ -60,7 +60,7 @@ variable.forEach((testVariable) => {
   const testNameSuffix = testVariable.description;
   test(`#Gating test :  ${testNameSuffix}`, async (t) => {
     async function beforeTest() {
-      if (coreKitInstance.status === COREKIT_STATUS.INITIALIZED) await criticalResetAccount(coreKitInstance);
+      if (coreKitInstance.status === COREKIT_STATUS.INITIALIZED) await criticalResetAccount(coreKitInstance, manualSync);
     }
 
     t.after(async function () {
diff --git a/tests/importRecovery.spec.ts b/tests/importRecovery.spec.ts
@@ -2,7 +2,6 @@ import assert from "node:assert";
 import test from "node:test";
 
 import { tssLib as tssLibDKLS } from "@toruslabs/tss-dkls-lib";
-import { tssLib as tssLibFROST } from "@toruslabs/tss-frost-lib";
 
 import {  MemoryStorage, sigToRSV, TssLibType, TssShareType, WEB3AUTH_NETWORK, Web3AuthMPCCoreKit } from "../src";
 import { bufferToElliptic, criticalResetAccount, mockLogin  } from "./setup";
@@ -55,10 +54,7 @@ export const ImportTest = async (testVariable: ImportKeyTestVariable) => {
   async function resetAccount(email: string) {
     const kit = await newCoreKitInstance(email);
     console.log('tss pub key', kit.state.tssPubKey)
-    await criticalResetAccount(kit);
-    if (testVariable.manualSync) {
-      await kit.commitChanges();
-    }
+    await criticalResetAccount(kit, testVariable.manualSync);
     await kit.logout();
     // await new AsyncStorage(kit._storageKey, storageInstance).resetStore();
   }
diff --git a/tests/login-disableHashedFactorKey.spec.ts b/tests/login-disableHashedFactorKey.spec.ts
@@ -78,7 +78,7 @@ variable.forEach((testVariable) => {
         verifierId: parsedToken.email,
         idToken,
       });
-      await criticalResetAccount(resetInstance);
+      await criticalResetAccount(resetInstance, manualSync);
       await new AsyncStorage(resetInstance._storageKey, storageInstance).resetStore();
     }
 
diff --git a/tests/login.spec.ts b/tests/login.spec.ts
@@ -76,7 +76,7 @@ variable.forEach((testVariable) => {
         verifierId: parsedToken.email,
         idToken,
       });
-      await criticalResetAccount(resetInstance);
+      await criticalResetAccount(resetInstance, manualSync);
       await new AsyncStorage(resetInstance._storageKey, storageInstance).resetStore();
     }
 
diff --git a/tests/securityQuestion.spec.ts b/tests/securityQuestion.spec.ts
@@ -29,7 +29,7 @@ export const TssSecurityQuestionsTest = async (newInstance: () => Promise<Web3Au
   test(`#Tss Security Question - ${testVariable.manualSync} `, async function (t) {
     async function beforeTest() {
       const coreKitInstance = await newInstance();
-      await criticalResetAccount(coreKitInstance);
+      await criticalResetAccount(coreKitInstance, testVariable.manualSync);
       await coreKitInstance.logout();
 
       await new AsyncStorage(coreKitInstance._storageKey, storageInstance).resetStore();
diff --git a/tests/sessionTime.spec.ts b/tests/sessionTime.spec.ts
@@ -76,7 +76,7 @@ variable.forEach(async (testVariable) => {
     });
 
     async function beforeTest() {
-      if (coreKitInstance.status === COREKIT_STATUS.INITIALIZED) await criticalResetAccount(coreKitInstance);
+      if (coreKitInstance.status === COREKIT_STATUS.INITIALIZED) await criticalResetAccount(coreKitInstance, manualSync);
     }
 
     t.after(async function () {
@@ -86,7 +86,7 @@ variable.forEach(async (testVariable) => {
 
     await beforeTest();
 
-    await t.test("`sessionTime` should be equal to `sessionTokenDuration` from #Login", async function (t) {
+    await t.test("`sessionTime` should be equal to `sessionTokenDuration` from #Login", async function () {
       // mocklogin
       const { idToken, parsedToken } = await mockLogin(email);
 
diff --git a/tests/setup.ts b/tests/setup.ts
@@ -22,7 +22,7 @@ export const mockLogin2 = async (email: string) => {
   return { idToken, parsedToken };
 };
 
-export const criticalResetAccount = async (coreKitInstance: Web3AuthMPCCoreKit): Promise<void> => {
+export const criticalResetAccount = async (coreKitInstance: Web3AuthMPCCoreKit, manualSync: boolean): Promise<void> => {
   // This is a critical function that should only be used for testing purposes
   // Resetting your account means clearing all the metadata associated with it from the metadata server
   // The key details will be deleted from our server and you will not be able to recover your account
@@ -37,6 +37,9 @@ export const criticalResetAccount = async (coreKitInstance: Web3AuthMPCCoreKit):
       privKey: new BN(coreKitInstance.state.postBoxKey!, "hex"),
       input: { message: "KEY_NOT_FOUND" },
     });
+    if (manualSync) {
+      await coreKitInstance.commitChanges();
+    }
   }
 };
 
diff --git a/tests/sfaImport.spec.ts b/tests/sfaImport.spec.ts
@@ -28,7 +28,7 @@ export const ImportSFATest = async (testVariable: ImportKeyTestVariable) => {
 
   async function resetAccount(email: string) {
     const kit = await newCoreKitInstance(email);
-    await criticalResetAccount(kit);
+    await criticalResetAccount(kit, testVariable.manualSync);
     await kit.logout();
     await new AsyncStorage(kit._storageKey, storageInstance).resetStore();
   }

Original file line number	Diff line number	Diff line change
`@@ -75,7 +75,7 @@ export const FactorManipulationTest = async (testVariable: FactorTestVariable) =`
`75`	`75`
`76`	`76`	`async function beforeTest() {`
`77`	`77`	`const resetInstance = await newInstance();`
`78`		`- await criticalResetAccount(resetInstance);`
	`78`	`+ await criticalResetAccount(resetInstance, testVariable.manualSync);`
`79`	`79`	`await resetInstance.logout();`
`80`	`80`	`await new AsyncStorage(resetInstance._storageKey, testVariable.storage).resetStore();`
`81`	`81`	`}`
Original file line number	Diff line number	Diff line change
`@@ -60,7 +60,7 @@ variable.forEach((testVariable) => {`
`60`	`60`	`const testNameSuffix = testVariable.description;`
`61`	`61`	test(`#Gating test : ${testNameSuffix}`, async (t) => {
`62`	`62`	`async function beforeTest() {`
`63`		`- if (coreKitInstance.status === COREKIT_STATUS.INITIALIZED) await criticalResetAccount(coreKitInstance);`
	`63`	`+ if (coreKitInstance.status === COREKIT_STATUS.INITIALIZED) await criticalResetAccount(coreKitInstance, manualSync);`
`64`	`64`	`}`
`65`	`65`
`66`	`66`	`t.after(async function () {`
Original file line number	Diff line number	Diff line change
`@@ -78,7 +78,7 @@ variable.forEach((testVariable) => {`
`78`	`78`	`verifierId: parsedToken.email,`
`79`	`79`	`idToken,`
`80`	`80`	`});`
`81`		`- await criticalResetAccount(resetInstance);`
	`81`	`+ await criticalResetAccount(resetInstance, manualSync);`
`82`	`82`	`await new AsyncStorage(resetInstance._storageKey, storageInstance).resetStore();`
`83`	`83`	`}`
`84`	`84`
Original file line number	Diff line number	Diff line change
`@@ -76,7 +76,7 @@ variable.forEach((testVariable) => {`
`76`	`76`	`verifierId: parsedToken.email,`
`77`	`77`	`idToken,`
`78`	`78`	`});`
`79`		`- await criticalResetAccount(resetInstance);`
	`79`	`+ await criticalResetAccount(resetInstance, manualSync);`
`80`	`80`	`await new AsyncStorage(resetInstance._storageKey, storageInstance).resetStore();`
`81`	`81`	`}`
`82`	`82`
Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,7 @@ export const ImportSFATest = async (testVariable: ImportKeyTestVariable) => {`
`28`	`28`
`29`	`29`	`async function resetAccount(email: string) {`
`30`	`30`	`const kit = await newCoreKitInstance(email);`
`31`		`- await criticalResetAccount(kit);`
	`31`	`+ await criticalResetAccount(kit, testVariable.manualSync);`
`32`	`32`	`await kit.logout();`
`33`	`33`	`await new AsyncStorage(kit._storageKey, storageInstance).resetStore();`
`34`	`34`	`}`