chore: mdxish cleanup

__tests__/helpers.ts

@@ -1,8 +1,54 @@
+import type { Element, Root, RootContent } from 'hast';
+
 import * as rdmd from '@readme/markdown-legacy';
 
 import { vi } from 'vitest';
 
 import { run, compile, migrate as baseMigrate, mdastV6 } from '../index';
+import { type RMDXModule } from '../types';
+
+export function findElementByTagName(node: Root | RootContent, tagName: string): Element | null {
+  if ('type' in node && node.type === 'element' && 'tagName' in node && node.tagName === tagName) {
+    return node;
+  }
+
+  if ('children' in node && Array.isArray(node.children)) {
+    return node.children.reduce<Element | null>((found, child) => {
+      if (found) return found;
+      return findElementByTagName(child, tagName);
+    }, null);
+  }
+
+  return null;
+}
+
+export function findElementsByTagName(node: Root | RootContent, tagName: string): Element[] {
+  const results: Element[] = [];
+
+  if ('type' in node && node.type === 'element' && 'tagName' in node && node.tagName === tagName) {
+    results.push(node);
+  }
+
+  if ('children' in node && Array.isArray(node.children)) {
+    node.children.forEach(child => {
+      results.push(...findElementsByTagName(child, tagName));
+    });
+  }
+
+  return results;
+}
+
+export const stubModule = {
+  default: () => null,
+  Toc: null,
+  toc: [],
+} as unknown as RMDXModule;
+
+export const makeComponents = (...names: string[]) =>
+  names.reduce<Record<string, RMDXModule>>((acc, name) => {
+    acc[name] = stubModule;
+    return acc;
+  }, {});
 
 export const silenceConsole =
   (prop: keyof Console = 'error', impl = () => {}) =>

__tests__/lib/mdxish/gemoji.test.ts → __tests__/lib/gemoji.test.ts

@@ -1,4 +1,4 @@
-import { mix } from '../../../lib';
+import { mix } from '../../lib';
 
 describe('gemoji transformer', () => {
   it('should transform shortcodes back to emojis', () => {

__tests__/lib/mdxish/legacy-variables.test.ts

@@ -2,22 +2,7 @@ import type { CustomComponents } from '../../../types';
 import type { Element, Text } from 'hast';
 
 import { mdxish } from '../../../lib';
-
-function findElementByTagName(tree: Element, tagName: string): Element | null {
-  if (tree.tagName === tagName) {
-    return tree;
-  }
-  // Recursively search children
-  let result: Element | null = null;
-  tree.children.some(child => {
-    if ('tagName' in child) {
-      result = findElementByTagName(child as Element, tagName);
-      return result !== null;
-    }
-    return false;
-  });
-  return result;
-}
+import { findElementByTagName } from '../../helpers';
 
 describe('legacy variables resolution', () => {
   describe('basic resolution', () => {