改良测试用环境

Author: cyfung1031

src/app/service/content/create_context.ts

@@ -3,34 +3,11 @@ import { v4 as uuidv4 } from "uuid";
 import type { Message } from "@Packages/message/types";
 import EventEmitter from "eventemitter3";
 import { GMContextApiGet } from "./gm_context";
-import { createGMBase } from "./gm_api";
+import { createGMBase, type GMApiType } from "./gm_api";
+
+function _createContext(options:{[key:string]:any}, scriptGrants:Set<string> | string[]){
+  const context = createGMBase(options);
 
-// 构建沙盒上下文
-export function createContext(
-  scriptRes: ScriptRunResource,
-  GMInfo: any,
-  envPrefix: string,
-  message: Message,
-  scriptGrants: Set<string>
-) {
-  // 按照GMApi构建
-  const valueChangeListener = new Map<number, { name: string; listener: GMTypes.ValueChangeListener }>();
-  const EE: EventEmitter = new EventEmitter();
-  const context = createGMBase({
-    prefix: envPrefix,
-    message,
-    scriptRes,
-    valueChangeListener,
-    EE,
-    runFlag: uuidv4(),
-    eventId: 10000,
-    GM: { info: GMInfo },
-    GM_info: GMInfo,
-    window: {
-      onurlchange: null,
-    },
-    grantSet: new Set(),
-  });
   const grantedAPIs: { [key: string]: any } = {};
   const __methodInject__ = (grant: string): boolean => {
     const grantSet: Set<string> = context.grantSet;
@@ -64,6 +41,59 @@ export function createContext(
       g = g[part] || (g[part] = (grantedAPIs[s] || {}));
     }
   }
-  context.unsafeWindow = window;
+  return <GMApiType>context;
+}
+
+// 构建沙盒上下文
+export function createContext(
+  scriptRes: ScriptRunResource,
+  GMInfo: any,
+  envPrefix: string,
+  message: Message,
+  scriptGrants: Set<string> | string[]
+) {
+  // 按照GMApi构建
+  const valueChangeListener = new Map<number, { name: string; listener: GMTypes.ValueChangeListener }>();
+  const EE: EventEmitter = new EventEmitter();
+  const context = _createContext({
+    prefix: envPrefix,
+    message,
+    scriptRes,
+    valueChangeListener,
+    EE,
+    runFlag: uuidv4(),
+    eventId: 10000,
+    GM: { info: GMInfo },
+    GM_info: GMInfo,
+    window: {
+      onurlchange: null,
+    },
+    grantSet: new Set(),
+    unsafeWindow: window
+  }, scriptGrants);
+  return context;
+}
+
+// 构建测试用沙盒上下文
+export function createTestContext(
+  prefix: string,
+  message: Message,
+  scriptRes: ScriptRunResource,
+  scriptGrants: Set<string> | string[]
+) {
+
+  const valueChangeListener = new Map<number, { name: string; listener: GMTypes.ValueChangeListener }>();
+  const EE: EventEmitter = new EventEmitter();
+  const context = _createContext({
+    prefix,
+    message,
+    scriptRes,
+    valueChangeListener,
+    EE,
+    notificationTagMap: new Map(),
+    eventId: 0,
+    grantSet: new Set(),
+  }, scriptGrants);
+
   return context;
 }

src/app/service/content/gm_api.ts

@@ -108,8 +108,8 @@ class GM_Base implements IGM_Base {
   }
 }
 
-// GMApi 定义 外部用API函数。不使用@protected
-export default class GMApi extends GM_Base {
+// GMApi 定义 外部用API函数。不使用@protected。定義API用。不導出
+class GMApi extends GM_Base {
   /**
    * <tag, notificationId>
    */
@@ -1157,3 +1157,5 @@ export const { createGMBase } = GM_Base;
 
 // 從 GMApi 對象中解構出內部函數，用於後續本地使用，不導出
 const { _GM_getValue, _GM_cookie, _GM_setValue, _GM_xmlhttpRequest } = GMApi;
+
+export type GMApiType = InstanceType<typeof GMApi>;

tests/runtime/gm_api.test.ts

@@ -1,5 +1,5 @@
 import { type Script, ScriptDAO, type ScriptRunResource } from "@App/app/repo/scripts";
-import GMApi from "@App/app/service/content/gm_api";
+import { createTestContext } from "@App/app/service/content/create_context";
 import { initTestEnv, initTestGMApi } from "@Tests/utils";
 import { randomUUID } from "crypto";
 import { newMockXhr } from "mock-xmlhttprequest";
@@ -33,9 +33,9 @@ beforeAll(async () => {
 });
 
 describe("GM xmlHttpRequest", () => {
-  const gmApi = new GMApi("serviceWorker", msg, <ScriptRunResource>{
+  const gmApi = createTestContext("serviceWorker", msg, <ScriptRunResource>{
     uuid: script.uuid
-  });
+  }, ["GM_xmlhttpRequest"]);
   const mockXhr = newMockXhr();
   mockXhr.onSend = async (request) => {
     switch (request.url) {

tests/utils.test.ts

@@ -4,7 +4,7 @@ import { randomUUID } from "crypto";
 import { newMockXhr } from "mock-xmlhttprequest";
 import type { Script, ScriptRunResource } from "@App/app/repo/scripts";
 import { ScriptDAO } from "@App/app/repo/scripts";
-import GMApi from "@App/app/service/content/gm_api";
+import { createTestContext } from "@App/app/service/content/create_context";
 
 initTestEnv();
 
@@ -29,9 +29,9 @@ describe("测试GMApi环境", async () => {
     checktime: 0,
   };
   await new ScriptDAO().save(script);
-  const gmApi = new GMApi("serviceWorker", msg, <ScriptRunResource>{
+  const gmApi = createTestContext("serviceWorker", msg, <ScriptRunResource>{
     uuid: script.uuid
-  });
+  }, ["GM_xmlhttpRequest"]);
   const mockXhr = newMockXhr();
   mockXhr.onSend = async (request) => {
     return request.respond(200, {}, "example");