@@ -330,6 +330,11 @@ describe('Collection Tab Content store', function () {
330330 log : { error : sandbox . spy ( ) } ,
331331 debug : sandbox . spy ( ) ,
332332 } ;
333+ const preferences = {
334+ getConfigurableUserPreferences : sandbox . stub ( ) . resolves ( {
335+ enableGenAISampleDocumentPassing : true ,
336+ } ) ,
337+ } ;
333338
334339 const mockDataSchemaResponse : MockDataSchemaResponse = {
335340 content : {
@@ -361,7 +366,11 @@ describe('Collection Tab Content store', function () {
361366 const thunk = collectionTabModule . generateFakerMappings (
362367 mockAtlasConnectionInfo . current
363368 ) ;
364- await thunk ( dispatch , getState , { logger, atlasAiService } as any ) ;
369+ await thunk ( dispatch , getState , {
370+ logger,
371+ atlasAiService,
372+ preferences,
373+ } as any ) ;
365374
366375 // Assert
367376 expect ( dispatch ) . to . have . been . calledTwice ;
@@ -396,6 +405,11 @@ describe('Collection Tab Content store', function () {
396405 log : { error : sandbox . spy ( ) } ,
397406 debug : sandbox . spy ( ) ,
398407 } ;
408+ const preferences = {
409+ getConfigurableUserPreferences : sandbox . stub ( ) . resolves ( {
410+ enableGenAISampleDocumentPassing : true ,
411+ } ) ,
412+ } ;
399413
400414 const atlasAiService = {
401415 getMockDataSchema : sandbox . stub ( ) . resolves ( { } ) ,
@@ -405,7 +419,11 @@ describe('Collection Tab Content store', function () {
405419 const thunk = collectionTabModule . generateFakerMappings (
406420 mockAtlasConnectionInfo . current
407421 ) ;
408- await thunk ( dispatch , getState , { logger, atlasAiService } as any ) ;
422+ await thunk ( dispatch , getState , {
423+ logger,
424+ atlasAiService,
425+ preferences,
426+ } as any ) ;
409427
410428 // Assert
411429 expect ( dispatch ) . to . have . been . calledTwice ;
0 commit comments