Improve decompiler test assertion errors

Gekkio · Nov 15, 2022 · d91d00b · d91d00b
1 parent 7f18369
commit d91d00b
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/test/kotlin/fi/gekkio/ghidraboy/decompiler/DecompilerTest.kt b/src/test/kotlin/fi/gekkio/ghidraboy/decompiler/DecompilerTest.kt
@@ -120,9 +120,10 @@ class DecompilerTest : IntegrationTest() {
             assertTrue(it.decompileCompleted()) { "Decompilation did not complete" }
         }.decompiledFunction.c
 
+    private fun formatCode(code: String) = code.lineSequence()
+        .map { it.trim() }
+        .filter { it.isNotEmpty() }
+        .joinToString(separator = "\n")
     private fun assertDecompiled(function: ghidra.program.model.listing.Function, @Language("C") code: String) =
-        assertEquals(
-            code.lineSequence().map { it.trim() }.filter { it.isNotEmpty() }.toList(),
-            decompile(function).lineSequence().map { it.trim() }.filter { it.isNotEmpty() }.toList()
-        )
+        assertEquals(formatCode(code), formatCode(decompile(function)))
 }