Fix tests

Author: cmihaylov

test/unit/gemini.test.ts

@@ -34,10 +34,11 @@ describe('Gemini AI unit tests', () => {
         const res = await callGemini({ prompt: 'text prompt' })
 
         expect(res).not.toBe(null)
-        expect(res).toHaveLength(1)
+        expect(res).toHaveProperty('response')
+        expect(res.response).toHaveLength(1)
 
         const expectedStrategy = mockedStrategies[0]
-        const strategy = (res as Strategy[])[0]
+        const strategy = (res.response as Strategy[])[0]
         expect(strategy.name).toEqual(expectedStrategy.name)
         expect(strategy.risk).toEqual(expectedStrategy.risk)
         expect(strategy.actions.length).toEqual(expectedStrategy.actions.length)

test/unit/grok.test.ts

@@ -38,10 +38,11 @@ describe('Grok unit tests', () => {
         const res = await callGrok({ prompt: 'text prompt' })
 
         expect(res).not.toBe(null)
-        expect(res).toHaveLength(1)
+        expect(res).toHaveProperty('response')
+        expect(res.response).toHaveLength(1)
 
         const expectedStrategy = mockedStrategies[0]
-        const strategy = (res as Strategy[])[0]
+        const strategy = (res.response as Strategy[])[0]
         expect(strategy.name).toEqual(expectedStrategy.name)
         expect(strategy.risk).toEqual(expectedStrategy.risk)
         expect(strategy.actions.length).toEqual(expectedStrategy.actions.length)

test/unit/portfolio.test.ts

@@ -1,8 +1,8 @@
 import { EMPTY_PORTFOLIO_STRATEGIES } from '../../lib'
 import {
+    LlmProcessOutput,
     NetworkPortfolioLibResponse,
     PortfolioLibToken,
-    Strategy,
     StrategyRisk
 } from '../../lib/types'
 import {
@@ -31,18 +31,24 @@ const mockedNetworkPortfolioResult: NetworkPortfolioLibResponse = {
         }
     ]
 }
-const mockedStrategies: Strategy[] = [
-    {
-        actions: [
-            {
-                description: 'Example USDC strategy description',
-                tokens: 'USDC, ETH'
-            }
-        ],
-        name: 'Example USDC strategy name',
-        risk: StrategyRisk.LOW
-    }
-]
+const mockedLlmOutput: LlmProcessOutput = {
+    llm: {
+        provider: 'mock',
+        model: 'mock'
+    },
+    response: [
+        {
+            actions: [
+                {
+                    description: 'Example USDC strategy description',
+                    tokens: 'USDC, ETH'
+                }
+            ],
+            name: 'Example USDC strategy name',
+            risk: StrategyRisk.LOW
+        }
+    ]
+}
 
 jest.mock('ambire-common/dist/src/consts/networks', () => {
     const actual = jest.requireActual('ambire-common/dist/src/consts/networks')
@@ -99,30 +105,32 @@ describe('Portfolio unit tests', () => {
             address: TEST_WALLET,
             getPortfolio: getPortfolioVelcroV3,
             makePrompt: simplePrompt,
-            llmProcessor: () => Promise.resolve(mockedStrategies)
+            llmProcessor: () => Promise.resolve(mockedLlmOutput)
         })
 
         expect(res).toHaveProperty('address')
         expect(res).toHaveProperty('portfolio')
         expect(res).toHaveProperty('strategies')
         expect(res.address).toEqual(TEST_WALLET)
         expect(res.portfolio).toHaveLength(1)
-        expect(res.strategies).toBe(mockedStrategies)
+        expect(res.strategies).toHaveLength(1)
+        expect(res.strategies[0]).toBe(mockedLlmOutput)
     })
 
     test('should process address with empty portfolio and get hardcoded strategy for it', async () => {
         const res = await processAddress({
             address: TEST_WALLET,
             getPortfolio: () => Promise.resolve([]),
             makePrompt: simplePrompt,
-            llmProcessor: () => Promise.resolve(mockedStrategies)
+            llmProcessor: () => Promise.resolve(mockedLlmOutput)
         })
 
         expect(res).toHaveProperty('address')
         expect(res).toHaveProperty('portfolio')
         expect(res).toHaveProperty('strategies')
         expect(res.address).toEqual(TEST_WALLET)
         expect(res.portfolio).toHaveLength(0)
-        expect(res.strategies).toBe(EMPTY_PORTFOLIO_STRATEGIES)
+        expect(res.strategies).toHaveLength(1)
+        expect(res.strategies[0].response).toBe(EMPTY_PORTFOLIO_STRATEGIES)
     })
 })