@steerprotocol/sdk 1.26.0 → 1.28.0

package/src/__tests__/base/SubgraphClient.test.ts

@@ -0,0 +1,99 @@
+import { createPublicClient, http } from 'viem';
+import { polygon } from 'viem/chains';
+import { SubgraphClient } from '../../base/SubgraphClient.js';
+import { SteerResponse } from '../../types.js';
+
+// Create a concrete implementation for testing
+class TestSubgraphClient extends SubgraphClient {
+  public async testQuery<T>(query: string, variables?: Record<string, unknown>): Promise<SteerResponse<T>> {
+    return this.executeSubgraphQuery<T>(query, variables);
+  }
+
+  public getTestSubgraphUrl(chainId: number): string | undefined {
+    return this.getSubgraphUrl(chainId);
+  }
+}
+
+
+const RPC_URL = 'https://polygon-mainnet.g.alchemy.com/v2/VjQCxdNfXfyvPacuKwjsrhOsWc_g6X27'
+
+describe('SubgraphClient', () => {
+  let client: TestSubgraphClient;
+  
+  beforeEach(() => {
+    const viemClient = createPublicClient({
+      chain: polygon,
+      transport: http(RPC_URL)
+    });
+    client = new TestSubgraphClient(viemClient);
+  });
+
+  describe('supportsChain', () => {
+    it('should return true for supported chains', () => {
+      const result = client.supportsChain(137);
+      expect(result).toBe(true);
+    });
+
+    it('should return false for unsupported chains', () => {
+      const result = client.supportsChain(999999);
+      expect(result).toBe(false);
+    });
+  });
+
+  describe('executeSubgraphQuery', () => {
+    it('should throw error when no active chain is found', async () => {
+      const clientWithNoChain = new TestSubgraphClient(createPublicClient({
+        transport: http(RPC_URL)
+      }));
+
+      await expect(clientWithNoChain.testQuery('{ test }'))
+        .rejects
+        .toThrow('No active chain found');
+    });
+
+    it('should handle GraphQL errors gracefully', async () => {
+      global.fetch = jest.fn().mockImplementationOnce(() =>
+        Promise.resolve({
+          ok: true,
+          status: 200,
+          json: () => Promise.resolve({ errors: [{ message: 'GraphQL Error' }] }),
+        })
+      );
+
+      const response = await client.testQuery('{ test }');
+      expect(response.success).toBe(false);
+      expect(response.error).toBeDefined();
+      expect(response.status).toBe(500);
+    });
+
+    it('should handle network errors gracefully', async () => {
+      global.fetch = jest.fn().mockImplementationOnce(() =>
+        Promise.resolve({
+          ok: false,
+          status: 404,
+        })
+      );
+
+      const response = await client.testQuery('{ test }');
+      expect(response.success).toBe(false);
+      expect(response.error).toContain('HTTP error');
+      expect(response.status).toBe(500);
+    });
+
+    it('should handle successful queries', async () => {
+      const mockData = { protocol: { id: '1', totalValueLockedUSD: '1000000' } };
+      global.fetch = jest.fn().mockImplementationOnce(() =>
+        Promise.resolve({
+          ok: true,
+          status: 200,
+          json: () => Promise.resolve({ data: mockData }),
+        })
+      );
+
+      const response = await client.testQuery('{ protocol { id } }');
+      expect(response.success).toBe(true);
+      expect(response.status).toBe(200);
+      expect(response.data).toEqual(mockData);
+    });
+  });
+});