Add integration test to check flash cookie removal (close #139)

Author: dlecan

project-code/integration-tests/src/test/scala/AllTests.scala

@@ -285,6 +285,33 @@ abstract class AbstractPlay2WarTests extends FeatureSpec with GivenWhenThen with
         fail("Page not found")
       }
     }
+
+    scenario("Container set and remove flash cookie") {
+
+      // no flash cookie before test :)
+      var maybeFlashCookie = Option(webClient.getCookieManager.getCookie("PLAY_FLASH"))
+      maybeFlashCookie should be (None)
+
+      // still no flash cookie after this request
+      webClient.getPage(rootUrl + "/redirectLanding")
+      maybeFlashCookie = Option(webClient.getCookieManager.getCookie("PLAY_FLASH"))
+      maybeFlashCookie should be (None)
+
+      // Call page which sets Flash cookie, then redirect to redirectLanding page which removes cookie
+      val page = givenWhenGet("a page", "/flashing", "load a page which sets flash cookie")
+
+      Then("page body should contain 'Flash cookie: found'")
+      page.map { p =>
+        p.getWebResponse.getContentAsString should include("Flash cookie: found")
+      }.getOrElse {
+        fail("Page not found")
+      }
+
+      // no flash cookie after redirecting
+      maybeFlashCookie = Option(webClient.getCookieManager.getCookie("PLAY_FLASH"))
+      maybeFlashCookie should be (None)
+
+    }
   }
 
   /*

sample/common/app/controllers/Application.scala

@@ -46,8 +46,9 @@ object Application extends Controller {
     Ok(views.html.getCookies(mapCookies))
   }
 
-  def redirectLanding = Action {
-    Ok(views.html.redirectLanding())
+  def redirectLanding = Action { implicit request =>
+    val flashResult = flash.get("success").getOrElse("not found")
+    Ok(views.html.redirectLanding(flashResult))
   }
 
   def redirect = Action {
@@ -181,4 +182,11 @@ object Application extends Controller {
     Thread.sleep(java.util.concurrent.TimeUnit.SECONDS.toMillis(duration))
     Ok("")
   }
+
+  def flashing = Action {
+    Redirect(routes.Application.redirectLanding).flashing(
+      "success" -> "found"
+    )
+  }
+
 }

sample/common/app/views/redirectLanding.scala.html

@@ -1,5 +1,6 @@
-@()
+@(flashResult: String)
 
 @main("Redirect landing") {
     <p>redirect landing</p>
+    <p>Flash cookie: @flashResult</p>
 }

sample/common/conf/routes

@@ -34,5 +34,7 @@ GET     /clock                      controllers.Application.liveClock
 GET     /longRequest/:duration      controllers.JavaApplication.longRequest(duration: Long)
 GET     /slongRequest/:duration     controllers.Application.longRequest(duration: Long)
 
+GET     /flashing                   controllers.Application.flashing
+
 # Map static resources from the /public folder to the /assets URL path
 GET     /assets/*file               controllers.Assets.at(path="/public", file)