150 lines
3.5 KiB
TypeScript
150 lines
3.5 KiB
TypeScript
/**
|
|
* External dependencies
|
|
*/
|
|
import { recordEvent } from '@woocommerce/tracks';
|
|
import { __experimentalRequestJetpackToken as requestJetpackToken } from '@woocommerce/ai';
|
|
import apiFetch from '@wordpress/api-fetch';
|
|
/**
|
|
* Internal dependencies
|
|
*/
|
|
import { getCompletion } from '../services';
|
|
|
|
jest.mock( '@woocommerce/tracks', () => ( {
|
|
recordEvent: jest.fn(),
|
|
} ) );
|
|
|
|
jest.mock( '@woocommerce/ai', () => ( {
|
|
__experimentalRequestJetpackToken: jest.fn(),
|
|
} ) );
|
|
|
|
jest.mock( '@wordpress/api-fetch', () => jest.fn() );
|
|
|
|
describe( 'getCompletion', () => {
|
|
beforeEach( () => {
|
|
jest.clearAllMocks();
|
|
} );
|
|
|
|
it( 'should successfully get completion', async () => {
|
|
( requestJetpackToken as jest.Mock ).mockResolvedValue( {
|
|
token: 'fake_token',
|
|
} );
|
|
( apiFetch as unknown as jest.Mock ).mockResolvedValue( {
|
|
completion: JSON.stringify( { key: 'value' } ),
|
|
} );
|
|
const responseValidation = jest.fn( ( json ) => json );
|
|
|
|
const result = await getCompletion( {
|
|
queryId: 'query1',
|
|
prompt: 'test prompt',
|
|
responseValidation,
|
|
retryCount: 0,
|
|
version: '1',
|
|
} );
|
|
|
|
expect( result ).toEqual( { key: 'value' } );
|
|
expect( responseValidation ).toBeCalledWith( { key: 'value' } );
|
|
expect( recordEvent ).toBeCalledWith(
|
|
'customize_your_store_ai_completion_success',
|
|
{
|
|
query_id: 'query1',
|
|
retry_count: 0,
|
|
version: '1',
|
|
}
|
|
);
|
|
} );
|
|
|
|
it( 'should handle API fetch error', async () => {
|
|
( requestJetpackToken as jest.Mock ).mockResolvedValue( {
|
|
token: 'fake_token',
|
|
} );
|
|
( apiFetch as unknown as jest.Mock ).mockRejectedValue(
|
|
new Error( 'API error' )
|
|
);
|
|
|
|
await expect(
|
|
getCompletion( {
|
|
queryId: 'query1',
|
|
prompt: 'test prompt',
|
|
responseValidation: () => {},
|
|
retryCount: 0,
|
|
version: '1',
|
|
} )
|
|
).rejects.toThrow( 'API error' );
|
|
|
|
expect( recordEvent ).toBeCalledWith(
|
|
'customize_your_store_ai_completion_api_error',
|
|
{
|
|
query_id: 'query1',
|
|
retry_count: 0,
|
|
error_type: 'api_request_error',
|
|
version: '1',
|
|
}
|
|
);
|
|
} );
|
|
|
|
it( 'should handle JSON parse error', async () => {
|
|
( requestJetpackToken as jest.Mock ).mockResolvedValue( {
|
|
token: 'fake_token',
|
|
} );
|
|
( apiFetch as unknown as jest.Mock ).mockResolvedValue( {
|
|
completion: 'invalid json',
|
|
} );
|
|
|
|
await expect(
|
|
getCompletion( {
|
|
queryId: 'query1',
|
|
prompt: 'test prompt',
|
|
responseValidation: () => {},
|
|
retryCount: 0,
|
|
version: '1',
|
|
} )
|
|
).rejects.toThrow(
|
|
`Error validating Jetpack AI text completions response for query1`
|
|
);
|
|
|
|
expect( recordEvent ).toBeCalledWith(
|
|
'customize_your_store_ai_completion_response_error',
|
|
{
|
|
query_id: 'query1',
|
|
retry_count: 0,
|
|
error_type: 'json_parse_error',
|
|
response: 'invalid json',
|
|
version: '1',
|
|
}
|
|
);
|
|
} );
|
|
|
|
it( 'should handle validation error', async () => {
|
|
( requestJetpackToken as jest.Mock ).mockResolvedValue( {
|
|
token: 'fake_token',
|
|
} );
|
|
( apiFetch as unknown as jest.Mock ).mockResolvedValue( {
|
|
completion: JSON.stringify( { key: 'invalid value' } ),
|
|
} );
|
|
const responseValidation = jest.fn( () => {
|
|
throw new Error( 'Validation error' );
|
|
} );
|
|
|
|
await expect(
|
|
getCompletion( {
|
|
queryId: 'query1',
|
|
prompt: 'test prompt',
|
|
responseValidation,
|
|
retryCount: 0,
|
|
version: '1',
|
|
} )
|
|
).rejects.toThrow( 'Validation error' );
|
|
|
|
expect( recordEvent ).toBeCalledWith(
|
|
'customize_your_store_ai_completion_response_error',
|
|
{
|
|
query_id: 'query1',
|
|
retry_count: 0,
|
|
error_type: 'valid_json_invalid_values',
|
|
response: JSON.stringify( { key: 'invalid value' } ),
|
|
version: '1',
|
|
}
|
|
);
|
|
} );
|
|
} );
|