@eggjs/mock 7.0.0-beta.19 → 7.0.0-beta.20

package/dist/agent-BKYkjoCx.js

@@ -0,0 +1,111 @@
+import { rimraf, sleep } from "./utils-MhP_Krx1.js";
+import { formatOptions } from "./format_options-LRnReKde.js";
+import { context } from "./context-BIduDy1M.js";
+import { setCustomLoader } from "./mock_custom_loader-f_tHRojG.js";
+import { APP_INIT } from "./util-BpppqTXv.js";
+import path from "node:path";
+import { debuglog } from "node:util";
+import { importModule } from "@eggjs/utils";
+import { Base } from "sdk-base";
+import { detectPort } from "detect-port";
+
+//#region src/lib/parallel/agent.ts
+const debug = debuglog("egg/mock/lib/parallel/agent");
+var MockAgent = class extends Base {
+	baseDir;
+	[APP_INIT] = false;
+	#initOnListeners = /* @__PURE__ */ new Set();
+	#initOnceListeners = /* @__PURE__ */ new Set();
+	_instance;
+	constructor(options) {
+		super({ initMethod: "_init" });
+		this.options = options;
+		this.baseDir = this.options.baseDir;
+	}
+	async _init() {
+		if (this.options.beforeInit) {
+			await this.options.beforeInit(this);
+			delete this.options.beforeInit;
+		}
+		if (this.options.clean !== false) {
+			const logDir = path.join(this.options.baseDir, "logs");
+			try {
+				await rimraf(logDir);
+			} catch (err) {
+				console.error(`remove log dir ${logDir} failed: ${err.stack}`);
+			}
+			const runDir = path.join(this.options.baseDir, "run");
+			try {
+				await rimraf(runDir);
+			} catch (err) {
+				console.error(`remove run dir ${runDir} failed: ${err.stack}`);
+			}
+		}
+		this.options.clusterPort = await detectPort();
+		process.env.CLUSTER_PORT = String(this.options.clusterPort);
+		debug("get clusterPort %s", this.options.clusterPort);
+		const { Agent } = await importModule(this.options.framework);
+		const agent = this._instance = new Agent({ ...this.options });
+		Object.assign(agent.context, context);
+		setCustomLoader(agent);
+		debug("agent instantiate");
+		this[APP_INIT] = true;
+		debug("this[APP_INIT] = true");
+		this.#bindEvents();
+		await agent.ready();
+		const msg = {
+			action: "egg-ready",
+			data: this.options
+		};
+		agent.messenger.onMessage(msg);
+		debug("agent ready");
+	}
+	#bindEvents() {
+		debug("bind cache events to agent");
+		for (const args of this.#initOnListeners) {
+			debug("on(%s), use cache and pass to agent", args);
+			this._instance.on(args[0], args[1]);
+			this.removeListener(args[0], args[1]);
+		}
+		for (const args of this.#initOnceListeners) {
+			debug("once(%s), use cache and pass to agent", args);
+			this._instance.once(args[0], args[1]);
+			this.removeListener(args[0], args[1]);
+		}
+	}
+	on(...args) {
+		if (this[APP_INIT]) {
+			debug("on(%s), pass to agent", args);
+			this._instance.on(args[0], args[1]);
+		} else {
+			debug("on(%s), cache it because agent has not init", args);
+			this.#initOnListeners.add(args);
+			super.on(args[0], args[1]);
+		}
+		return this;
+	}
+	once(...args) {
+		if (this[APP_INIT]) {
+			debug("once(%s), pass to agent", args);
+			this._instance.once(args[0], args[1]);
+		} else {
+			debug("once(%s), cache it because agent has not init", args);
+			this.#initOnceListeners.add(args);
+			super.on(args[0], args[1]);
+		}
+		return this;
+	}
+	/**
+	* close agent
+	*/
+	async _close() {
+		if (this._instance) await this._instance.close();
+		else await sleep(200);
+	}
+};
+function createAgent(options) {
+	return new MockAgent(formatOptions(options));
+}
+
+//#endregion
+export { MockAgent, createAgent };

package/dist/agent-CHCe8tnW.d.ts

