From e778b3678e7489c23204dd6968bf1e25ca73fecd Mon Sep 17 00:00:00 2001
From: "anubhava.gupta" <anubhava.gupta@salesforce.com>
Date: Wed, 6 May 2026 18:01:03 +0530
Subject: [PATCH] Using Real RSA key instead of random BigInteger generated
 RSAPrivateKeySpec

---
 .../einsteinbot/sdk/auth/JwtBearerOAuthTest.java    | 13 +++----------
 1 file changed, 3 insertions(+), 10 deletions(-)

diff --git a/src/test/java/com/salesforce/einsteinbot/sdk/auth/JwtBearerOAuthTest.java b/src/test/java/com/salesforce/einsteinbot/sdk/auth/JwtBearerOAuthTest.java
index 2b81986..010b894 100644
--- a/src/test/java/com/salesforce/einsteinbot/sdk/auth/JwtBearerOAuthTest.java
+++ b/src/test/java/com/salesforce/einsteinbot/sdk/auth/JwtBearerOAuthTest.java
@@ -16,14 +16,10 @@
 
 import com.salesforce.einsteinbot.sdk.cache.Cache;
 import java.io.IOException;
-import java.math.BigInteger;
-import java.security.KeyFactory;
 import java.security.PrivateKey;
-import java.security.spec.RSAPrivateKeySpec;
 import java.time.Instant;
 import java.time.temporal.ChronoUnit;
 import java.util.Optional;
-import java.util.Random;
 import okhttp3.mockwebserver.MockResponse;
 import okhttp3.mockwebserver.MockWebServer;
 import okhttp3.mockwebserver.RecordedRequest;
@@ -83,12 +79,9 @@ void initialize() {
   }
 
   private static PrivateKey createTestPrivateKey() throws Exception {
-    KeyFactory keyFactory = KeyFactory.getInstance("RSA");
-    BigInteger randomBigInteger = new BigInteger(512, new Random());
-
-    RSAPrivateKeySpec privateKeySpec = new RSAPrivateKeySpec(randomBigInteger, randomBigInteger);
-
-    return keyFactory.generatePrivate(privateKeySpec);
+    java.security.KeyPairGenerator keyGen = java.security.KeyPairGenerator.getInstance("RSA");
+    keyGen.initialize(2048);
+    return keyGen.generateKeyPair().getPrivate();
   }
 
   @Test