@@ -20,12 +20,7 @@ import { AmazonQInlineCompletionItemProvider, InlineCompletionManager } from '..
2020import { RecommendationService } from '../../../../../src/app/inline/recommendationService'
2121import { SessionManager } from '../../../../../src/app/inline/sessionManager'
2222import { createMockDocument , createMockTextEditor , getTestWindow , installFakeClock } from 'aws-core-vscode/test'
23- import {
24- noInlineSuggestionsMsg ,
25- ReferenceHoverProvider ,
26- ReferenceInlineProvider ,
27- ReferenceLogViewProvider ,
28- } from 'aws-core-vscode/codewhisperer'
23+ import { noInlineSuggestionsMsg , ReferenceHoverProvider , ReferenceLogViewProvider } from 'aws-core-vscode/codewhisperer'
2924import { InlineGeneratingMessage } from '../../../../../src/app/inline/inlineGeneratingMessage'
3025import { LineTracker } from '../../../../../src/app/inline/stateTracker/lineTracker'
3126import { InlineTutorialAnnotation } from '../../../../../src/app/inline/tutorials/inlineTutorialAnnotation'
@@ -238,15 +233,13 @@ describe('InlineCompletionManager', () => {
238233 let provider : AmazonQInlineCompletionItemProvider
239234 let getAllRecommendationsStub : sinon . SinonStub
240235 let recommendationService : RecommendationService
241- let setInlineReferenceStub : sinon . SinonStub
242236 let inlineTutorialAnnotation : InlineTutorialAnnotation
243237
244238 beforeEach ( ( ) => {
245239 const lineTracker = new LineTracker ( )
246240 const activeStateController = new InlineGeneratingMessage ( lineTracker )
247241 inlineTutorialAnnotation = new InlineTutorialAnnotation ( lineTracker , mockSessionManager )
248242 recommendationService = new RecommendationService ( mockSessionManager , activeStateController )
249- setInlineReferenceStub = sandbox . stub ( ReferenceInlineProvider . instance , 'setInlineReference' )
250243
251244 mockSessionManager = {
252245 getActiveSession : getActiveSessionStub ,
@@ -288,14 +281,6 @@ describe('InlineCompletionManager', () => {
288281 inlineTutorialAnnotation
289282 )
290283 await provider . provideInlineCompletionItems ( mockDocument , mockPosition , mockContext , mockToken )
291- assert ( setInlineReferenceStub . calledOnce )
292- assert (
293- setInlineReferenceStub . calledWithExactly (
294- mockPosition . line ,
295- mockSuggestions [ 0 ] . insertText ,
296- fakeReferences
297- )
298- )
299284 } ) ,
300285 it ( 'should add a range to the completion item when missing' , async function ( ) {
301286 provider = new AmazonQInlineCompletionItemProvider (
0 commit comments