component-tree.ts

import type { TestInstance, TestNode } from 'test-renderer';

import { screen } from '../screen';

/**
 * Checks if the given element is a host element.
 * @param node The element to check.
 */
export function isTestInstance(node?: TestNode | null): node is TestInstance {
  return typeof node !== 'string' && typeof node?.type === 'string';
}

export function isElementMounted(element: TestInstance) {
  return getContainerElement(element) === screen.container;
}

/**
 * Returns host siblings for given element.
 * @param instance The element start traversing from.
 */
export function getInstanceSiblings(instance: TestInstance): TestInstance[] {
  // Should not happen
  const parent = instance.parent;
  if (!parent) {
    return [];
  }

  return parent.children.filter(
    (sibling) => typeof sibling !== 'string' && sibling !== instance,
  ) as TestInstance[];
}

/**
 * Returns the container element of the tree.
 *
 * @param instance The element start traversing from.
 * @returns The container element of the tree.
 */
export function getContainerElement(instance: TestInstance) {
  let current = instance;
  while (current.parent) {
    current = current.parent;
  }

  return current;
}