@@ -0,0 +1,24 @@
+import { MockApplicationOptions, MockOptions } from "./types-CjZ-JeCo.js";
+import { APP_INIT } from "./util-1xYBaaoh.js";
+import { Agent } from "egg";
+import { Base } from "sdk-base";
+
+//#region src/lib/parallel/agent.d.ts
+declare class MockAgent extends Base {
+  #private;
+  options: MockApplicationOptions;
+  baseDir: string;
+  [APP_INIT]: boolean;
+  _instance: Agent;
+  constructor(options: MockApplicationOptions);
+  _init(): Promise<void>;
+  on(...args: any[]): this;
+  once(...args: any[]): this;
+  /**
+   * close agent
+   */
+  _close(): Promise<void>;
+}
+declare function createAgent(options: MockOptions): MockAgent;
+//#endregion
+export { MockAgent, createAgent };

package/dist/agent-DZ_fHoxJ.js

@@ -0,0 +1,48 @@
+import { getMockAgent, restoreMockAgent } from "./mock_agent-CxT7Q0_N.js";
+import { createMockHttpClient } from "./mock_httpclient-B6UVI7GS.js";
+import { Agent } from "egg";
+import { mock, restore } from "mm";
+
+//#region src/app/extend/agent.ts
+var AgentUnittest = class extends Agent {
+	_mockHttpClient;
+	/**
+	* mock httpclient
+	* @alias mockHttpClient
+	* @function App#mockHttpclient
+	*/
+	mockHttpclient(mockUrl, mockMethod, mockResult) {
+		return this.mockHttpClient(mockUrl, mockMethod, mockResult);
+	}
+	/**
+	* mock httpclient
+	* @function App#mockHttpClient
+	*/
+	mockHttpClient(mockUrl, mockMethod, mockResult) {
+		if (!this._mockHttpClient) this._mockHttpClient = createMockHttpClient(this);
+		return this._mockHttpClient(mockUrl, mockMethod, mockResult);
+	}
+	/**
+	* get mock httpclient agent
+	* @function Agent#mockHttpclientAgent
+	*/
+	mockAgent() {
+		return getMockAgent(this);
+	}
+	async mockAgentRestore() {
+		await restoreMockAgent();
+	}
+	/**
+	* @see mm#restore
+	* @function Agent#mockRestore
+	*/
+	mockRestore = restore;
+	/**
+	* @see mm
+	* @function Agent#mm
+	*/
+	mm = mock;
+};
+
+//#endregion
+export { AgentUnittest };

package/dist/agent_handler-DeUtoRVT.js

@@ -0,0 +1,26 @@
+import { getEggOptions } from "./utils-MhP_Krx1.js";
+import { createAgent } from "./agent-BKYkjoCx.js";
+import { debuglog } from "node:util";
+
+//#region src/lib/agent_handler.ts
+const debug = debuglog("egg/mock/lib/agent_handler");
+let agent;
+async function setupAgent() {
+	debug("setupAgent call, env.ENABLE_MOCHA_PARALLEL: %s, process.env.AUTO_AGENT: %s, agent: %s", process.env.ENABLE_MOCHA_PARALLEL, process.env.AUTO_AGENT, !!agent);
+	if (agent) {
+		await agent.ready();
+		return agent;
+	}
+	if (process.env.ENABLE_MOCHA_PARALLEL && process.env.AUTO_AGENT) {
+		agent = createAgent(getEggOptions());
+		await agent.ready();
+	}
+	return agent;
+}
+async function closeAgent() {
+	debug("setupAgent call, agent: %s", !!agent);
+	if (agent) await agent.close();
+}
+
+//#endregion
+export { closeAgent, setupAgent };

package/dist/app/extend/agent.d.ts

@@ -1,4 +1,4 @@
-import { MockHttpClientMethod, MockResultFunction, MockResultOptions } from "../../lib/mock_httpclient.js";
+import { MockHttpClientMethod, MockResultFunction, MockResultOptions } from "../../mock_httpclient-1EgQkMxi.js";
 import { Agent } from "egg";
 import { mock, restore } from "mm";
 import * as urllib0 from "urllib";

package/dist/app/extend/agent.js

