Skip to content

Commit 4a5dc46

Browse files
committed
adapt unit test
1 parent 3e43c3f commit 4a5dc46

File tree

1 file changed

+20
-2
lines changed

1 file changed

+20
-2
lines changed

packages/compass-collection/src/stores/collection-tab.spec.ts

Lines changed: 20 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)