Add test to verify `compress` decorator is properly passing through irrelevant results

2 files changed, 6 insertions, 5 deletions

diff --git a/cask/src/cask/decorators/compress.scala b/cask/src/cask/decorators/compress.scala
index 61d76f6..17931fd 100644
--- a/cask/src/cask/decorators/compress.scala
+++ b/cask/src/cask/decorators/compress.scala
@@ -11,7 +11,8 @@ class compress extends cask.RawDecorator{
       .toSeq
       .flatMap(_.asScala)
       .flatMap(_.split(", "))
-    delegate(Map()).transform{ case v: cask.Response[_]  =>
+    val r = delegate(Map())
+    val finalResult = r.transform{ case v: cask.Response.Raw  =>
       val (newData, newHeaders) = if (acceptEncodings.exists(_.toLowerCase == "gzip")) {
         new Response.Data {
           def write(out: OutputStream): Unit = {
@@ -37,6 +38,6 @@ class compress extends cask.RawDecorator{
         v.cookies
       )
     }
-
+    finalResult
   }
 }
diff --git a/cask/src/cask/router/Result.scala b/cask/src/cask/router/Result.scala
index e38ee28..3c2f957 100644
--- a/cask/src/cask/router/Result.scala
+++ b/cask/src/cask/router/Result.scala
@@ -9,7 +9,7 @@ package cask.router
  */
 sealed trait Result[+T]{
   def map[V](f: T => V): Result[V]
-  def transform[V](f: PartialFunction[T, V]): Result[V]
+  def transform[V](f: PartialFunction[Any, V]): Result[V]
 }
 object Result{
 
@@ -19,7 +19,7 @@ object Result{
    */
   case class Success[T](value: T) extends Result[T]{
     def map[V](f: T => V) = Success(f(value))
-    def transform[V](f: PartialFunction[T, V]) = f.lift(value) match {
+    def transform[V](f: PartialFunction[Any, V]) = f.lift(value) match {
       case None => Success(value).asInstanceOf[Result[V]]
       case Some(res) => Success(res)
     }
@@ -30,7 +30,7 @@ object Result{
    */
   sealed trait Error extends Result[Nothing]{
     def map[V](f: Nothing => V) = this
-    def transform[V](f: PartialFunction[Nothing, V]) = this
+    def transform[V](f: PartialFunction[Any, V]) = this
   }