@@ -1540,14 +1540,16 @@ describe('API Methods', () => {
15401540describe ( 'API Handlers' , ( ) => {
15411541 const columnSendRequestStarted = jest . fn ( ) ;
15421542 const columnSendRequestResolved = jest . fn ( ) ;
1543- const sendRequestSpy = jest . fn ( ) ;
1543+ const sendRequestPromptSpy = jest . fn ( ) ;
1544+ const sendRequestDataSpy = jest . fn ( ) ;
15441545 const abortSpy = jest . fn ( ) ;
15451546
15461547 beforeEach ( ( ) => {
15471548 beforeTest ( ) ;
15481549 columnSendRequestStarted . mockClear ( ) ;
15491550 columnSendRequestResolved . mockClear ( ) ;
1550- sendRequestSpy . mockClear ( ) ;
1551+ sendRequestPromptSpy . mockClear ( ) ;
1552+ sendRequestDataSpy . mockClear ( ) ;
15511553 abortSpy . mockClear ( ) ;
15521554 } ) ;
15531555
@@ -1568,8 +1570,9 @@ describe('API Handlers', () => {
15681570 } ,
15691571 } ) ;
15701572 const columnAiIntegration = new AIIntegration ( {
1571- sendRequest ( { prompt } ) : RequestResult {
1572- sendRequestSpy ( prompt ) ;
1573+ sendRequest ( { prompt, data } ) : RequestResult {
1574+ sendRequestPromptSpy ( prompt ) ;
1575+ sendRequestDataSpy ( data ) ;
15731576 return aiIntegrationResult ( ) ;
15741577 } ,
15751578 } ) ;
@@ -1694,7 +1697,7 @@ describe('API Handlers', () => {
16941697 jest . advanceTimersByTime ( 10000 ) ;
16951698 expect ( columnSendRequestStarted ) . toHaveBeenCalledTimes ( 1 ) ;
16961699 expect ( columnSendRequestResolved ) . toHaveBeenCalledTimes ( 1 ) ;
1697- expect ( sendRequestSpy ) . toHaveBeenCalledWith ( expect . objectContaining ( {
1700+ expect ( sendRequestPromptSpy ) . toHaveBeenCalledWith ( expect . objectContaining ( {
16981701 user : expect . stringContaining ( 'Data: {"2":{"id":2,"name":"Name 2","value":20}}' ) ,
16991702 } ) ) ;
17001703
@@ -1734,13 +1737,18 @@ describe('API Handlers', () => {
17341737 jest . advanceTimersByTime ( 10000 ) ;
17351738 expect ( columnSendRequestStarted ) . toHaveBeenCalledTimes ( 1 ) ;
17361739 expect ( columnSendRequestResolved ) . toHaveBeenCalledTimes ( 1 ) ;
1737- expect ( sendRequestSpy ) . toHaveBeenCalledWith ( expect . objectContaining ( {
1738- user : expect . stringContaining ( 'Data: {"2":{"id":2,"name":"Name 2","value":20}}' ) ,
1740+ expect ( sendRequestDataSpy ) . toHaveBeenCalledWith ( expect . objectContaining ( {
1741+ additionalInfo : { customData : 'My custom data' } ,
17391742 } ) ) ;
17401743
17411744 await Promise . resolve ( ) ;
17421745 expect ( abortSpy ) . toHaveBeenCalledTimes ( 1 ) ;
17431746 } ) ;
1747+
1748+ // TODO: Implement after cache is done
1749+ // if('should take into account useCache property', async () => {
1750+
1751+ // });
17441752 } ) ;
17451753
17461754 // describe('onAIRequestCompleted', () => {
0 commit comments