in FindAllBy and FindAll query helper types, spread Arguments to support no arguments

Author: dturcotte

types/__tests__/type-tests.ts

@@ -132,6 +132,50 @@ export async function testQueryHelpers() {
   screenWithCustomQueries.getByAutomationId(['id', 'automationId'])
   await screenWithCustomQueries.findAllByAutomationId('id', {}, {timeout: 1000})
   await screenWithCustomQueries.findByAutomationId('id', {}, {timeout: 1000})
+
+  // contrived example of a custom query with no arguments beyond container
+  function queryAllByFoo(container: HTMLElement) {
+    return queryAllByText(container, 'Foo')
+  }
+
+  const [
+    queryByTextFoo,
+    getAllByTextFoo,
+    getByTextFoo,
+    findAllByTextFoo,
+    findByTextFoo,
+  ] = buildQueries(
+    queryAllByFoo,
+    createIdRelatedErrorHandler(
+      `Found multiple elements with text Foo`,
+      'Multiple error',
+    ),
+    createIdRelatedErrorHandler(
+      `Unable to find an element with text Foo`,
+      'Missing error',
+    ),
+  )
+
+  queryByTextFoo(element)
+  getAllByTextFoo(element)
+  getByTextFoo(element)
+  await findAllByTextFoo(element)
+  await findByTextFoo(element)
+
+  const screenWithCustomFooQueries = within(document.body, {
+    ...queries,
+    queryByTextFoo,
+    getAllByTextFoo,
+    getByTextFoo,
+    findAllByTextFoo,
+    findByTextFoo,
+  })
+
+  screenWithCustomFooQueries.queryByTextFoo()
+  screenWithCustomFooQueries.getAllByTextFoo()
+  screenWithCustomFooQueries.getByTextFoo()
+  await screenWithCustomFooQueries.findAllByTextFoo()
+  await screenWithCustomFooQueries.findByTextFoo()
 }
 
 export function testBoundFunctions() {

types/query-helpers.d.ts

@@ -50,12 +50,12 @@ export type GetAllBy<Arguments extends any[]> = QueryMethod<
   HTMLElement[]
 >
 export type FindAllBy<Arguments extends any[]> = QueryMethod<
-  [Arguments[0], Arguments[1]?, waitForOptions?],
+  [...Arguments, waitForOptions?],
   Promise<HTMLElement[]>
 >
 export type GetBy<Arguments extends any[]> = QueryMethod<Arguments, HTMLElement>
 export type FindBy<Arguments extends any[]> = QueryMethod<
-  [Arguments[0], Arguments[1]?, waitForOptions?],
+  [...Arguments, waitForOptions?],
   Promise<HTMLElement>
 >