io-to-data.7 : extract generate

Duncan McGregor · dmcg · commit 2f7a7056046a · 2021-06-23T22:24:14.000+01:00
diff --git a/src/main/java/travelator/marketing/HighValueCustomersReport.kt b/src/main/java/travelator/marketing/HighValueCustomersReport.kt
@@ -5,13 +5,18 @@ import java.io.Writer
 
 @Throws(IOException::class)
 fun generate(writer: Writer, lines: List<String>) {
+    val resultLines = generate(lines)
+    writer.append(resultLines.joinToString("\n"))
+}
+
+fun generate(lines: List<String>): List<String> {
     val valuableCustomers = lines
         .toValuableCustomers()
         .sortedBy(CustomerData::score)
     val resultLines = listOf("ID\tName\tSpend") +
         valuableCustomers.map(CustomerData::outputLine) +
         valuableCustomers.summarised()
-    writer.append(resultLines.joinToString("\n"))
+    return resultLines
 }
 
 private fun List<String>.toValuableCustomers() = withoutHeader()
