Refactor decorators into a more traditional delegation model, and use that to implement endpoint-scoped transactions using Quill

1 files changed, 6 insertions, 2 deletions

diff --git a/cask/test/src/test/cask/FailureTests.scala b/cask/test/src/test/cask/FailureTests.scala
index 3ed4249..de3b438 100644
--- a/cask/test/src/test/cask/FailureTests.scala
+++ b/cask/test/src/test/cask/FailureTests.scala
@@ -1,11 +1,15 @@
 package test.cask
 
-import cask.model.ParamContext
+import cask.internal.Router
+import cask.model.{ParamContext, Response}
 import utest._
 
 object FailureTests extends TestSuite {
   class myDecorator extends cask.Decorator {
-    def getRawParams(ctx: ParamContext) = Right(cask.Decor("extra" -> 31337))
+    def wrapMethodOutput(ctx: ParamContext,
+                         delegate: Map[String, Input] => Router.Result[Output]): Router.Result[Response] = {
+      delegate(Map("extra" -> 31337))
+    }
   }
   val tests = Tests{