fix jest unit test

elastic · Mar 23, 2021 · 45597a9 · 45597a9
1 parent 213ef13
commit 45597a9
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 47 deletions.
diff --git a/x-pack/plugins/lists/public/exceptions/components/builder/and_badge.test.tsx b/x-pack/plugins/lists/public/exceptions/components/builder/and_badge.test.tsx
@@ -6,48 +6,47 @@
  */
 
 import React from 'react';
-import { EuiFlexItem } from '@elastic/eui';
-import styled from 'styled-components';
-
-import { AndOrBadge } from '../and_or_badge';
-
-const MyInvisibleAndBadge = styled(EuiFlexItem)`
-  visibility: hidden;
-`;
-
-const MyFirstRowContainer = styled(EuiFlexItem)`
-  padding-top: 20px;
-`;
-
-interface BuilderAndBadgeProps {
-  entriesLength: number;
-  exceptionItemIndex: number;
-}
-
-export const BuilderAndBadgeComponent = React.memo<BuilderAndBadgeProps>(
-  ({ entriesLength, exceptionItemIndex }) => {
-    const badge = <AndOrBadge includeAntennas type="and" />;
-
-    if (entriesLength > 1 && exceptionItemIndex === 0) {
-      return (
-        <MyFirstRowContainer grow={false} data-test-subj="exceptionItemEntryFirstRowAndBadge">
-          {badge}
-        </MyFirstRowContainer>
-      );
-    } else if (entriesLength <= 1) {
-      return (
-        <MyInvisibleAndBadge grow={false} data-test-subj="exceptionItemEntryInvisibleAndBadge">
-          {badge}
-        </MyInvisibleAndBadge>
-      );
-    } else {
-      return (
-        <EuiFlexItem grow={false} data-test-subj="exceptionItemEntryAndBadge">
-          {badge}
-        </EuiFlexItem>
-      );
-    }
-  }
-);
-
-BuilderAndBadgeComponent.displayName = 'BuilderAndBadge';
+import { ThemeProvider } from 'styled-components';
+import { mount } from 'enzyme';
+
+import { getMockTheme } from '../../../common/test_utils/kibana_react.mock';
+
+import { BuilderAndBadgeComponent } from './and_badge';
+
+const mockTheme = getMockTheme({ eui: { euiColorLightShade: '#ece' } });
+
+describe('BuilderAndBadgeComponent', () => {
+  test('it renders exceptionItemEntryFirstRowAndBadge for very first exception item in builder', () => {
+    const wrapper = mount(
+      <ThemeProvider theme={mockTheme}>
+        <BuilderAndBadgeComponent entriesLength={2} exceptionItemIndex={0} />
+      </ThemeProvider>
+    );
+
+    expect(
+      wrapper.find('[data-test-subj="exceptionItemEntryFirstRowAndBadge"]').exists()
+    ).toBeTruthy();
+  });
+
+  test('it renders exceptionItemEntryInvisibleAndBadge if "entriesLength" is 1 or less', () => {
+    const wrapper = mount(
+      <ThemeProvider theme={mockTheme}>
+        <BuilderAndBadgeComponent entriesLength={1} exceptionItemIndex={0} />
+      </ThemeProvider>
+    );
+
+    expect(
+      wrapper.find('[data-test-subj="exceptionItemEntryInvisibleAndBadge"]').exists()
+    ).toBeTruthy();
+  });
+
+  test('it renders regular "and" badge if exception item is not the first one and includes more than one entry', () => {
+    const wrapper = mount(
+      <ThemeProvider theme={mockTheme}>
+        <BuilderAndBadgeComponent entriesLength={2} exceptionItemIndex={1} />
+      </ThemeProvider>
+    );
+
+    expect(wrapper.find('[data-test-subj="exceptionItemEntryAndBadge"]').exists()).toBeTruthy();
+  });
+});
diff --git a/x-pack/plugins/lists/public/exceptions/components/builder/exception_item_renderer.test.tsx b/x-pack/plugins/lists/public/exceptions/components/builder/exception_item_renderer.test.tsx
@@ -27,8 +27,6 @@ const mockTheme = getMockTheme({
 const mockKibanaHttpService = coreMock.createStart().http;
 const { autocomplete: autocompleteStartMock } = dataPluginMock.createStartContract();
 
-jest.mock('../../../../common/lib/kibana');
-
 describe('BuilderExceptionListItemComponent', () => {
   const getValueSuggestionsMock = jest.fn().mockResolvedValue(['value 1', 'value 2']);