remove getHostChildren

Author: mdjastrzebski

src/helpers/__tests__/component-tree.test.tsx

@@ -2,13 +2,7 @@ import React from 'react';
 import { Text, TextInput, View } from 'react-native';
 
 import { render, screen } from '../..';
-import {
-  getHostChildren,
-  getHostParent,
-  getHostSelves,
-  getHostSiblings,
-  getContainerElement,
-} from '../component-tree';
+import { getHostSelves, getHostSiblings, getContainerElement } from '../component-tree';
 
 function ZeroHostChildren() {
   return <></>;
@@ -24,49 +18,6 @@ function MultipleHostChildren() {
   );
 }
 
-describe('getHostChildren()', () => {
-  it('returns host children for host component', () => {
-    render(
-      <View testID="grandparent">
-        <View testID="parent">
-          <View testID="subject" />
-          <Text testID="sibling">Hello</Text>
-        </View>
-      </View>,
-    );
-
-    const hostSubject = screen.getByTestId('subject');
-    expect(getHostChildren(hostSubject)).toEqual([]);
-
-    const hostSibling = screen.getByTestId('sibling');
-    expect(getHostChildren(hostSibling)).toEqual([]);
-
-    const hostParent = screen.getByTestId('parent');
-    expect(getHostChildren(hostParent)).toEqual([hostSubject, hostSibling]);
-
-    const hostGrandparent = screen.getByTestId('grandparent');
-    expect(getHostChildren(hostGrandparent)).toEqual([hostParent]);
-  });
-
-  it('returns host children for composite component', () => {
-    render(
-      <View testID="parent">
-        <MultipleHostChildren />
-        <View testID="subject" />
-        <View testID="sibling" />
-      </View>,
-    );
-
-    expect(getHostChildren(screen.getByTestId('parent'))).toEqual([
-      screen.getByTestId('child1'),
-      screen.getByTestId('child2'),
-      screen.getByTestId('child3'),
-      screen.getByTestId('subject'),
-      screen.getByTestId('sibling'),
-    ]);
-  });
-});
-
 describe('getHostSelves()', () => {
   it('returns passed element for host components', () => {
     render(

src/helpers/component-tree.ts

@@ -14,32 +14,6 @@ export function isElementMounted(element: HostElement) {
   return getContainerElement(element) === screen.container;
 }
 
-/**
- * Returns host children for given element.
- * @param element The element start traversing from.
- */
-export function getHostChildren(element: HostElement | null): HostElement[] {
-  if (element == null) {
-    return [];
-  }
-
-  const hostChildren: HostElement[] = [];
-
-  element.children.forEach((child) => {
-    if (typeof child !== 'object') {
-      return;
-    }
-
-    if (isHostElement(child)) {
-      hostChildren.push(child);
-    } else {
-      hostChildren.push(...getHostChildren(child));
-    }
-  });
-
-  return hostChildren;
-}
-
 /**
  * Return the array of host elements that represent the passed element.
  *

src/matchers/to-be-empty-element.ts

@@ -1,24 +1,25 @@
-import type { HostElement } from 'universal-test-renderer';
 import { matcherHint, RECEIVED_COLOR } from 'jest-matcher-utils';
 import redent from 'redent';
+import type { HostElement } from 'universal-test-renderer';
 
-import { getHostChildren } from '../helpers/component-tree';
+import { isHostElement } from '../helpers/component-tree';
 import { formatElementList } from '../helpers/format-element';
 import { checkHostElement } from './utils';
 
 export function toBeEmptyElement(this: jest.MatcherContext, element: HostElement) {
   checkHostElement(element, toBeEmptyElement, this);
 
-  const hostChildren = getHostChildren(element);
+  // TODO check
+  const children = element.children.filter((child) => isHostElement(child));
 
   return {
-    pass: hostChildren.length === 0,
+    pass: children.length === 0,
     message: () => {
       return [
         matcherHint(`${this.isNot ? '.not' : ''}.toBeEmptyElement`, 'element', ''),
         '',
         'Received:',
-        `${RECEIVED_COLOR(redent(formatElementList(hostChildren), 2))}`,
+        `${RECEIVED_COLOR(redent(formatElementList(children), 2))}`,
       ].join('\n');
     },
   };