Как проверить маршрут akka-http для обработки исключения?

У меня есть простое приложение akka-http для загрузки файлов кусками на сервер. У него есть два маршрута: первый / открывает HTML-форму для поиска файла, а второй маршрут связывает кнопку загрузки с логикой, которая разделяет файл на куски и выгружает его. Затем я создал модульный тест (он же Spec), чтобы проверить эти два маршрута. Первый очень простой, он просто определяет, что я могу открыть веб-страницу. а вот второе я бы хотел протестировать:

если в форме нет файла, я должен указать исключение в спецификации
Если в форме есть файл, я проверяю логику и проверяю, действительно ли он загружает файл.

Итак, это мое приложение akka-http:

import akka.Done
import akka.actor.ActorSystem
import akka.http.scaladsl.Http
import akka.http.scaladsl.model.{ContentTypes, HttpEntity, Multipart}
import akka.http.scaladsl.server.Directives._
import akka.stream.ThrottleMode
import akka.stream.scaladsl.{FileIO, Sink, Source}
import akka.util.ByteString

import java.io.File
import scala.concurrent.Future
import scala.concurrent.duration._
import scala.util.{Failure, Success}

object UploadingFiles {
  implicit val system = ActorSystem("UploadingFiles")
  val NO_OF_MESSAGES = 1
  val filesRoutes = {
    (pathEndOrSingleSlash & get) {
      complete(
        HttpEntity(
          ContentTypes.`text/html(UTF-8)`,
          """
            |<html>
            |  <body>
            |    <form action="http://localhost:8080/upload" method="post" enctype="multipart/form-data">
            |      <input type="file" name="myFile" multiple>
            |      <button type="submit">Upload</button>
            |    </form>
            |  </body>
            |</html>
          """.stripMargin
        )
      )
    } ~ (path("upload") & post & extractLog) { log =>
      // handling uploading files using multipart/form-data
      entity(as[Multipart.FormData]) { formData =>
        // handle file payload
        val partsSource: Source[Multipart.FormData.BodyPart, Any] = formData.parts
        val filePartsSink: Sink[Multipart.FormData.BodyPart, Future[Done]] =
          Sink.foreach[Multipart.FormData.BodyPart] { bodyPart =>
            if (bodyPart.name == "myFile") {
              // create a file
              val filename = "download/" + bodyPart.filename.getOrElse("tempFile_" + System.currentTimeMillis())
              val file = new File(filename)

              log.info(s"writing to file: $filename")
              val fileContentsSource: Source[ByteString, _] = bodyPart.entity.dataBytes
              val fileContentsSink: Sink[ByteString, _] = FileIO.toPath(file.toPath)

              val publishRate = NO_OF_MESSAGES / 1
              // writing the data to the file using akka-stream graph
              fileContentsSource
                .throttle(publishRate, 2 seconds, publishRate, ThrottleMode.shaping)
                .runWith(fileContentsSink)
            }
          }
        val writeOperationFuture = partsSource.runWith(filePartsSink)
        onComplete(writeOperationFuture) {
          case Success(value) => complete("file uploaded =)")
          case Failure(exception) => complete(s"file failed to upload: $exception")
        }
      }
    }
  }

  def main(args: Array[String]): Unit = {
    println("access the browser at: localhost:8080")
    Http().newServerAt("localhost", 8080).bindFlow(filesRoutes)
  }
}

Я хотел бы поймать строку с исключением s"file failed to upload: $exception" в приложении akka-http в моей спецификации:

import akka.http.scaladsl.model.StatusCodes
import akka.http.scaladsl.testkit.{RouteTestTimeout, ScalatestRouteTest}
import org.github.felipegutierrez.explore.akka.classic.http.server.highlevel.UploadingFiles.filesRoutes
import org.scalatest.matchers.should.Matchers
import org.scalatest.wordspec.AnyWordSpec

import scala.concurrent.duration._

class UploadingFilesSpec extends AnyWordSpec
  with Matchers
  with ScalatestRouteTest {

  implicit val timeout: RouteTestTimeout = RouteTestTimeout(2 seconds)

  "A basic GET request to open the html form" should {
    "return OK [200]" in {
      Get("/") ~> filesRoutes ~> check {
        status shouldBe StatusCodes.OK
      }
    }
  }
  "A POST request upload without a file" should {
    "return NOT OK" in {
      Post("/upload") ~> filesRoutes ~> check {
        handled should ===(false) // THIS WORKS

        // THIS DOES NOT WORK
        rejection should ===(UnsupportedRequestContentTypeRejection(Set(ContentTypes.`text/plain(UTF-8)`)))
        println(rejections)
        rejections.foreach { e: Rejection =>
          println(e.toString)
          println(e.asInstanceOf[UnsupportedRequestContentTypeRejection].contentType)
        }
        // THIS DOES NOT WORK
        rejection should ===(
          UnsupportedRequestContentTypeRejection(
            Set(ContentTypes.`text/plain(UTF-8)`),
            Some(ContentTypes.`text/plain(UTF-8)`)
          )
        )
      }
    }
  }
}

Но похоже, что тестовый комплект тестирует все содержимое исключения, и я хочу сделать его универсальным для любого вида исключения. Вот ошибка:

akka.http.scaladsl.server.UnsupportedRequestContentTypeRejection@bdc8014 did not equal akka.http.scaladsl.server.UnsupportedRequestContentTypeRejection@1f443fae
ScalaTestFailureLocation: org.github.felipegutierrez.explore.akka.classic.http.server.highlevel.UploadingFilesSpec at (UploadingFilesSpec.scala:30)
Expected :akka.http.scaladsl.server.UnsupportedRequestContentTypeRejection@1f443fae
Actual   :akka.http.scaladsl.server.UnsupportedRequestContentTypeRejection@bdc8014

Felipe 03.02.2021 источник

Ответы (1)

arrow_upward
1
arrow_downward

Обратите внимание, что _ 1_, как вы ожидаете в своем маршруте, имеет следующий тип носителя:

def mediaType = MediaTypes.`multipart/form-data`

Способ создать ContentTypeRange из него:

ContentTypeRange(MediaRange.apply(MediaTypes.`multipart/form-data`))

и у него нет типа содержимого.

Следовательно, должно работать следующее:

"A POST request upload without a file" should {
  "return NOT OK" in {
    Post("/upload") ~> filesRoutes ~> check {
      handled should ===(false) // THIS WORKS

      rejection should ===(UnsupportedRequestContentTypeRejection(
        Set(ContentTypeRange(MediaRange.apply(MediaTypes.`multipart/form-data`))), 
        Some(ContentTypes.NoContentType))
      )
    }
  }
}

Tomer Shetah 03.02.2021

Как проверить маршрут akka-http для обработки исключения?

Ответы (1)

Похожие вопросы