feat: agent context can be extended (#3478)

eggjs · Feb 20, 2019 · 7147b23 · 7147b23
1 parent a2f0d96
commit 7147b23
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 1 deletion.
diff --git a/lib/loader/agent_worker_loader.js b/lib/loader/agent_worker_loader.js
@@ -18,6 +18,8 @@ class AgentWorkerLoader extends EggLoader {
 
   load() {
     this.loadAgentExtend();
+    this.loadContextExtend();
+
     this.loadCustomAgent();
   }
 }

diff --git a/test/app/extend/context.test.js b/test/app/extend/context.test.js
@@ -146,6 +146,21 @@ describe('test/app/extend/context.test.js', () => {
     });
   });
 
+  describe('agent anonymous context can be extended', () => {
+    let app;
+    before(() => {
+      app = utils.app('apps/custom-context-getlogger');
+      return app.ready();
+    });
+    after(() => app.close());
+
+    it('should extend context as app', () => {
+      const ctx = app.agent.createAnonymousContext();
+      const logger = ctx.getLogger('foo');
+      logger.info('hello');
+    });
+  });
+
   describe('properties', () => {
     let app;
     before(() => {

diff --git a/test/lib/egg.test.js b/test/lib/egg.test.js
@@ -135,7 +135,7 @@ describe('test/lib/egg.test.js', () => {
 
     it('should read timing data', function* () {
       let json = readJson(path.join(baseDir, `run/agent_timing_${process.pid}.json`));
-      assert(json.length === 32);
+      assert(json.length === 39);
       assert(json[0].name === 'Application Start');
       assert(json[0].pid === process.pid);