add truncate body length option to toCurl

http4k · Sep 23, 2018 · 4330f44 · 4330f44
1 parent e1709d7
commit 4330f44
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/http4k-core/src/main/kotlin/org/http4k/core/curl.kt b/http4k-core/src/main/kotlin/org/http4k/core/curl.kt
@@ -4,14 +4,14 @@ import org.http4k.appendIfNotBlank
 import org.http4k.appendIfNotEmpty
 import org.http4k.quoted
 
-fun Request.toCurl(): String =
-    StringBuilder("curl")
-        .append(" -X $method")
-        .appendIfNotEmpty(headers, " " + headers.joinToString(" ") { """-H ${(it.first + ":" + it.second).quoted()}""" })
-        .appendIfNotBlank(body.toString(), " --data ${body.toString().truncated().quoted()}")
-        .append(" \"$uri\"")
-        .toString()
+fun Request.toCurl(truncateBodyLength: Int = 256): String =
+        StringBuilder("curl")
+                .append(" -X $method")
+                .appendIfNotEmpty(headers, " " + headers.joinToString(" ") { """-H ${(it.first + ":" + it.second).quoted()}""" })
+                .appendIfNotBlank(body.toString(), " --data ${body.toString().truncated(truncateBodyLength).quoted()}")
+                .append(" \"$uri\"")
+                .toString()
 
-private fun String.truncated(): String = if (length > 256)
+private fun String.truncated(truncateBodyLength: Int): String = if (length > truncateBodyLength)
     substring(0..127) + "[truncated]" + substring(length - 128)
 else this
diff --git a/http4k-core/src/test/kotlin/org/http4k/CurlTest.kt b/http4k-core/src/test/kotlin/org/http4k/CurlTest.kt
@@ -63,7 +63,7 @@ class CurlTest {
     @Test
     fun `limits the entity if it's too large`() {
         val largeBody = (0..500).joinToString(" ")
-        val curl = org.http4k.core.Request(Method.GET, "http://httpbin.org").body(largeBody).toCurl()
+        val curl = Request(Method.GET, "http://httpbin.org").body(largeBody).toCurl(256)
         val data = "data \"([^\"]+)\"".toRegex().find(curl)?.groupValues?.get(1)!!
         assertThat(data.length, equalTo(256 + "[truncated]".length))
     }