@@ -59,7 +59,10 @@ function calculateTotal(items: number[]): number {
5959export { calculateTotal };`
6060
6161 const result = strategy . applyDiff ( originalContent , diffContent )
62- expect ( result ) . toBe ( expected )
62+ expect ( result . success ) . toBe ( true )
63+ if ( result . success ) {
64+ expect ( result . content ) . toBe ( expected )
65+ }
6366 } )
6467
6568 it ( 'should successfully apply a diff adding a new method' , ( ) => {
@@ -93,7 +96,10 @@ export { calculateTotal };`
9396}`
9497
9598 const result = strategy . applyDiff ( originalContent , diffContent )
96- expect ( result ) . toBe ( expected )
99+ expect ( result . success ) . toBe ( true )
100+ if ( result . success ) {
101+ expect ( result . content ) . toBe ( expected )
102+ }
97103 } )
98104
99105 it ( 'should successfully apply a diff modifying imports' , ( ) => {
@@ -128,7 +134,10 @@ function App() {
128134}`
129135
130136 const result = strategy . applyDiff ( originalContent , diffContent )
131- expect ( result ) . toBe ( expected )
137+ expect ( result . success ) . toBe ( true )
138+ if ( result . success ) {
139+ expect ( result . content ) . toBe ( expected )
140+ }
132141 } )
133142
134143 it ( 'should successfully apply a diff with multiple hunks' , ( ) => {
@@ -190,7 +199,10 @@ async function processFile(path: string) {
190199export { processFile };`
191200
192201 const result = strategy . applyDiff ( originalContent , diffContent )
193- expect ( result ) . toBe ( expected )
202+ expect ( result . success ) . toBe ( true )
203+ if ( result . success ) {
204+ expect ( result . content ) . toBe ( expected )
205+ }
194206 } )
195207
196208 it ( 'should handle empty original content' , ( ) => {
@@ -207,7 +219,10 @@ export { processFile };`
207219}\n`
208220
209221 const result = strategy . applyDiff ( originalContent , diffContent )
210- expect ( result ) . toBe ( expected )
222+ expect ( result . success ) . toBe ( true )
223+ if ( result . success ) {
224+ expect ( result . content ) . toBe ( expected )
225+ }
211226 } )
212227 } )
213228} )
0 commit comments