@@ -24,6 +24,19 @@ describe('showEdits', function () {
24
24
let sessionStub : any
25
25
let itemStub : InlineCompletionItemWithReferences
26
26
27
+ // Helper function to create mock SVG result
28
+ function createMockSvgResult ( overrides : Partial < any > = { } ) {
29
+ return {
30
+ svgImage : vscode . Uri . file ( '/path/to/generated.svg' ) ,
31
+ startLine : 5 ,
32
+ newCode : 'console.log("Hello World");' ,
33
+ origionalCodeHighlightRange : [ { line : 5 , start : 0 , end : 10 } ] ,
34
+ addedCharacterCount : 25 ,
35
+ deletedCharacterCount : 0 ,
36
+ ...overrides ,
37
+ }
38
+ }
39
+
27
40
beforeEach ( function ( ) {
28
41
sandbox = sinon . createSandbox ( )
29
42
@@ -54,7 +67,7 @@ describe('showEdits', function () {
54
67
id : sharedModuleId ,
55
68
filename : sharedModuleId ,
56
69
loaded : true ,
57
- parent : null ,
70
+ parent : undefined ,
58
71
children : [ ] ,
59
72
exports : mockSharedModule ,
60
73
paths : [ ] ,
@@ -133,14 +146,7 @@ describe('showEdits', function () {
133
146
134
147
it ( 'should successfully generate and display SVG when all parameters are valid' , async function ( ) {
135
148
// Setup successful SVG generation
136
- const mockSvgResult = {
137
- svgImage : vscode . Uri . file ( '/path/to/generated.svg' ) ,
138
- startLine : 5 ,
139
- newCode : 'console.log("Hello World");' ,
140
- origionalCodeHighlightRange : [ { line : 5 , start : 0 , end : 10 } ] ,
141
- addedCharacterCount : 25 ,
142
- deletedCharacterCount : 0 ,
143
- }
149
+ const mockSvgResult = createMockSvgResult ( )
144
150
svgGenerationServiceStub . generateDiffSvg . resolves ( mockSvgResult )
145
151
146
152
await showEdits ( itemStub , editorStub as unknown as vscode . TextEditor , sessionStub , languageClientStub )
@@ -175,14 +181,7 @@ describe('showEdits', function () {
175
181
176
182
it ( 'should log error when SVG generation returns empty result' , async function ( ) {
177
183
// Setup SVG generation to return undefined svgImage
178
- const mockSvgResult = {
179
- svgImage : undefined as any ,
180
- startLine : 5 ,
181
- newCode : 'console.log("Hello World");' ,
182
- origionalCodeHighlightRange : [ { line : 5 , start : 0 , end : 10 } ] ,
183
- addedCharacterCount : 25 ,
184
- deletedCharacterCount : 0 ,
185
- }
184
+ const mockSvgResult = createMockSvgResult ( { svgImage : undefined as any } )
186
185
svgGenerationServiceStub . generateDiffSvg . resolves ( mockSvgResult )
187
186
188
187
await showEdits ( itemStub , editorStub as unknown as vscode . TextEditor , sessionStub , languageClientStub )
@@ -219,14 +218,7 @@ describe('showEdits', function () {
219
218
220
219
it ( 'should catch and log error when displaySvgDecoration throws exception' , async function ( ) {
221
220
// Setup successful SVG generation
222
- const mockSvgResult = {
223
- svgImage : vscode . Uri . file ( '/path/to/generated.svg' ) ,
224
- startLine : 5 ,
225
- newCode : 'console.log("Hello World");' ,
226
- origionalCodeHighlightRange : [ { line : 5 , start : 0 , end : 10 } ] ,
227
- addedCharacterCount : 25 ,
228
- deletedCharacterCount : 0 ,
229
- }
221
+ const mockSvgResult = createMockSvgResult ( )
230
222
svgGenerationServiceStub . generateDiffSvg . resolves ( mockSvgResult )
231
223
232
224
// Setup displaySvgDecoration to throw an error
@@ -262,14 +254,7 @@ describe('showEdits', function () {
262
254
} as any
263
255
264
256
// Setup successful SVG generation
265
- const mockSvgResult = {
266
- svgImage : vscode . Uri . file ( '/path/to/generated.svg' ) ,
267
- startLine : 5 ,
268
- newCode : 'console.log("Hello World");' ,
269
- origionalCodeHighlightRange : [ { line : 5 , start : 0 , end : 10 } ] ,
270
- addedCharacterCount : 25 ,
271
- deletedCharacterCount : 0 ,
272
- }
257
+ const mockSvgResult = createMockSvgResult ( )
273
258
svgGenerationServiceStub . generateDiffSvg . resolves ( mockSvgResult )
274
259
275
260
await showEdits (
0 commit comments