@@ -38,18 +38,25 @@ suite('SqlCellStatusBarProvider', () => {
3838 assert . isUndefined ( result ) ;
3939 } ) ;
4040
41- test ( 'returns only variable status bar item for SQL cells without integration ID' , async ( ) => {
41+ test ( 'returns status bar items for SQL cells without integration ID' , async ( ) => {
4242 const cell = createMockCell ( 'sql' , { } ) ;
4343
4444 const result = await provider . provideCellStatusBarItems ( cell , cancellationToken ) ;
4545
4646 assert . isDefined ( result ) ;
4747 assert . isArray ( result ) ;
4848 const items = result as any [ ] ;
49- assert . strictEqual ( items . length , 1 ) ;
49+ assert . strictEqual ( items . length , 2 ) ;
50+
51+ // Check "No integration connected" status bar item
52+ const integrationItem = items [ 0 ] ;
53+ assert . strictEqual ( integrationItem . text , '$(database) No integration connected' ) ;
54+ assert . strictEqual ( integrationItem . alignment , 1 ) ;
55+ assert . isDefined ( integrationItem . command ) ;
56+ assert . strictEqual ( integrationItem . command . command , 'deepnote.switchSqlIntegration' ) ;
5057
5158 // Check variable status bar item
52- const variableItem = items [ 0 ] ;
59+ const variableItem = items [ 1 ] ;
5360 assert . strictEqual ( variableItem . text , 'Variable: df' ) ;
5461 assert . strictEqual ( variableItem . alignment , 1 ) ;
5562 assert . isDefined ( variableItem . command ) ;
0 commit comments