vite-plugin-vue-testid 1.0.7 → 1.0.8

package/README.en.md

@@ -13,6 +13,7 @@ A Vite plugin that auto-injects `data-testid` into Vue component templates via c
 - **Date panel mode awareness** — date / month / year / decade panels are distinguished automatically
 - **MutationObserver re-injection** — New DOM elements from panel switching, tree node expand/collapse are detected and injected automatically
 - **Extensible** — Custom component lists, testId generation rules, and overlay injection strategies are supported
+- **prefixCls compatible** — Supports ant-design-vue global ConfigProvider custom `prefixCls` with auto-detection or manual config
 
 ---
 
@@ -243,6 +244,14 @@ const cleanup = setupAntDropdownTestIds({
    */
   attributeName: 'data-testid',
 
+  /**
+   * ant-design-vue global CSS class prefix.
+   * Required when ConfigProvider prefixCls is set to a non-default value.
+   * When not set, auto-detects from DOM (CSS vars → button class → fallback 'ant').
+   * @default auto-detect || 'ant'
+   */
+  prefixCls: 'myapp',
+
   /**
    * Custom panel selector → injection strategy mapping
    * key: CSS selector
@@ -255,6 +264,7 @@ const cleanup = setupAntDropdownTestIds({
       // ctx.trigger       — trigger HTMLElement | null
       // ctx.triggerTestId — the trigger's testId value
       // ctx.attrName      — testId attribute name
+      // ctx.prefixCls     — ant-design-vue CSS class prefix
       const prefix = `${ctx.triggerTestId}-dropdown`
       ctx.container.setAttribute(ctx.attrName, prefix)
       // ... custom injection logic

package/README.md

@@ -13,6 +13,7 @@
 - **日期面板模式感知** — date / month / year / decade 面板自动区分
 - **MutationObserver 重注入** — 面板切换、树节点展开/折叠时自动检测并注入新 DOM
 - **可扩展** — 支持自定义组件列表、testId 生成规则、弹出面板注入策略
+- **prefixCls 兼容** — 支持 ant-design-vue 全局 ConfigProvider 自定义 `prefixCls`，自动检测或手动配置
 
 ---
 
@@ -243,6 +244,14 @@ const cleanup = setupAntDropdownTestIds({
    */
   attributeName: 'data-testid',
 
+  /**
+   * ant-design-vue 全局 CSS 类名前缀。
+   * 如果通过 ConfigProvider 配置了 prefixCls，必须传入相同值。
+   * 未配置时自动从 DOM 检测（扫描 CSS 变量 → button 类名 → 回退 'ant'）。
+   * @default auto-detect || 'ant'
+   */
+  prefixCls: 'myapp',
+
   /**
    * 自定义面板选择器 → 注入策略映射
    * key: CSS 选择器
@@ -251,10 +260,11 @@ const cleanup = setupAntDropdownTestIds({
   panels: {
     '.ant-picker-dropdown': undefined,                     // 禁用内置 Picker 面板注入
     '.my-custom-dropdown': (ctx) => {
-      // ctx.container   — 面板容器 HTMLElement
-      // ctx.trigger     — 触发器 HTMLElement | null
+      // ctx.container     — 面板容器 HTMLElement
+      // ctx.trigger       — 触发器 HTMLElement | null
       // ctx.triggerTestId — 触发器的 testId 值
-      // ctx.attrName    — testId 属性名
+      // ctx.attrName      — testId 属性名
+      // ctx.prefixCls     — ant-design-vue CSS 类名前缀
       const prefix = `${ctx.triggerTestId}-dropdown`
       ctx.container.setAttribute(ctx.attrName, prefix)
       // ... 自定义注入逻辑

package/dist/runtime.d.ts

@@ -7,6 +7,9 @@
  * import { setupAntDropdownTestIds } from 'vite-plugin-vue-testid/runtime'
  *
  * setupAntDropdownTestIds()
+ *
+ * // 如果全局配置了 prefixCls（如 ConfigProvider prefixCls="myapp"）：
+ * setupAntDropdownTestIds({ prefixCls: 'myapp' })
  * ```
  */
 export interface RuntimeOptions {
@@ -15,6 +18,13 @@ export interface RuntimeOptions {
      * @default 'data-testid'
      */
     attributeName?: string;
+    /**
+     * ant-design-vue 全局 CSS 类名前缀。
+     * 如果通过 ConfigProvider 配置了 prefixCls，需要在此传入相同值。
+     * 未配置时自动从 DOM 检测，检测失败回退到 'ant'。
+     * @default auto-detect || 'ant'
+     */
+    prefixCls?: string;
     /**
      * 需要监听的 teleport 面板 CSS 选择器映射
      * key: 面板容器选择器
@@ -31,6 +41,8 @@ export interface PanelContext {
     triggerTestId: string;
     /** testId 属性名 */
     attrName: string;
+    /** ant-design-vue 当前生效的 CSS 类名前缀 */
+    prefixCls: string;
 }
 export type PanelInjectStrategy = (ctx: PanelContext) => void;
 /**

package/dist/runtime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"runtime.d.ts","sourceRoot":"","sources":["../src/runtime.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,MAAM,WAAW,cAAc;IAC7B;;;OAGG;IACH,aAAa,CAAC,EAAE,MAAM,CAAA;IAEtB;;;;OAIG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,mBAAmB,GAAG,SAAS,CAAC,CAAA;CACzD;AAED,MAAM,WAAW,YAAY;IAC3B,kBAAkB;IAClB,SAAS,EAAE,WAAW,CAAA;IACtB,iBAAiB;IACjB,OAAO,EAAE,WAAW,GAAG,IAAI,CAAA;IAC3B,oBAAoB;IACpB,aAAa,EAAE,MAAM,CAAA;IACrB,iBAAiB;IACjB,QAAQ,EAAE,MAAM,CAAA;CACjB;AAED,MAAM,MAAM,mBAAmB,GAAG,CAAC,GAAG,EAAE,YAAY,KAAK,IAAI,CAAA;AA+a7D;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkCG;AACH,wBAAgB,uBAAuB,CAAC,IAAI,GAAE,cAAmB,GAAG,MAAM,IAAI,CAkF7E;AAED;;;GAGG;AACH,wBAAgB,sBAAsB,CAAC,IAAI,GAAE,cAAmB,GAAG,IAAI,CA+CtE"}
+{"version":3,"file":"runtime.d.ts","sourceRoot":"","sources":["../src/runtime.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAEH,MAAM,WAAW,cAAc;IAC7B;;;OAGG;IACH,aAAa,CAAC,EAAE,MAAM,CAAA;IAEtB;;;;;OAKG;IACH,SAAS,CAAC,EAAE,MAAM,CAAA;IAElB;;;;OAIG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,mBAAmB,GAAG,SAAS,CAAC,CAAA;CACzD;AAED,MAAM,WAAW,YAAY;IAC3B,kBAAkB;IAClB,SAAS,EAAE,WAAW,CAAA;IACtB,iBAAiB;IACjB,OAAO,EAAE,WAAW,GAAG,IAAI,CAAA;IAC3B,oBAAoB;IACpB,aAAa,EAAE,MAAM,CAAA;IACrB,iBAAiB;IACjB,QAAQ,EAAE,MAAM,CAAA;IAChB,oCAAoC;IACpC,SAAS,EAAE,MAAM,CAAA;CAClB;AAED,MAAM,MAAM,mBAAmB,GAAG,CAAC,GAAG,EAAE,YAAY,KAAK,IAAI,CAAA;AA4e7D;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkCG;AACH,wBAAgB,uBAAuB,CAAC,IAAI,GAAE,cAAmB,GAAG,MAAM,IAAI,CAqF7E;AAED;;;GAGG;AACH,wBAAgB,sBAAsB,CAAC,IAAI,GAAE,cAAmB,GAAG,IAAI,CAiDtE"}

package/dist/runtime.js

@@ -1,8 +1,40 @@
 // src/runtime.ts
-function getPanelMode(panel) {
-  if (panel.classList.contains("ant-picker-month-panel")) return "month";
-  if (panel.classList.contains("ant-picker-year-panel")) return "year";
-  if (panel.classList.contains("ant-picker-decade-panel")) return "decade";
+function cls(prefix, name) {
+  return `.${prefix}-${name}`;
+}
+function detectPrefixCls() {
+  try {
+    const styles = getComputedStyle(document.documentElement);
+    for (let i = 0; i < styles.length; i++) {
+      const prop = styles[i];
+      const m = prop.match(/^--(ant|[a-z]+)-/);
+      if (m && m[1]) return m[1];
+    }
+    const buttons = document.querySelectorAll('[class*="-btn"]');
+    for (const btn of buttons) {
+      for (const clsName of btn.classList) {
+        if (clsName.endsWith("-btn") && clsName.length > 4) {
+          return clsName.slice(0, -4);
+        }
+      }
+    }
+    const antElements = document.querySelectorAll('[class*="ant-"]');
+    if (antElements.length > 0) return "ant";
+  } catch {
+  }
+  return "ant";
+}
+var _cachedPrefixCls = null;
+function resolvePrefixCls(userPrefix) {
+  if (userPrefix) return userPrefix;
+  if (_cachedPrefixCls !== null) return _cachedPrefixCls;
+  _cachedPrefixCls = detectPrefixCls();
+  return _cachedPrefixCls;
+}
+function getPanelMode(panel, pfx) {
+  if (panel.classList.contains(`${pfx}-picker-month-panel`)) return "month";
+  if (panel.classList.contains(`${pfx}-picker-year-panel`)) return "year";
+  if (panel.classList.contains(`${pfx}-picker-decade-panel`)) return "decade";
   return "date";
 }
 var MONTH_NAMES = [
@@ -19,8 +51,8 @@ var MONTH_NAMES = [
   "nov",
   "dec"
 ];
-function injectHeaderViewButtons(panel, prefix, attrName, suffix) {
-  const headerView = panel.querySelector(".ant-picker-header-view");
+function injectHeaderViewButtons(panel, prefix, attrName, suffix, pfx) {
+  const headerView = panel.querySelector(cls(pfx, "picker-header-view"));
   if (!headerView) return;
   const buttons = headerView.querySelectorAll("button");
   buttons.forEach((btn) => {
@@ -43,25 +75,25 @@ function injectHeaderViewButtons(panel, prefix, attrName, suffix) {
   });
 }
 var _observedContainers = /* @__PURE__ */ new WeakSet();
-function injectSinglePanel(panel, prefix, attrName, suffix) {
-  const mode = getPanelMode(panel);
+function injectSinglePanel(panel, prefix, attrName, suffix, pfx) {
+  const mode = getPanelMode(panel, pfx);
   const setAttr = (selector, name) => {
     const el = panel.querySelector(selector);
     if (el && !el.hasAttribute(attrName)) {
       el.setAttribute(attrName, `${prefix}-${name}${suffix}`);
     }
   };
-  setAttr(".ant-picker-header-super-prev-btn", "super-prev");
-  setAttr(".ant-picker-header-prev-btn", "prev");
-  setAttr(".ant-picker-header-next-btn", "next");
-  setAttr(".ant-picker-header-super-next-btn", "super-next");
-  setAttr(".ant-picker-header-view", `header-${mode}`);
-  injectHeaderViewButtons(panel, prefix, attrName, suffix);
-  injectPanelCells(panel, prefix, attrName, suffix);
+  setAttr(cls(pfx, "picker-header-super-prev-btn"), "super-prev");
+  setAttr(cls(pfx, "picker-header-prev-btn"), "prev");
+  setAttr(cls(pfx, "picker-header-next-btn"), "next");
+  setAttr(cls(pfx, "picker-header-super-next-btn"), "super-next");
+  setAttr(cls(pfx, "picker-header-view"), `header-${mode}`);
+  injectHeaderViewButtons(panel, prefix, attrName, suffix, pfx);
+  injectPanelCells(panel, prefix, attrName, suffix, pfx);
 }
-function injectPanelCells(panel, prefix, attrName, suffix) {
-  const mode = getPanelMode(panel);
-  const cells = panel.querySelectorAll(".ant-picker-cell-inner");
+function injectPanelCells(panel, prefix, attrName, suffix, pfx) {
+  const mode = getPanelMode(panel, pfx);
+  const cells = panel.querySelectorAll(cls(pfx, "picker-cell-inner"));
   cells.forEach((cell) => {
     const el = cell;
     if (el.hasAttribute(attrName)) return;
@@ -86,30 +118,29 @@ function injectPanelCells(panel, prefix, attrName, suffix) {
         break;
     }
   });
-  injectMenuItems(attrName);
 }
 function injectPickerPanel(ctx) {
-  const { container, triggerTestId, attrName } = ctx;
+  const { container, triggerTestId, attrName, prefixCls: pfx } = ctx;
   const prefix = `${triggerTestId}-dropdown`;
   container.setAttribute(attrName, prefix);
-  const isRange = container.classList.contains("ant-picker-dropdown-range");
+  const isRange = container.classList.contains(`${pfx}-picker-dropdown-range`);
   const injectAllPanels = () => {
-    const panels = container.querySelectorAll(".ant-picker-panel");
+    const panels = container.querySelectorAll(cls(pfx, "picker-panel"));
     if (isRange && panels.length >= 2) {
-      injectSinglePanel(panels[0], prefix, attrName, "-left");
-      injectSinglePanel(panels[1], prefix, attrName, "-right");
+      injectSinglePanel(panels[0], prefix, attrName, "-left", pfx);
+      injectSinglePanel(panels[1], prefix, attrName, "-right", pfx);
     } else {
       const panel = panels[0] || container;
-      injectSinglePanel(panel, prefix, attrName, "");
+      injectSinglePanel(panel, prefix, attrName, "", pfx);
     }
-    const presets = container.querySelectorAll(".ant-picker-presets .ant-tag");
+    const presets = container.querySelectorAll(`${cls(pfx, "picker-presets")} .${pfx}-tag`);
     presets.forEach((tag) => {
       const el = tag;
       if (el.hasAttribute(attrName)) return;
       const text = el.textContent?.trim() || "";
       el.setAttribute(attrName, `${prefix}-preset-${text}`);
     });
-    const timeColumns = container.querySelectorAll(".ant-picker-time-panel-column");
+    const timeColumns = container.querySelectorAll(cls(pfx, "picker-time-panel-column"));
     timeColumns.forEach((col, i) => {
       const el = col;
       if (el.hasAttribute(attrName)) return;
@@ -128,9 +159,9 @@ function injectPickerPanel(ctx) {
         btn.setAttribute(attrName, `${prefix}-${name}`);
       }
     };
-    setFooterBtn(".ant-picker-ok button", "ok");
-    setFooterBtn(".ant-picker-now button", "now");
-    setFooterBtn(".ant-picker-now-btn", "now");
+    setFooterBtn(`${cls(pfx, "picker-ok")} button`, "ok");
+    setFooterBtn(`${cls(pfx, "picker-now")} button`, "now");
+    setFooterBtn(cls(pfx, "picker-now-btn"), "now");
   };
   injectAllPanels();
   if (!_observedContainers.has(container)) {
@@ -146,14 +177,14 @@ function injectPickerPanel(ctx) {
   }
 }
 function injectCascaderPanel(ctx) {
-  const { container, triggerTestId, attrName } = ctx;
+  const { container, triggerTestId, attrName, prefixCls: pfx } = ctx;
   const prefix = `${triggerTestId}-dropdown`;
   container.setAttribute(attrName, prefix);
-  const menus = container.querySelectorAll(".ant-cascader-menu");
+  const menus = container.querySelectorAll(cls(pfx, "cascader-menu"));
   menus.forEach((menu, i) => {
     const el = menu;
     el.setAttribute(attrName, `${prefix}-menu-${i}`);
-    const items = menu.querySelectorAll(".ant-cascader-menu-item");
+    const items = menu.querySelectorAll(cls(pfx, "cascader-menu-item"));
     items.forEach((item) => {
       const li = item;
       const text = li.textContent?.trim() || "";
@@ -162,28 +193,28 @@ function injectCascaderPanel(ctx) {
   });
 }
 function injectSelectPanel(ctx) {
-  const { container, triggerTestId, attrName } = ctx;
+  const { container, triggerTestId, attrName, prefixCls: pfx } = ctx;
   const prefix = `${triggerTestId}-dropdown`;
   container.setAttribute(attrName, prefix);
   const injectAllItems = () => {
-    const options = container.querySelectorAll(".ant-select-item-option");
+    const options = container.querySelectorAll(cls(pfx, "select-item-option"));
     options.forEach((opt) => {
       const el = opt;
       if (el.hasAttribute(attrName)) return;
       const title = el.getAttribute("title") || el.textContent?.trim() || "";
       el.setAttribute(attrName, `${prefix}-option-${title}`);
     });
-    const treeNodes = container.querySelectorAll(".ant-select-tree-treenode");
+    const treeNodes = container.querySelectorAll(cls(pfx, "select-tree-treenode"));
     treeNodes.forEach((node) => {
       const el = node;
-      const content = el.querySelector(".ant-select-tree-node-content-wrapper");
+      const content = el.querySelector(cls(pfx, "select-tree-node-content-wrapper"));
       const title = content?.getAttribute("title") || content?.textContent?.trim() || "";
       if (content && !content.hasAttribute(attrName)) {
         content.setAttribute(attrName, `${prefix}-tree-${title}`);
       }
-      const switcher = el.querySelector(".ant-select-tree-switcher");
+      const switcher = el.querySelector(cls(pfx, "select-tree-switcher"));
       if (switcher && !switcher.hasAttribute(attrName)) {
-        const isOpen = switcher.classList.contains("ant-select-tree-switcher_open");
+        const isOpen = switcher.classList.contains(`${pfx}-select-tree-switcher_open`);
         const state = isOpen ? "collapse" : "expand";
         switcher.setAttribute(attrName, `${prefix}-tree-switcher-${state}-${title}`);
       }
@@ -202,17 +233,19 @@ function injectSelectPanel(ctx) {
     observer.observe(container, { childList: true, subtree: true });
   }
 }
-var DEFAULT_PANELS = {
-  ".ant-picker-dropdown": injectPickerPanel,
-  ".ant-cascader-dropdown": injectCascaderPanel,
-  ".ant-select-dropdown": injectSelectPanel
-};
+function buildDefaultPanels(pfx) {
+  return {
+    [cls(pfx, "picker-dropdown")]: injectPickerPanel,
+    [cls(pfx, "cascader-dropdown")]: injectCascaderPanel,
+    [cls(pfx, "select-dropdown")]: injectSelectPanel
+  };
+}
 function getTriggerTestId(trigger, attrName) {
   if (!trigger) return "unknown-picker";
   return trigger.getAttribute(attrName) || trigger.getAttribute("id") || "unknown-picker";
 }
-function findActiveTrigger(attrName) {
-  const focused = document.querySelector(".ant-picker-focused");
+function findActiveTrigger(attrName, pfx) {
+  const focused = document.querySelector(cls(pfx, "picker-focused"));
   if (focused) {
     if (focused.hasAttribute(attrName)) return focused;
     const inner = focused.querySelector(`[${attrName}]`);
@@ -220,9 +253,9 @@ function findActiveTrigger(attrName) {
     const input = focused.querySelector("input[id]");
     if (input) return input;
   }
-  const selectOpen = document.querySelector(".ant-select-open");
+  const selectOpen = document.querySelector(cls(pfx, "select-open"));
   if (selectOpen?.hasAttribute(attrName)) return selectOpen;
-  const cascader = document.querySelector(".ant-cascader-picker-focused");
+  const cascader = document.querySelector(cls(pfx, "cascader-picker-focused"));
   if (cascader?.hasAttribute(attrName)) return cascader;
   const expanded = document.querySelector(
     `[${attrName}][aria-expanded="true"]`
@@ -230,11 +263,11 @@ function findActiveTrigger(attrName) {
   if (expanded) return expanded;
   return null;
 }
-function injectMenuItems(attrName, root = document) {
+function injectMenuItems(attrName, pfx, root = document) {
   const itemRules = [
-    { selector: ".ant-menu-item", prefix: "a-menu-item-" },
-    { selector: ".ant-menu-submenu", textSelector: ".ant-menu-submenu-title", prefix: "a-sub-menu-" },
-    { selector: ".ant-menu-item-group", textSelector: ".ant-menu-item-group-title", prefix: "a-menu-item-group-" }
+    { selector: cls(pfx, "menu-item"), prefix: "a-menu-item-" },
+    { selector: cls(pfx, "menu-submenu"), textSelector: cls(pfx, "menu-submenu-title"), prefix: "a-sub-menu-" },
+    { selector: cls(pfx, "menu-item-group"), textSelector: cls(pfx, "menu-item-group-title"), prefix: "a-menu-item-group-" }
   ];
   const injectSingle = (el, rule) => {
     if (el.hasAttribute(attrName)) return;
@@ -261,7 +294,9 @@ function injectMenuItems(attrName, root = document) {
 }
 function setupAntDropdownTestIds(opts = {}) {
   const attrName = opts.attributeName ?? "data-testid";
-  const panels = { ...DEFAULT_PANELS, ...opts.panels };
+  const pfx = resolvePrefixCls(opts.prefixCls);
+  const defaultPanels = buildDefaultPanels(pfx);
+  const panels = { ...defaultPanels, ...opts.panels };
   const observer = new MutationObserver((mutations) => {
     for (const mutation of mutations) {
       for (const node of mutation.addedNodes) {
@@ -273,21 +308,22 @@ function setupAntDropdownTestIds(opts = {}) {
             if (container.hasAttribute(attrName)) continue;
             if (strategy) {
               queueMicrotask(() => {
-                const trigger = findActiveTrigger(attrName);
+                const trigger = findActiveTrigger(attrName, pfx);
                 const triggerTestId = getTriggerTestId(trigger, attrName);
                 const ctx = {
                   container,
                   trigger,
                   triggerTestId,
-                  attrName
+                  attrName,
+                  prefixCls: pfx
                 };
                 strategy(ctx);
               });
             }
           }
         }
-        const uploadByTestId = node.matches(`.ant-upload[${attrName}]`) ? [node] : Array.from(node.querySelectorAll(`.ant-upload[${attrName}]`));
-        const uploadById = uploadByTestId.length === 0 ? node.matches(`.ant-upload[id]`) ? [node] : Array.from(node.querySelectorAll(`.ant-upload[id]`)) : [];
+        const uploadByTestId = node.matches(`${cls(pfx, "upload")}[${attrName}]`) ? [node] : Array.from(node.querySelectorAll(`${cls(pfx, "upload")}[${attrName}]`));
+        const uploadById = uploadByTestId.length === 0 ? node.matches(`${cls(pfx, "upload")}[id]`) ? [node] : Array.from(node.querySelectorAll(`${cls(pfx, "upload")}[id]`)) : [];
         const uploads = [...uploadByTestId, ...uploadById];
         for (const upload of uploads) {
           if (!(upload instanceof HTMLElement)) continue;
@@ -301,33 +337,35 @@ function setupAntDropdownTestIds(opts = {}) {
             }
           }
         }
-        injectMenuItems(attrName, node instanceof HTMLElement ? node : document);
+        injectMenuItems(attrName, pfx, node instanceof HTMLElement ? node : document);
       }
     }
   });
   observer.observe(document.body, { childList: true, subtree: true });
   requestAnimationFrame(() => {
-    injectMenuItems(attrName);
+    injectMenuItems(attrName, pfx);
   });
   return () => observer.disconnect();
 }
 function injectCurrentDropdowns(opts = {}) {
   const attrName = opts.attributeName ?? "data-testid";
-  const panels = { ...DEFAULT_PANELS, ...opts.panels };
+  const pfx = resolvePrefixCls(opts.prefixCls);
+  const defaultPanels = buildDefaultPanels(pfx);
+  const panels = { ...defaultPanels, ...opts.panels };
   for (const [selector, strategy] of Object.entries(panels)) {
     const containers = document.querySelectorAll(selector);
     containers.forEach((container) => {
       if (!(container instanceof HTMLElement)) return;
       if (container.hasAttribute(attrName)) return;
       if (!strategy) return;
-      const trigger = findActiveTrigger(attrName);
+      const trigger = findActiveTrigger(attrName, pfx);
       const triggerTestId = getTriggerTestId(trigger, attrName);
-      strategy({ container, trigger, triggerTestId, attrName });
+      strategy({ container, trigger, triggerTestId, attrName, prefixCls: pfx });
     });
   }
   const uploads = [
-    ...Array.from(document.querySelectorAll(`.ant-upload[${attrName}]`)),
-    ...Array.from(document.querySelectorAll(`.ant-upload[id]`))
+    ...Array.from(document.querySelectorAll(`${cls(pfx, "upload")}[${attrName}]`)),
+    ...Array.from(document.querySelectorAll(`${cls(pfx, "upload")}[id]`))
   ];
   const seen = /* @__PURE__ */ new Set();
   uploads.forEach((upload) => {
@@ -344,7 +382,7 @@ function injectCurrentDropdowns(opts = {}) {
       }
     }
   });
-  injectMenuItems(attrName);
+  injectMenuItems(attrName, pfx);
 }
 export {
   injectCurrentDropdowns,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vite-plugin-vue-testid",
-  "version": "1.0.7",
+  "version": "1.0.8",
   "description": "Vite plugin to auto-inject data-testid into Vue component templates for E2E testing",
   "type": "module",
   "main": "./dist/index.js",