@midwayjs/mock 3.2.2 → 3.3.2

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/mock.d.ts

@@ -1,5 +1,5 @@
 import { IMidwayApplication, IMidwayContext } from '@midwayjs/core';
-export declare function mockSession(app: IMidwayApplication, key: string, value: string): void;
+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;

package/dist/mock.js

@@ -6,6 +6,9 @@ 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.');
     }
@@ -35,17 +38,32 @@ function mockHeader(app, headerKey, headerValue) {
 exports.mockHeader = mockHeader;
 function mockClassProperty(clzz, propertyName, value) {
     const mockService = getMockService();
-    return mockService.mockClassProperty(clzz, propertyName, value);
+    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();
-    return mockService.mockProperty(obj, key, value);
+    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();
-    mockService.restore();
+    if (mockService) {
+        mockService.restore();
+    }
+    else {
+        core_1.MidwayMockService.prepareMocks = [];
+    }
 }
 exports.restoreAllMocks = restoreAllMocks;
 function mockContext(app, key, value) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@midwayjs/mock",
-  "version": "3.2.2",
+  "version": "3.3.2",
   "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.2",
+    "@midwayjs/core": "^3.3.2",
     "@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": "6bc097035302d01ae901d76a10e080db8554a864"
+  "gitHead": "2c7c235421ba904f8943efe31a71f452089d5d53"
 }