From 945bcb194e64228bd7ab7c63f9f0dee6b5ff9b5f Mon Sep 17 00:00:00 2001
From: chenos <chenlinxh@gmail.com>
Date: Thu, 21 Sep 2023 14:54:27 +0800
Subject: [PATCH] feat: add testid (#2434)

Co-authored-by: Rain <958414905@qq.com>
---
 packages/core/client/src/pm/PluginManagerLink.tsx |  1 +
 .../src/schema-component/antd/action/Action.tsx   |  1 +
 .../antd/block-item/BlockItem.tsx                 | 15 +++++++++++++++
 .../src/schema-component/antd/menu/Menu.tsx       |  3 ++-
 .../schema-component/core/DesignableSwitch.tsx    |  1 +
 .../components/DuplicateAction.tsx                |  1 +
 6 files changed, 21 insertions(+), 1 deletion(-)

diff --git a/packages/core/client/src/pm/PluginManagerLink.tsx b/packages/core/client/src/pm/PluginManagerLink.tsx
index 5cffc26847..de13cdd336 100644
--- a/packages/core/client/src/pm/PluginManagerLink.tsx
+++ b/packages/core/client/src/pm/PluginManagerLink.tsx
@@ -14,6 +14,7 @@ export const PluginManagerLink = () => {
   return (
     <Tooltip title={t('Plugin manager')}>
       <Button
+        data-testid={'pm-button'}
         icon={<ApiOutlined />}
         title={t('Plugin manager')}
         onClick={() => {
diff --git a/packages/core/client/src/schema-component/antd/action/Action.tsx b/packages/core/client/src/schema-component/antd/action/Action.tsx
index 1f75d11c65..bac2b877cf 100644
--- a/packages/core/client/src/schema-component/antd/action/Action.tsx
+++ b/packages/core/client/src/schema-component/antd/action/Action.tsx
@@ -77,6 +77,7 @@ export const Action: ComposedAction = observer(
       }
       return (
         <SortableItem
+          data-testid={`${fieldSchema['x-action'] || fieldSchema.name}-action`}
           {...others}
           loading={field?.data?.loading}
           icon={icon ? <Icon type={icon} /> : null}
diff --git a/packages/core/client/src/schema-component/antd/block-item/BlockItem.tsx b/packages/core/client/src/schema-component/antd/block-item/BlockItem.tsx
index 7cf28d052c..2e989604a7 100644
--- a/packages/core/client/src/schema-component/antd/block-item/BlockItem.tsx
+++ b/packages/core/client/src/schema-component/antd/block-item/BlockItem.tsx
@@ -1,14 +1,29 @@
 import { css } from '@emotion/css';
+import { useFieldSchema } from '@formily/react';
 import cls from 'classnames';
 import React from 'react';
 import { SortableItem } from '../../common';
 import { useDesigner, useProps } from '../../hooks';
 
+const getTestId = (schema) => {
+  const resource = schema?.['x-decorator-props']?.['resource'];
+  if (resource) {
+    return `${resource}-resource`;
+  }
+  const field = schema['x-collection-field'];
+  if (field) {
+    return `${field}-field`;
+  }
+  return `${schema.name}-item`;
+};
+
 export const BlockItem: React.FC<any> = (props) => {
   const { className, children } = useProps(props);
   const Designer = useDesigner();
+  const schema = useFieldSchema();
   return (
     <SortableItem
+      data-testid={getTestId(schema)}
       className={cls(
         'nb-block-item',
         className,
diff --git a/packages/core/client/src/schema-component/antd/menu/Menu.tsx b/packages/core/client/src/schema-component/antd/menu/Menu.tsx
index 1b12841db9..1e8d7fb4ff 100644
--- a/packages/core/client/src/schema-component/antd/menu/Menu.tsx
+++ b/packages/core/client/src/schema-component/antd/menu/Menu.tsx
@@ -196,7 +196,7 @@ const HeaderMenu = ({
       key: 'x-designer-button',
       disabled: true,
       style: { padding: '0 8px', order: 9999 },
-      label: render({ style: { background: 'none' } }),
+      label: render({ 'data-testid': 'add-menu-item-button-in-header', style: { background: 'none' } }),
       notdelete: true,
     };
     const result = getMenuItems(() => {
@@ -281,6 +281,7 @@ const SideMenu = ({
         key: 'x-designer-button',
         disabled: true,
         label: render({
+          'data-testid': 'add-menu-item-button-in-side',
           insert: (s) => {
             const dn = createDesignable({
               t,
diff --git a/packages/core/client/src/schema-component/core/DesignableSwitch.tsx b/packages/core/client/src/schema-component/core/DesignableSwitch.tsx
index 026af6e538..4ca990f25c 100644
--- a/packages/core/client/src/schema-component/core/DesignableSwitch.tsx
+++ b/packages/core/client/src/schema-component/core/DesignableSwitch.tsx
@@ -19,6 +19,7 @@ export const DesignableSwitch = () => {
   return (
     <Tooltip title={t('UI Editor')}>
       <Button
+        data-testid={'ui-editor-button'}
         // selected={designable}
         icon={<HighlightOutlined />}
         title={t('UI Editor')}
diff --git a/packages/core/client/src/schema-initializer/components/DuplicateAction.tsx b/packages/core/client/src/schema-initializer/components/DuplicateAction.tsx
index 275c87423f..915e928270 100644
--- a/packages/core/client/src/schema-initializer/components/DuplicateAction.tsx
+++ b/packages/core/client/src/schema-initializer/components/DuplicateAction.tsx
@@ -123,6 +123,7 @@ export const DuplicateAction = observer((props: any) => {
             </a>
           ) : (
             <Button
+              data-testid={'duplicate-action'}
               disabled={disabled}
               style={{
                 opacity: designable && field?.data?.hidden && 0.1,