@@ -4,24 +4,18 @@ import { TranslationService } from '../utils';
44import { handleCategorySelection } from './category' ;
55
66// Mock Choices.js
7- vi . mock ( 'choices.js' , ( ) => {
8- return {
9- default : vi . fn ( ) ,
10- } ;
11- } ) ;
7+ vi . mock ( 'choices.js' , ( ) => ( {
8+ default : vi . fn ( ) ,
9+ } ) ) ;
1210
1311// Mock TranslationService
14- vi . mock ( '../utils' , ( ) => {
15- const mockTranslationService = {
16- translations : new Map ( ) ,
17- loadTranslations : vi . fn ( ) . mockResolvedValue ( undefined ) ,
18- translate : vi . fn ( ) . mockImplementation ( ( key : string ) => `translated_${ key } ` ) ,
19- } ;
20-
21- return {
22- TranslationService : vi . fn ( ( ) => mockTranslationService ) ,
23- } ;
24- } ) ;
12+ vi . mock ( '../utils' , ( ) => ( {
13+ TranslationService : vi . fn ( function ( ) {
14+ this . translations = new Map ( ) ;
15+ this . loadTranslations = vi . fn ( ) . mockResolvedValue ( undefined ) ;
16+ this . translate = vi . fn ( ) . mockImplementation ( ( key : string ) => `translated_${ key } ` ) ;
17+ } ) ,
18+ } ) ) ;
2519
2620describe ( 'handleCategorySelection' , ( ) => {
2721 const mockChoices = vi . mocked ( Choices ) ;
@@ -72,7 +66,11 @@ describe('handleCategorySelection', () => {
7266 loadTranslations : vi . fn ( ) . mockResolvedValue ( undefined ) ,
7367 translate : vi . fn ( ) . mockReturnValue ( 'translated' ) ,
7468 } ;
75- mockTranslationService . mockReturnValue ( mockInstance ) ;
69+ mockTranslationService . mockImplementation ( function ( ) {
70+ this . translations = mockInstance . translations ;
71+ this . loadTranslations = mockInstance . loadTranslations ;
72+ this . translate = mockInstance . translate ;
73+ } ) ;
7674
7775 await handleCategorySelection ( ) ;
7876
0 commit comments