Merge pull request leanovate#39 from leanovate/issue_38

add test for issue 38

src/test/scala/mockws/StreamingTest.scala

@@ -1,14 +1,16 @@
 package mockws
 
+import akka.NotUsed
 import akka.stream.scaladsl.Source
 import akka.util.ByteString
 import mockws.MockWSHelpers._
 import org.scalatest.prop.PropertyChecks
 import org.scalatest.{FunSuite, Matchers}
 import play.api.http.HttpEntity
 import play.api.libs.ws.WSClient
-import play.api.mvc.MultipartFormData.DataPart
+import play.api.mvc.MultipartFormData.{DataPart, FilePart, Part}
 import play.api.mvc.{ResponseHeader, Result}
+import play.api.mvc.Results._
 import play.api.test.FakeRequest
 import play.api.test.Helpers._
 
@@ -46,6 +48,28 @@ class StreamingTest extends FunSuite with Matchers with PropertyChecks {
     ws.close()
   }
 
+  test("mock WS supports streaming of MultipartFormData") {
+    val ws = MockWS {
+      case (PUT, "/") =>
+        Action { request ⇒
+          request.body.asMultipartFormData match {
+            case None ⇒ InternalServerError("error")
+            case Some(data) ⇒ Ok(data.dataParts.mkString(", "))
+          }
+        }
+    }
+
+    val fileData: Source[Part[Source[ByteString, _]], NotUsed] = Source(
+      FilePart("file", "", Some(BINARY), Source.single(ByteString("test"))) ::
+        DataPart("key 1", "data 1") ::
+        DataPart("key 2", "data 2") ::
+        Nil)
+
+    val response = await(ws.url("/").put(fileData))
+    response.body shouldEqual "key 2 -> Vector(data 2), key 1 -> Vector(data 1)"
+    ws.close()
+  }
+
   test("mock WS supports method in stream") {
     def testedController(ws: WSClient) = Action.async {
       ws.url("/").withMethod("POST").stream().map { resp =>