feat(queries): allow overriding queries in getQueriesForElement

src/__tests__/get-queries-for-element.js

@@ -0,0 +1,43 @@
+import {getQueriesForElement} from '../get-queries-for-element'
+import {queries} from '..'
+
+test('uses default queries', () => {
+  const container = document.createElement('div')
+  const boundQueries = getQueriesForElement(container)
+  expect(Object.keys(boundQueries)).toEqual(Object.keys(queries))
+})
+
+test('accepts custom queries', () => {
+  const container = document.createElement('div')
+  const customQuery = jest.fn()
+  const boundQueries = getQueriesForElement(container, {
+    customQuery,
+  })
+  expect(boundQueries.customQuery).toBeDefined()
+})
+
+test('binds functions to container', () => {
+  const container = document.createElement('div')
+  const mock = jest.fn()
+  function customQuery(element) {
+    return mock(element)
+  }
+  const boundQueries = getQueriesForElement(container, {
+    customQuery,
+  })
+  boundQueries.customQuery()
+  expect(mock).toHaveBeenCalledWith(container)
+})
+
+test('accepts an array of queries', () => {
+  const container = document.createElement('div')
+  const customQuery = jest.fn()
+  const boundQueries = getQueriesForElement(container, [
+    queries,
+    {
+      customQuery,
+    },
+  ])
+  expect(boundQueries.getAllByText).toBeDefined()
+  expect(boundQueries.customQuery).toBeDefined()
+})

src/get-queries-for-element.js

@@ -1,7 +1,22 @@
-import * as queries from './queries'
+import * as defaultQueries from './queries'
 
-function getQueriesForElement(element) {
-  return Object.entries(queries).reduce((helpers, [key, fn]) => {
+/**
+ * @typedef {{[key: string]: Function}} FuncMap
+ */
+
+/**
+ * @param {HTMLElement} element container
+ * @param {FuncMap|FuncMap[]} queries object of functions, or array of objects
+ * @returns {FuncMap} returns object of functions bound to container
+ */
+function getQueriesForElement(element, queries = defaultQueries) {
+  let flattenedQueries
+  if (Array.isArray(queries)) {
+    flattenedQueries = Object.assign({}, ...queries)
+  } else {
+    flattenedQueries = queries
+  }
+  return Object.entries(flattenedQueries).reduce((helpers, [key, fn]) => {
     helpers[key] = fn.bind(null, element)
     return helpers
   }, {})

typings/get-queries-for-element.d.ts

@@ -12,4 +12,7 @@ export type BoundFunctions<T> = {[P in keyof T]: BoundFunction<T[P]>}
 
 export function getQueriesForElement(
   element: HTMLElement,
+  queriesToBind:
+    | BoundFunctions<typeof queries>
+    | BoundFunctions<typeof queries>[],
 ): BoundFunctions<typeof queries>