Merge pull request #135 from ing-bank/bug/content-length-size

Bug - fakeEntity content length

src/main/scala/com/ing/wbaa/rokku/proxy/handler/parsers/CacheHelpers.scala

@@ -39,11 +39,18 @@ object CacheHelpers {
   // https://github.com/akka/akka-http/issues/377
   // return a pseudo Default entity that contains the content-length and an Empty data stream
   def generateFakeEntity(contentLength: Int): HttpEntity.Default = {
-    HttpEntity.Default(
-      ContentType.WithMissingCharset(MediaTypes.`text/plain`),
-      contentLength,
-      Source(ByteString() :: Nil)
-    )
+    val entityWithSize: Int => HttpEntity.Default = cl =>
+      HttpEntity.Default(
+        ContentType.WithMissingCharset(MediaTypes.`text/plain`),
+        cl,
+        Source(ByteString() :: Nil)
+      )
+
+    if (contentLength == 0) {
+      entityWithSize(1)
+    } else {
+      entityWithSize(contentLength)
+    }
   }
 
 }

src/test/scala/com/ing/wbaa/rokku/proxy/cache/CacheHelpersSpec.scala

@@ -0,0 +1,15 @@
+package com.ing.wbaa.rokku.proxy.cache
+
+import com.ing.wbaa.rokku.proxy.handler.parsers.CacheHelpers._
+import org.scalatest.diagrams.Diagrams
+import org.scalatest.wordspec.AnyWordSpec
+
+class CacheHelpersSpec extends AnyWordSpec with Diagrams {
+
+  "Hazelcast Cache helpers" should {
+    "returns non empty Entity if incorrect CL" in {
+      assert(generateFakeEntity(0).contentLength == 1)
+    }
+  }
+
+}