fix(loader): loadPlugin can be extended

eggjs · May 3, 2017 · 5aa9bee · 5aa9bee
1 parent 89b4df9
commit 5aa9bee
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 2 deletions.
diff --git a/lib/loader/agent_worker_loader.js b/lib/loader/agent_worker_loader.js
@@ -12,7 +12,7 @@ class AgentWorkerLoader extends EggLoader {
    * loadPlugin first, then loadConfig
    */
   loadConfig() {
-    super.loadPlugin();
+    this.loadPlugin();
     super.loadConfig();
   }
 

diff --git a/lib/loader/app_worker_loader.js b/lib/loader/app_worker_loader.js
@@ -13,7 +13,7 @@ class AppWorkerLoader extends EggLoader {
    * @since 1.0.0
    */
   loadConfig() {
-    super.loadPlugin();
+    this.loadPlugin();
     super.loadConfig();
   }
 

diff --git a/test/lib/core/loader/load_plugin.test.js b/test/lib/core/loader/load_plugin.test.js
@@ -4,7 +4,9 @@ const should = require('should');
 const path = require('path');
 const fs = require('fs');
 const mm = require('egg-mock');
+const assert = require('assert');
 const AppWorkerLoader = require('../../../../').AppWorkerLoader;
+const AgentWorkerLoader = require('../../../../').AgentWorkerLoader;
 const utils = require('../../../utils');
 
 const EGG_BASE = path.join(__dirname, '../../../../');
@@ -297,4 +299,35 @@ describe('test/lib/core/loader/load_plugin.test.js', () => {
       from: path.join(baseDir, 'config/plugin.js'),
     });
   });
+
+  it('should customize loadPlugin', () => {
+    const baseDir = utils.getFilepath('apps/loader-plugin');
+    class CustomAppLoader extends AppWorkerLoader {
+      loadPlugin() {
+        this.hasAppLoadPlugin = true;
+        super.loadPlugin();
+      }
+    }
+    const appLoader = new CustomAppLoader({
+      baseDir,
+      app,
+      logger,
+    });
+    appLoader.loadConfig();
+    assert(appLoader.hasAppLoadPlugin === true);
+
+    class CustomAgentLoader extends AgentWorkerLoader {
+      loadPlugin() {
+        this.hasAgentLoadPlugin = true;
+        super.loadPlugin();
+      }
+    }
+    const agentLoader = new CustomAgentLoader({
+      baseDir,
+      app,
+      logger,
+    });
+    agentLoader.loadConfig();
+    assert(agentLoader.hasAgentLoadPlugin === true);
+  });
 });