Skip to content

Commit f13950b

Browse files
committed
Update tests per comments
1 parent 5163ed7 commit f13950b

File tree

1 file changed

+13
-26
lines changed

1 file changed

+13
-26
lines changed

packages/compass-collection/src/components/collection-header-actions/collection-header-actions.spec.tsx

Lines changed: 13 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -12,10 +12,7 @@ import {
1212
import type { PreferencesAccess } from 'compass-preferences-model';
1313
import { createSandboxFromDefaultPreferences } from 'compass-preferences-model';
1414
import { PreferencesProvider } from 'compass-preferences-model/provider';
15-
import {
16-
TelemetryProvider,
17-
ExperimentTestName,
18-
} from '@mongodb-js/compass-telemetry/provider';
15+
import { ExperimentTestName } from '@mongodb-js/compass-telemetry/provider';
1916
import { CompassExperimentationProvider } from '@mongodb-js/compass-telemetry';
2017
import type { ConnectionInfo } from '@mongodb-js/compass-connections/provider';
2118

@@ -54,30 +51,21 @@ describe('CollectionHeaderActions [Component]', function () {
5451
};
5552

5653
const ui = (
57-
<TelemetryProvider options={{ sendTrack: sinon.stub() }}>
58-
<CompassExperimentationProvider
59-
useAssignment={mockUseAssignment}
60-
assignExperiment={sinon.stub()}
54+
<CompassExperimentationProvider
55+
useAssignment={mockUseAssignment}
56+
assignExperiment={sinon.stub()}
57+
>
58+
<WorkspacesServiceProvider
59+
value={workspaceService as WorkspacesService}
6160
>
62-
<WorkspacesServiceProvider
63-
value={workspaceService as WorkspacesService}
64-
>
65-
<PreferencesProvider value={preferences}>
66-
<CollectionHeaderActions {...defaultProps} />
67-
</PreferencesProvider>
68-
</WorkspacesServiceProvider>
69-
</CompassExperimentationProvider>
70-
</TelemetryProvider>
61+
<PreferencesProvider value={preferences}>
62+
<CollectionHeaderActions {...defaultProps} />
63+
</PreferencesProvider>
64+
</WorkspacesServiceProvider>
65+
</CompassExperimentationProvider>
7166
);
7267

73-
if (connectionInfo) {
74-
// For tests that need Atlas metadata (Mock Data Generator button visibility, etc.)
75-
return await renderWithActiveConnection(ui, connectionInfo);
76-
} else {
77-
// For tests that only need basic component rendering (readonly checks, view buttons, etc.)
78-
const { render } = await import('@mongodb-js/testing-library-compass');
79-
return render(ui);
80-
}
68+
return await renderWithActiveConnection(ui, connectionInfo);
8169
};
8270

8371
context('when the collection is not readonly', function () {
@@ -340,7 +328,6 @@ describe('CollectionHeaderActions [Component]', function () {
340328
expect(
341329
screen.getByTestId('collection-header-generate-mock-data-button')
342330
).to.exist;
343-
expect(screen.getByText('Generate Mock Data')).to.exist;
344331
});
345332

346333
it('should call useAssignment with correct parameters', async function () {

0 commit comments

Comments
 (0)