@@ -1,48 +1,5 @@
-import { getMockAgent, restoreMockAgent } from "../../lib/mock_agent.js";
-import { createMockHttpClient } from "../../lib/mock_httpclient.js";
-import { Agent } from "egg";
-import { mock, restore } from "mm";
+import "../../mock_agent-CxT7Q0_N.js";
+import "../../mock_httpclient-B6UVI7GS.js";
+import { AgentUnittest } from "../../agent-DZ_fHoxJ.js";
 
-//#region src/app/extend/agent.ts
-var AgentUnittest = class extends Agent {
-	_mockHttpClient;
-	/**
-	* mock httpclient
-	* @alias mockHttpClient
-	* @function App#mockHttpclient
-	*/
-	mockHttpclient(mockUrl, mockMethod, mockResult) {
-		return this.mockHttpClient(mockUrl, mockMethod, mockResult);
-	}
-	/**
-	* mock httpclient
-	* @function App#mockHttpClient
-	*/
-	mockHttpClient(mockUrl, mockMethod, mockResult) {
-		if (!this._mockHttpClient) this._mockHttpClient = createMockHttpClient(this);
-		return this._mockHttpClient(mockUrl, mockMethod, mockResult);
-	}
-	/**
-	* get mock httpclient agent
-	* @function Agent#mockHttpclientAgent
-	*/
-	mockAgent() {
-		return getMockAgent(this);
-	}
-	async mockAgentRestore() {
-		await restoreMockAgent();
-	}
-	/**
-	* @see mm#restore
-	* @function Agent#mockRestore
-	*/
-	mockRestore = restore;
-	/**
-	* @see mm
-	* @function Agent#mm
-	*/
-	mm = mock;
-};
-
-//#endregion
 export { AgentUnittest as default };

package/dist/app/extend/application.d.ts

