From eeb9e66823111c394b344c2c56d59bed707dee1c Mon Sep 17 00:00:00 2001
From: himanshu <himanshuchawla2014@gmail.com>
Date: Fri, 7 Feb 2025 13:00:12 +0530
Subject: [PATCH] fix tests

---
 src/helpers/nodeUtils.ts     | 8 ++++++--
 test/onekey.test.ts          | 2 +-
 test/sapphire_devnet.test.ts | 2 +-
 3 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/helpers/nodeUtils.ts b/src/helpers/nodeUtils.ts
index 45b786e..bbae09e 100644
--- a/src/helpers/nodeUtils.ts
+++ b/src/helpers/nodeUtils.ts
@@ -456,9 +456,11 @@ export async function retrieveOrImportShare(params: {
 
   const promiseArrRequest = [];
 
-  // if dkg is not used, we need to get existing key or import new shares from client
-  const getExistingKeyOrImportNewShares = !useDkg;
+  // if dkg is not used and may be its not a existing key,
+  // we need to get existing key or import new shares from client
+  const getExistingKeyOrImportNewShares = !useDkg && !isExistingKey;
   if (overrideExistingKey) {
+    // import new key flow
     isImportingShares = true;
     const proxyEndpointNum = getProxyCoordinatorEndpointIndex(endpoints, verifier, verifierParams.verifier_id);
     const items: Record<string, unknown>[] = [];
@@ -503,6 +505,7 @@ export async function retrieveOrImportShare(params: {
     ).catch((err) => log.error("share req", err));
     promiseArrRequest.push(p);
   } else if (getExistingKeyOrImportNewShares) {
+    // no dkg login and registration flow
     const proxyEndpointNum = getProxyCoordinatorEndpointIndex(endpoints, verifier, verifierParams.verifier_id);
     const items: Record<string, unknown>[] = [];
     for (let i = 0; i < endpoints.length; i += 1) {
@@ -547,6 +550,7 @@ export async function retrieveOrImportShare(params: {
     ).catch((err) => log.error("share req", err));
     promiseArrRequest.push(p);
   } else {
+    // dkg login and registration flow
     for (let i = 0; i < endpoints.length; i += 1) {
       const p = post<JRPCResponse<ShareRequestResult>>(
         endpoints[i],
diff --git a/test/onekey.test.ts b/test/onekey.test.ts
index 3b5890a..73811c9 100644
--- a/test/onekey.test.ts
+++ b/test/onekey.test.ts
@@ -177,7 +177,7 @@ describe("torus onekey", function () {
     expect(publicAddress.oAuthKeyData.walletAddress).to.not.equal(null);
   });
 
-  it("should be able to key assign via login", async function () {
+  it.only("should be able to key assign via login", async function () {
     const email = faker.internet.email();
     const token = generateIdToken(email, "ES256");
     const verifierDetails = { verifier: TORUS_TEST_VERIFIER, verifierId: email };
diff --git a/test/sapphire_devnet.test.ts b/test/sapphire_devnet.test.ts
index 2f7e819..5c6126e 100644
--- a/test/sapphire_devnet.test.ts
+++ b/test/sapphire_devnet.test.ts
@@ -381,7 +381,7 @@ describe("torus utils sapphire devnet", function () {
         token,
         nodeDetails.torusNodePub,
         {},
-        true,
+        false,
         false
       )
     );