@steerprotocol/sdk 1.30.8-test-algebra-plugins.2 → 1.31.0

package/src/__tests__/base/vault/single-asset/calculateLimitPrice.test.ts

@@ -13,15 +13,18 @@ describe('calculateLimitPrice', () => {
 
 
 
-  beforeAll(() => { // Create real clients for blockchain interaction
-      publicClient = createPublicClient({chain: polygon, transport: http()});
+  beforeEach(() => { // Create real clients for blockchain interaction
+      publicClient = createPublicClient({
+        chain: polygon,
+        transport: http('https://polygon-bor-rpc.publicnode.com')
+      });
 
 
       // Initialize SteerClient with real clients
   });
 
   describe('successful calculations', () => {
-    it.only('should calculate limit price for token0 → token1 swap (zeroForOne: true)', async () => {
+    it('should calculate limit price for token0 → token1 swap (zeroForOne: true)', async () => {
       const result = await calculateLimitPrice(publicClient, {
         pool: mockPoolAddress,
         slippageBP: 500, // 5% slippage
@@ -137,7 +140,16 @@ describe('calculateLimitPrice', () => {
   });
 
   describe('error handling', () => {
-    it('should reject unsupported AMM types', async () => {
+    it('should process Algebra AMM pools when pool state is available', async () => {
+      publicClient.readContract = jest.fn().mockResolvedValueOnce([
+        BigInt('1753088363026579664086182258483316'),
+        200101,
+        0,
+        0,
+        0,
+        true,
+      ]).mockResolvedValueOnce(18);
+
       const result = await calculateLimitPrice(publicClient, {
         pool: mockPoolAddress,
         slippageBP: 500,
@@ -147,9 +159,8 @@ describe('calculateLimitPrice', () => {
         token1: mockToken1
       });
 
-      expect(result.success).toBe(false);
-      expect(result.status).toBe(500);
-      expect(result.error).toContain('AMM type 1 not yet supported');
+      expect(result.success).toBe(true);
+      expect(result.status).toBe(200);
     });
 
     it('should handle contract read failures', async () => {
@@ -166,7 +177,7 @@ describe('calculateLimitPrice', () => {
 
       expect(result.success).toBe(false);
       expect(result.status).toBe(500);
-      expect(result.error).toContain('Failed to calculate limit price');
+      expect(result.error).toContain('Contract read failed');
     });
 
     it('should handle zero slippage', async () => {
@@ -303,12 +314,19 @@ describe('calculateLimitPrice', () => {
       });
     });
 
-    it('should reject unsupported AMM types for slot0', async () => {
+    it('should support Algebra slot0 reads when pool globalState exists', async () => {
+      publicClient.readContract = jest.fn().mockResolvedValue([
+        BigInt('1753088363026579664086182258483316'),
+        200101,
+        0,
+        0,
+        0,
+        true,
+      ]);
       const result = await getPoolSlot0(publicClient, mockPoolAddress, AMMType.Algebra);
 
-      expect(result.success).toBe(false);
-      expect(result.status).toBe(500);
-      expect(result.error).toContain('AMM type 1 not yet supported');
+      expect(result.success).toBe(true);
+      expect(result.status).toBe(200);
     });
 
     it('should handle contract read failures for slot0', async () => {
@@ -318,7 +336,7 @@ describe('calculateLimitPrice', () => {
 
       expect(result.success).toBe(false);
       expect(result.status).toBe(500);
-      expect(result.error).toContain('Failed to get pool slot0 data');
+      expect(result.error).toContain('Contract read failed');
     });
   });
-}); 
+}); 

package/src/__tests__/base/vault/single-asset/calculateSwapAmount.test.ts

@@ -55,13 +55,16 @@ describe('calculateSwapAmount', () => {
     const mockContractAddress = '0x9903C4ED61A71d0d5C322097adfF752d6339F871';
 
 
-    beforeAll(() => { // Create real clients for blockchain interaction
-        publicClient = createPublicClient({chain: polygon, transport: http()});
+    beforeEach(() => { // Create real clients for blockchain interaction
+        publicClient = createPublicClient({
+            chain: polygon,
+            transport: http('https://polygon-bor-rpc.publicnode.com')
+        });
 
         // Initialize SteerClient with real clients
     });
     describe('successful calculations', () => {
-        it.only('should calculate swap amount for token0 deposit', async () => {
+        it('should calculate swap amount for token0 deposit', async () => {
 
             const result = await calculateSwapAmount(publicClient, {
                 depositAmount: parseEther('100000'),
@@ -198,7 +201,7 @@ describe('calculateSwapAmount', () => {
         });
 
         it('should handle invalid contract response', async () => { // Mock invalid response (not an array)
-            publicClient.readContract = jest.fn().mockResolvedValue('invalid response');
+            publicClient.readContract = jest.fn().mockResolvedValue(undefined);
 
             const result = await calculateSwapAmount(publicClient, {
                 depositAmount: parseEther('100'),