stub contexts return values

diff --git a/test/stub.js b/test/stub.js
index aa8f54801191cb39a59714067471a7e9afb31e36..352d9519ded35525ac52f9da63ce5e6c1d76f490 100644 (file)
--- a/test/stub.js
+++ b/test/stub.js
@@ -108,12 +108,12 @@ function stubCreator(Abstract, metadata = stubMetadata()) {
       this._sinonInit();
     }
 
-    async context(fn) {
-      await fn({});
+    async context(fn, ...args) {
+      return await fn({}, ...args);
     }
 
-    async transaction(dbCtx, fn) {
-      await fn(dbCtx);
+    async transaction(dbCtx, fn, ...args) {
+      return await fn(dbCtx, ...args);
     }
 
     _sinonInit() {