@midwayjs/mock 3.2.0 → 3.3.1

package/dist/client/index.d.ts

@@ -1,5 +1,5 @@
 export * from './http';
 export * from './rabbitMQ';
 export * from './socketio';
-export * from './ws';
+export * from './ws.client';
 //# sourceMappingURL=index.d.ts.map

package/dist/client/index.js

@@ -17,5 +17,5 @@ Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./http"), exports);
 __exportStar(require("./rabbitMQ"), exports);
 __exportStar(require("./socketio"), exports);
-__exportStar(require("./ws"), exports);
+__exportStar(require("./ws.client"), exports);
 //# sourceMappingURL=index.js.map

package/dist/client/{ws.d.ts → ws.client.d.ts}

@@ -3,4 +3,4 @@ import * as WebSocket from 'ws';
 import * as http from 'http';
 import * as url from 'url';
 export declare function createWebSocketClient(address: string | url.URL, options?: WebSocket.ClientOptions | http.ClientRequestArgs): Promise<WebSocket>;
-//# sourceMappingURL=ws.d.ts.map
+//# sourceMappingURL=ws.client.d.ts.map

package/dist/client/{ws.js → ws.client.js}

@@ -11,4 +11,4 @@ async function createWebSocketClient(address, options) {
     });
 }
 exports.createWebSocketClient = createWebSocketClient;
-//# sourceMappingURL=ws.js.map
+//# sourceMappingURL=ws.client.js.map

package/dist/index.d.ts

@@ -1,4 +1,5 @@
 export { create, close, createApp, createFunctionApp, createLightApp, } from './creator';
 export * from './client/index';
 export { transformFrameworkToConfiguration } from './utils';
+export * from './mock';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.js

@@ -24,4 +24,5 @@ Object.defineProperty(exports, "createLightApp", { enumerable: true, get: functi
 __exportStar(require("./client/index"), exports);
 var utils_1 = require("./utils");
 Object.defineProperty(exports, "transformFrameworkToConfiguration", { enumerable: true, get: function () { return utils_1.transformFrameworkToConfiguration; } });
+__exportStar(require("./mock"), exports);
 //# sourceMappingURL=index.js.map

package/dist/mock.d.ts

@@ -0,0 +1,8 @@
+import { IMidwayApplication, IMidwayContext } from '@midwayjs/core';
+export declare function mockSession(app: IMidwayApplication, key: string, value: any): void;
+export declare function mockHeader(app: IMidwayApplication, headerKey: string, headerValue: string): void;
+export declare function mockClassProperty(clzz: new (...args: any[]) => any, propertyName: string, value: any): void;
+export declare function mockProperty(obj: any, key: string, value: any): void;
+export declare function restoreAllMocks(): void;
+export declare function mockContext(app: IMidwayApplication, key: string | ((ctx: IMidwayContext) => void), value?: PropertyDescriptor | any): void;
+//# sourceMappingURL=mock.d.ts.map

package/dist/mock.js

@@ -0,0 +1,74 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.mockContext = exports.restoreAllMocks = exports.mockProperty = exports.mockClassProperty = exports.mockHeader = exports.mockSession = void 0;
+const core_1 = require("@midwayjs/core");
+function getMockService(app) {
+    if (!app) {
+        app = (0, core_1.getCurrentMainApp)();
+    }
+    if (!app) {
+        return null;
+    }
+    if (!app.getApplicationContext) {
+        throw new Error('[mock]: app.getApplicationContext is undefined.');
+    }
+    const applicationContext = app.getApplicationContext();
+    const mockService = applicationContext.get(core_1.MidwayMockService);
+    if (!mockService) {
+        throw new Error('[mock]: MidwayMockService is undefined.');
+    }
+    return mockService;
+}
+function mockSession(app, key, value) {
+    const mockService = getMockService(app);
+    mockService.mockContext(app, (ctx) => {
+        if (!ctx.session) {
+            ctx.session = {};
+        }
+        ctx.session[key] = value;
+    });
+}
+exports.mockSession = mockSession;
+function mockHeader(app, headerKey, headerValue) {
+    const mockService = getMockService(app);
+    mockService.mockContext(app, (ctx) => {
+        ctx.headers[headerKey] = headerValue;
+    });
+}
+exports.mockHeader = mockHeader;
+function mockClassProperty(clzz, propertyName, value) {
+    const mockService = getMockService();
+    if (!mockService) {
+        return core_1.MidwayMockService.mockClassProperty(clzz, propertyName, value);
+    }
+    else {
+        return mockService.mockClassProperty(clzz, propertyName, value);
+    }
+}
+exports.mockClassProperty = mockClassProperty;
+function mockProperty(obj, key, value) {
+    const mockService = getMockService();
+    if (!mockService) {
+        return core_1.MidwayMockService.mockProperty(obj, key, value);
+    }
+    else {
+        return mockService.mockProperty(obj, key, value);
+    }
+}
+exports.mockProperty = mockProperty;
+function restoreAllMocks() {
+    const mockService = getMockService();
+    if (mockService) {
+        mockService.restore();
+    }
+    else {
+        core_1.MidwayMockService.prepareMocks = [];
+    }
+}
+exports.restoreAllMocks = restoreAllMocks;
+function mockContext(app, key, value) {
+    const mockService = getMockService(app);
+    mockService.mockContext(app, key, value);
+}
+exports.mockContext = mockContext;
+//# sourceMappingURL=mock.js.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@midwayjs/mock",
-  "version": "3.2.0",
+  "version": "3.3.1",
   "description": "create your test app from midway framework",
   "main": "dist/index",
   "typings": "dist/index.d.ts",
@@ -26,7 +26,7 @@
   },
   "license": "MIT",
   "devDependencies": {
-    "@midwayjs/core": "^3.2.0",
+    "@midwayjs/core": "^3.3.1",
     "@midwayjs/decorator": "^3.1.6",
     "@midwayjs/logger": "^2.15.0",
     "@types/amqplib": "0.8.2",
@@ -45,5 +45,5 @@
     "type": "git",
     "url": "http://github.com/midwayjs/midway.git"
   },
-  "gitHead": "cee3e5ddbb4f0cc5f80176a7b1e271e809686a97"
+  "gitHead": "f603b622b205ec126de628eeab59f90045a75dd2"
 }