test: unit test for optioanl inject

Author: gxkl

core/core-decorator/test/decorators.test.ts

@@ -90,8 +90,8 @@ describe('test/decorator.test.ts', () => {
       assert.deepStrictEqual(injectConstructors, [
         { refIndex: 0, refName: 'xCache', objName: 'fooCache' },
         { refIndex: 1, refName: 'cache', objName: 'cache' },
-        { refIndex: 2, refName: 'optional1', objName: 'optional1' },
-        { refIndex: 3, refName: 'optional2', objName: 'optional2' },
+        { refIndex: 2, refName: 'optional1', objName: 'optional1', optional: true },
+        { refIndex: 3, refName: 'optional2', objName: 'optional2', optional: true },
       ]);
     });
   });

plugin/tegg/test/Inject.test.ts

@@ -0,0 +1,60 @@
+import mm from 'egg-mock';
+import path from 'path';
+import assert from 'assert';
+import { BarService } from './fixtures/apps/optional-inject/app/modules/module-a/BarService';
+import { FooService } from './fixtures/apps/optional-inject/app/modules/module-a/FooService';
+
+describe('plugin/tegg/test/Inject.test.ts', () => {
+  let app;
+
+  beforeEach(async () => {
+    mm(process.env, 'EGG_TYPESCRIPT', true);
+    mm(process, 'cwd', () => {
+      return path.join(__dirname, '..');
+    });
+  });
+
+  afterEach(async () => {
+    await app.close();
+    mm.restore();
+  });
+
+  describe('optional', () => {
+    beforeEach(async () => {
+      app = mm.app({
+        baseDir: path.join(__dirname, 'fixtures/apps/optional-inject'),
+        framework: require.resolve('egg'),
+      });
+      await app.ready();
+    });
+
+    it('should work with property', async () => {
+      const barService: BarService = await app.getEggObject(BarService);
+      const res = barService.bar();
+      assert.deepStrictEqual(res, {
+        nil1: 'Y',
+        nil2: 'Y',
+      });
+    });
+
+    it('should work with constructor', async () => {
+      const fooService: FooService = await app.getEggObject(FooService);
+      const res = fooService.foo();
+      assert.deepStrictEqual(res, {
+        nil1: 'Y',
+        nil2: 'Y',
+      });
+    });
+  });
+
+  it('should throw error if no proto found', async () => {
+    app = mm.app({
+      baseDir: path.join(__dirname, 'fixtures/apps/invalid-inject'),
+      framework: require.resolve('egg'),
+    });
+    await assert.rejects(
+      app.ready(),
+      /EggPrototypeNotFound: Object doesNotExist not found in LOAD_UNIT:a/,
+    );
+  });
+});

plugin/tegg/test/fixtures/apps/invalid-inject/app/modules/module-a/BarService.ts

@@ -0,0 +1,11 @@
+import { SingletonProto, Inject } from '@eggjs/tegg';
+
+@SingletonProto()
+export class BarService {
+  @Inject()
+  doesNotExist: object;
+
+  bar() {
+    console.log(this.doesNotExist);
+  }
+}

plugin/tegg/test/fixtures/apps/invalid-inject/app/modules/module-a/package.json

@@ -0,0 +1,6 @@
+{
+  "name": "module-a",
+  "eggModule": {
+    "name": "a"
+  }
+}

plugin/tegg/test/fixtures/apps/invalid-inject/config/config.default.js

@@ -0,0 +1,20 @@
+'use strict';
+
+const path = require('path');
+
+module.exports = function(appInfo) {
+  const config = {
+    keys: 'test key',
+    customLogger: {
+      xxLogger: {
+        file: path.join(appInfo.root, 'logs/xx.log'),
+      },
+    },
+    security: {
+      csrf: {
+        ignoreJSON: false,
+      },
+    },
+  };
+  return config;
+};

plugin/tegg/test/fixtures/apps/invalid-inject/config/plugin.js

@@ -0,0 +1,13 @@
+'use strict';
+
+exports.tracer = {
+  package: 'egg-tracer',
+  enable: true,
+};
+
+exports.teggConfig = {
+  package: '@eggjs/tegg-config',
+  enable: true,
+};
+
+exports.watcher = false;

plugin/tegg/test/fixtures/apps/invalid-inject/package.json

@@ -0,0 +1,3 @@
+{
+  "name": "egg-app"
+}

plugin/tegg/test/fixtures/apps/optional-inject/app/modules/module-a/BarService.ts

@@ -0,0 +1,17 @@
+import { SingletonProto, Inject, InjectOptional } from '@eggjs/tegg';
+
+@SingletonProto()
+export class BarService {
+  @Inject({ optional: true })
+  doesNotExist1: object;
+
+  @InjectOptional()
+  doesNotExist2: object;
+
+  bar() {
+    return {
+      nil1: this.doesNotExist1 ? 'N' : 'Y',
+      nil2: this.doesNotExist2 ? 'N' : 'Y',
+    };
+  }
+}

plugin/tegg/test/fixtures/apps/optional-inject/app/modules/module-a/FooService.ts

@@ -0,0 +1,16 @@
+import { SingletonProto, Inject, InjectOptional } from '@eggjs/tegg';
+
+@SingletonProto()
+export class FooService {
+  constructor(
+    @Inject({ optional: true }) readonly doesNotExist1: object,
+    @InjectOptional() readonly doesNotExist2: object,
+  ) {}
+
+  foo() {
+    return {
+      nil1: this.doesNotExist1 ? 'N' : 'Y',
+      nil2: this.doesNotExist2 ? 'N' : 'Y',
+    };
+  }
+}

plugin/tegg/test/fixtures/apps/optional-inject/app/modules/module-a/package.json

@@ -0,0 +1,6 @@
+{
+  "name": "module-a",
+  "eggModule": {
+    "name": "a"
+  }
+}