Fix chunk encoding reading

apollographql · Sep 16, 2024 · b7e2419 · b7e2419
1 parent bdd7f2d
commit b7e2419
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/apollo-mockserver/src/commonMain/kotlin/com/apollographql/mockserver/http.kt b/apollo-mockserver/src/commonMain/kotlin/com/apollographql/mockserver/http.kt
@@ -49,13 +49,16 @@ internal suspend fun readRequest(reader: Reader): MockRequestBase {
   }
 
   suspend fun readBytes(size: Long): Buffer {
+    var remaining = size
     val buffer2 = Buffer()
     while (buffer2.size < size) {
       if (reader.buffer.size == 0L) {
         reader.fillBuffer()
       }
 
-      buffer2.write(reader.buffer, minOf(size, reader.buffer.size))
+      val toRead = minOf(remaining, reader.buffer.size)
+      buffer2.write(reader.buffer, toRead)
+      remaining -= toRead
     }
 
     return buffer2