Updated testing

Signed-off-by: Dan Dong <[email protected]>

src/plugins/vis_builder/public/application/components/data_tab/field_bucket.test.tsx

@@ -0,0 +1,77 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import React from 'react';
+// @ts-ignore
+import { mountWithIntl } from 'test_utils/enzyme_helpers';
+import { IndexPatternField } from '../../../../../data/common';
+// @ts-ignore
+import { findTestSubject } from '@elastic/eui/lib/test';
+import { FieldBucket } from './field_bucket';
+import { Bucket } from './types';
+import { EuiSmallButtonIcon } from '@elastic/eui';
+
+const mockUseIndexPatterns = jest.fn(() => ({ selected: 'mockIndexPattern' }));
+const mockUseOnAddFilter = jest.fn();
+jest.mock('../../utils/use', () => ({
+  useIndexPatterns: jest.fn(() => mockUseIndexPatterns),
+  useOnAddFilter: jest.fn(() => mockUseOnAddFilter),
+}));
+
+describe('visBuilder field bucket', function () {
+  function mountComponent(field: IndexPatternField, bucket: Bucket) {
+    const compProps = { field, bucket };
+    return mountWithIntl(<FieldBucket {...compProps} />);
+  }
+
+  it('should render with buttons if field is filterable', async () => {
+    // const props = {
+    //   key: 'bucket1',
+    //   bucket: {
+    //     count: 100,
+    //     display: 'display',
+    //     percent: 5,
+    //     value: '12',
+    //   },
+    //   field: new IndexPatternField(
+    //     {
+    //       name: 'bytes',
+    //       type: 'number',
+    //       esTypes: ['long'],
+    //       count: 10,
+    //       scripted: false,
+    //       searchable: true,
+    //       aggregatable: true,
+    //       readFromDocValues: true,
+    //     },
+    //     'bytes'
+    //   ),
+    // };
+    // mountWithIntl(<FieldBucket {...props} />);
+    const field = new IndexPatternField(
+      {
+        name: 'bytes',
+        type: 'number',
+        esTypes: ['long'],
+        count: 10,
+        scripted: false,
+        searchable: true,
+        aggregatable: true,
+        readFromDocValues: true,
+      },
+      'bytes'
+    );
+    const bucket = {
+      display: `display`,
+      value: `value`,
+      percent: 25,
+      count: 100,
+    };
+    const comp = mountComponent(field, bucket);
+    expect(findTestSubject(comp, 'plus-bytes-value').length).toBe(1);
+    expect(findTestSubject(comp, 'minus-bytes-value').length).toBe(1);
+    expect(comp.find(EuiSmallButtonIcon)).toMatchSnapshot();
+  });
+});