ScalatestRouteTest does not support session test

[epiphyllum]

in my example:
there are three route: login, upload, download
to test it, the client must login first to upload
then with the result from previous upload, can only the client initiate the download test
it seems that ScalatestRouteTest does not support this kinds of tests

package com.yimei.zflow.util.asset.routes

import akka.actor.ActorSystem
import akka.http.scaladsl.marshallers.sprayjson.SprayJsonSupport
import akka.http.scaladsl.model.{ContentTypes, HttpEntity, Multipart}
import akka.http.scaladsl.server.Route
import akka.http.scaladsl.testkit.ScalatestRouteTest
import com.typesafe.config.ConfigFactory
import com.yimei.zflow.util.FlywayDB
import com.yimei.zflow.util.asset.routes.Models._
import com.yimei.zflow.util.organ.OrganSession
import org.scalatest.{Inside, Matchers, WordSpec}

object AssetRouteUT extends {
  implicit val coreSystem: ActorSystem = ActorSystem("TestSystem", ConfigFactory.parseString(
    """
      |database {
      |  jdbcUrl = "jdbc:mysql://127.0.0.1/cyflow?useUnicode=true&characterEncoding=utf8"
      |  username = "mysql"
      |  password = "mysql"
      |}
      |file.root = "/tmp"
      |akka.http.session.server-secret = "1234567891234567891234567891234567891234567890000012345678901234"
      |
    """.stripMargin))
} with AssetRoute {
  val fileField = "file"

  def prepare() = {
    val config = coreSystem.settings.config
    val jdbcUrl = config.getString("database.jdbcUrl")
    val username = config.getString("database.username")
    val password = config.getString("database.password")
    val flyway = new FlywayDB(jdbcUrl, username, password);
    flyway.drop()
    flyway.migrate()
  }

  import akka.http.scaladsl.server.Directives._
  def loginRoute: Route = path("login") {
    organSetSession(OrganSession("hary", "uid", "party", "instanceId", "company")) { ctx =>
      ctx.complete("ok")
    }
  }
}

/**
  * Created by hary on 17/1/10.
  */
class AssetRouteTest extends WordSpec with Matchers with Inside with ScalatestRouteTest with SprayJsonSupport {

  import AssetRouteUT.{assetRoute, loginRoute, prepare}

  override protected def beforeAll(): Unit = {
    super.beforeAll()
    prepare

    // must be tested with login session

  }

  val xml = "<int>42</int>"  // 上传文件
  val simpleMultipartUpload =
    Multipart.FormData(Multipart.FormData.BodyPart.Strict(
      "file",
      HttpEntity(ContentTypes.`text/xml(UTF-8)`, xml),
      Map("filename" → "age.xml")))

  var assetId: String = null

  "AssetRouteTest" should {

    "login is ok" in {
      Post("/login") ~> loginRoute ~> check {
        responseAs[String] shouldBe "ok"
      }

    }

    "uploadFile is ok" in {
      Post("/upload") ~> assetRoute ~> check {
        inside(responseAs[UploadResult]) {
          case UploadResult(id) =>
            assetId = id
            1 shouldBe 1
        }
      }
    }

    "downloadFile is in" in {
      Get("/download/" + assetId) ~> assetRoute ~> check {
        responseAs[String] shouldEqual xml
      }
    }
  }

}

[epiphyllum]

it seems that each test request is indepenent

[epiphyllum]

maybe, I can add the cookie header in subsequent test request to make it work

[epiphyllum]

it seems that each test request is indepentent in ScalatestRouteTest, when I add Set-Cookie,
authorization is still failed。

package com.yimei.zflow.util.asset.routes

import akka.actor.ActorSystem
import akka.http.scaladsl.marshallers.sprayjson.SprayJsonSupport
import akka.http.scaladsl.model.{ContentTypes, HttpEntity, HttpHeader, Multipart}
import akka.http.scaladsl.server.Route
import akka.http.scaladsl.testkit.ScalatestRouteTest
import com.typesafe.config.ConfigFactory
import com.yimei.zflow.util.FlywayDB
import com.yimei.zflow.util.asset.routes.Models._
import com.yimei.zflow.util.organ.OrganSession
import org.scalatest.{Inside, Matchers, WordSpec}

object AssetRouteUT extends {
  implicit val coreSystem: ActorSystem = ActorSystem("TestSystem", ConfigFactory.parseString(
    """
      |database {
      |  jdbcUrl = "jdbc:mysql://127.0.0.1/cyflow?useUnicode=true&characterEncoding=utf8"
      |  username = "mysql"
      |  password = "mysql"
      |}
      |file.root = "/tmp"
      |akka.http.session.server-secret = "1234567891234567891234567891234567891234567890000012345678901234"
      |
    """.stripMargin))
} with AssetRoute {
  val fileField = "file"

  def prepare() = {
    val config = coreSystem.settings.config
    val jdbcUrl = config.getString("database.jdbcUrl")
    val username = config.getString("database.username")
    val password = config.getString("database.password")
    val flyway = new FlywayDB(jdbcUrl, username, password);
    flyway.drop()
    flyway.migrate()
  }

  import akka.http.scaladsl.server.Directives._
  def loginRoute: Route = path("login") {
    organSetSession(OrganSession("hary", "uid", "party", "instanceId", "company")) { ctx =>
      ctx.complete("ok")
    }
  }
}

/**
  * Created by hary on 17/1/10.
  */
class AssetRouteTest extends WordSpec with Matchers with Inside with ScalatestRouteTest with SprayJsonSupport {

  import AssetRouteUT.{assetRoute, loginRoute, prepare}

  override protected def beforeAll(): Unit = {
    super.beforeAll()
    prepare

    // must be tested with login session

  }

  val xml = "<int>42</int>"  // 上传文件
  val simpleMultipartUpload =
    Multipart.FormData(Multipart.FormData.BodyPart.Strict(
      "file",
      HttpEntity(ContentTypes.`text/xml(UTF-8)`, xml),
      Map("filename" → "age.xml")))

  var assetId: String = null

  var cookie: HttpHeader = null

  "AssetRouteTest" should {

    "login is ok" in {
      Post("/login") ~> loginRoute ~> check {
        cookie = header("Set-Cookie").get
        responseAs[String] shouldBe "ok"
        println("cookie is " + cookie.toString)
      }

    }

    "uploadFile is ok" in {
      Post("/upload") ~> addHeader(cookie) ~> assetRoute ~> check {
        inside(responseAs[UploadResult]) {
          case UploadResult(id) =>
            assetId = id
            1 shouldBe 1
        }
        cookie = header("Set-Cookie").get
      }
    }

    "downloadFile is in" in {
      Get("/download/" + assetId) ~> addHeader(cookie) ~> assetRoute ~> check {
        responseAs[String] shouldEqual xml
      }
    }
  }

}

[adamw]

Yes, each test is independent, only the headers that you specify explicitly are sent. Previous responses have no influence on subsequent requests.

Maybe this will help: https://github.com/softwaremill/akka-http-session/blob/master/core/src/test/scala/com/softwaremill/session/OneOffTest.scala#L94

These are the tests of the set/get session directives. In there the /set endpoints sets a session (logs a user in), and /get uses a previously created session.

[Hajto]

@adamw Can I pre-generate cookie/JWT before sending checking ANY route (like call JWT generate(resource)) ?

[adamw]

@Hajto yes, you can do it through SessionManager.clientSessionManager.createCookie(data: T). You need to have an implicit session manager somewhere in order for the routes to work, so you can just use the same one.