@@ -1,175 +1,5 @@
-import { MockHttpClientMethod, MockResultFunction, MockResultOptions } from "../../lib/mock_httpclient.js";
-import { EggTestRequest } from "../../lib/supertest.js";
-import { MockOptions } from "../../lib/types.js";
-import { Application, Context } from "egg";
-import { mock } from "mm";
-import { IncomingMessage } from "node:http";
-import { Logger } from "egg-logger";
-import { MockAgent } from "urllib";
-
-//#region src/app/extend/application.d.ts
-interface MockContextOptions {
-  /**
-   * mock ctxStorage or not, default is `true`
-   */
-  mockCtxStorage?: boolean;
-  /**
-   * reuse ctxStorage or not, default is `true`
-   */
-  reuseCtxStorage?: boolean;
-}
-interface MockContextData {
-  headers?: Record<string, string | string[]>;
-  [key: string]: any;
-}
-interface MockContext extends Context {
-  service: any;
-}
-declare abstract class ApplicationUnittest extends Application {
-  [key: string]: any;
-  options: MockOptions & Application['options'];
-  _mockHttpClient?: MockHttpClientMethod;
-  agent: NonNullable<Application['agent']>;
-  /**
-   * mock Context
-   * @function App#mockContext
-   * @param {Object} data - ctx data
-   * @param {Object} [options] - mock ctx options
-   * @example
-   * ```js
-   * const ctx = app.mockContext({
-   *   user: {
-   *     name: 'Jason'
-   *   }
-   * });
-   * console.log(ctx.user.name); // Jason
-   *
-   * // controller
-   * module.exports = function*() {
-   *   this.body = this.user.name;
-   * };
-   * ```
-   */
-  mockContext(data?: MockContextData, options?: MockContextOptions): MockContext;
-  mockContextScope(fn: (ctx?: MockContext) => Promise<any>, data?: MockContextData): Promise<any>;
-  /**
-   * mock cookie session
-   * @function App#mockSession
-   * @param {Object} data - session object
-   */
-  mockSession(data: any): this;
-  /**
-   * Mock service
-   * @function App#mockService
-   * @param {String} service - name
-   * @param {String} methodName - method
-   * @param {Object|Function|Error} fn - mock you data
-   */
-  mockService(service: string | any, methodName: string, fn: any): this;
-  /**
-   * mock service that return error
-   * @function App#mockServiceError
-   * @param {String} service - name
-   * @param {String} methodName - method
-   * @param {Error} [err] - error information
-   */
-  mockServiceError(service: string | any, methodName: string, err?: string | Error): this;
-  _mockFn(obj: any, name: string, data: any): void;
-  /**
-   * mock request
-   * @function App#mockRequest
-   * @param {Request} req - mock request
-   */
-  mockRequest(req: MockContextData): IncomingMessage;
-  /**
-   * mock cookies
-   * @function App#mockCookies
-   */
-  mockCookies(cookies: Record<string, string | string[]>): this;
-  /**
-   * mock header
-   * @function App#mockHeaders
-   */
-  mockHeaders(headers: Record<string, string | string[]>): this;
-  /**
-   * mock csrf
-   * @function App#mockCsrf
-   * @since 1.11
-   */
-  mockCsrf(): this;
-  /**
-   * mock httpclient
-   * @alias mockHttpClient
-   * @function App#mockHttpclient
-   */
-  mockHttpclient(mockUrl: string | RegExp, mockMethod: string | string[] | MockResultOptions | MockResultFunction, mockResult?: MockResultOptions | MockResultFunction | string): this;
-  /**
-   * mock httpclient
-   * @function App#mockHttpClient
-   */
-  mockHttpClient(mockUrl: string | RegExp, mockMethod: string | string[] | MockResultOptions | MockResultFunction, mockResult?: MockResultOptions | MockResultFunction | string): this;
-  /**
-   * @deprecated Please use app.mockHttpClient instead of app.mockUrllib
-   */
-  mockUrllib(mockUrl: string | RegExp, mockMethod: string | string[] | MockResultOptions | MockResultFunction, mockResult?: MockResultOptions | MockResultFunction | string): this;
-  /**
-   * get mock httpclient agent
-   * @function App#mockHttpclientAgent
-   */
-  mockAgent(): MockAgent;
-  mockAgentRestore(): Promise<void>;
-  /**
-   * @see mm#restore
-   * @function App#mockRestore
-   */
-  mockRestore(): Promise<void>;
-  /**
-   * @see mm
-   * @function App#mm
-   */
-  get mm(): typeof mock;
-  /**
-   * override loadAgent
-   * @function App#loadAgent
-   */
-  loadAgent(): void;
-  /**
-   * mock serverEnv
-   * @function App#mockEnv
-   * @param {String} env - serverEnv
-   */
-  mockEnv(env: string): this;
-  /**
-   * http request helper
-   * @function App#httpRequest
-   * @return {SupertestRequest} req - supertest request
-   * @see https://github.com/visionmedia/supertest
-   */
-  httpRequest(): EggTestRequest;
-  /**
-   * collection logger message, then can be use on `expectLog()`
-   * @param {String|Logger} [logger] - logger instance, default is `app.logger`
-   * @function App#mockLog
-   */
-  mockLog(logger?: string | Logger): void;
-  __checkExpectLog(expectOrNot: boolean, str: string | RegExp, logger?: string | Logger): void;
-  /**
-   * expect str/regexp in the logger, if your server disk is slow, please call `mockLog()` first.
-   * @param {String|RegExp} str - test str or regexp
-   * @param {String|Logger} [logger] - logger instance, default is `ctx.logger`
-   * @function App#expectLog
-   */
-  expectLog(str: string | RegExp, logger?: string | Logger): void;
-  /**
-   * not expect str/regexp in the logger, if your server disk is slow, please call `mockLog()` first.
-   * @param {String|RegExp} str - test str or regexp
-   * @param {String|Logger} [logger] - logger instance, default is `ctx.logger`
-   * @function App#notExpectLog
-   */
-  notExpectLog(str: string | RegExp, logger?: string | Logger): void;
-  backgroundTasksFinished(): Promise<void>;
-  get _backgroundTasks(): Promise<any>[];
-  set _backgroundTasks(tasks: Promise<any>[]);
-}
-//#endregion
+import "../../mock_httpclient-1EgQkMxi.js";
+import "../../supertest-BIds28yL.js";
+import "../../types-CjZ-JeCo.js";
+import { ApplicationUnittest, MockContext, MockContextData, MockContextOptions } from "../../application-BPpvw5dg.js";
 export { MockContext, MockContextData, MockContextOptions, ApplicationUnittest as default };