@@ -12,10 +12,7 @@ import {
1212import type { PreferencesAccess } from 'compass-preferences-model' ;
1313import { createSandboxFromDefaultPreferences } from 'compass-preferences-model' ;
1414import { 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' ;
1916import { CompassExperimentationProvider } from '@mongodb-js/compass-telemetry' ;
2017import 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