Extract hook logic to useTestRunHooks

This may be used by consumers of <TestRunHooks /> to eg. conditionally
render a heading.

Author: badeball

src/components/app/TestRunHooks.tsx

@@ -1,25 +1,20 @@
-import { TestRunHookFinished } from '@cucumber/messages'
 import React, { FC } from 'react'
 
-import { useQueries } from '../../hooks/useQueries.js'
+import { useTestRunHooks } from '../../hooks/useTestRunHooks.js'
 import { TestRunHookOutcome } from '../results/TestRunHookOutcome.js'
 import styles from './TestRunHooks.module.scss'
 
 export const TestRunHooks: FC = () => {
-  const { cucumberQuery } = useQueries()
-  const testRunHooksFinished: ReadonlyArray<TestRunHookFinished> =
-    cucumberQuery.findAllTestRunHookFinished()
+  const hooks = useTestRunHooks()
+
   return (
     <ol aria-label="RunHooks" className={styles.hooks}>
-      {testRunHooksFinished.map((testRunHookFinished) => {
-        const testRunHook = cucumberQuery.findHookBy(testRunHookFinished)
-
+      {hooks.map(({ testRunHookFinished, hook }) => {
         return (
           <TestRunHookOutcome
+            key={hook.id}
             // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
-            key={testRunHook!.id}
-            // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
-            hook={testRunHook!}
+            hook={hook!}
             testRunHookFinished={testRunHookFinished}
           />
         )

src/hooks/helpers.ts

@@ -0,0 +1,6 @@
+export function ensure<T>(value: T | undefined, message: string): T {
+  if (!value) {
+    throw new Error(message)
+  }
+  return value
+}

src/hooks/useTestRunHooks.ts

@@ -0,0 +1,24 @@
+import { Hook, TestRunHookFinished } from '@cucumber/messages'
+
+import { ensure } from './helpers.js'
+import { useQueries } from './useQueries.js'
+
+type RunHooksList = { testRunHookFinished: TestRunHookFinished; hook: Hook }[]
+
+export function useTestRunHooks(): RunHooksList {
+  const { cucumberQuery } = useQueries()
+  const testRunHooksFinished: ReadonlyArray<TestRunHookFinished> =
+    cucumberQuery.findAllTestRunHookFinished()
+
+  return testRunHooksFinished.map((testRunHookFinished) => {
+    const hook = ensure(
+      cucumberQuery.findHookBy(testRunHookFinished),
+      'Expected testRunHookFinished to resolve with a hook'
+    )
+
+    return {
+      testRunHookFinished,
+      hook,
+    }
+  })
+}