feat: allow ctx.router setter (#3460)

app/extend/context.js

@@ -10,7 +10,7 @@ const LOCALS_LIST = Symbol('Context#localsList');
 const COOKIES = Symbol('Context#cookies');
 const CONTEXT_LOGGERS = Symbol('Context#logger');
 const CONTEXT_HTTPCLIENT = Symbol('Context#httpclient');
-
+const CONTEXT_ROUTER = Symbol('Context#router');
 
 const proto = module.exports = {
 
@@ -59,9 +59,19 @@ const proto = module.exports = {
    * ```
    */
   get router() {
-    return this.app.router;
+    if (!this[CONTEXT_ROUTER]) {
+      this[CONTEXT_ROUTER] = this.app.router;
+    }
+    return this[CONTEXT_ROUTER];
   },
 
+  /**
+   * Set router to Context, only use on EggRouter
+   * @param {EggRouter} val router instance
+   */
+  set router(val) {
+    this[CONTEXT_ROUTER] = val;
+  },
 
   /**
    * Get helper instance from {@link Application#Helper}

test/app/extend/context.test.js

@@ -154,7 +154,7 @@ describe('test/app/extend/context.test.js', () => {
     });
     after(() => app.close());
 
-    describe('ctx.router', () => {
+    describe('ctx.router getter and settter', () => {
       it('should work', () => {
         return app.httpRequest()
           .get('/')

test/fixtures/apps/context-config-app/app/controller/home.js

@@ -1,9 +1,12 @@
 'use strict';
 
-exports.index = function* () {
-  this.body = {
-    path: this.router.pathFor('home'),
-    foo: this.foo,
-    bar: this.bar()
+exports.index = async ctx => {
+  const router = ctx.router;
+  // set router ok too
+  ctx.router = router;
+  ctx.body = {
+    path: ctx.router.pathFor('home'),
+    foo: ctx.foo,
+    bar: ctx.bar()
   };
 };