feat: support for within across page loads/reloads (#32)

* feat: support for within across page loads

* lint

Author: benmonro

src/index.js

@@ -31,25 +31,18 @@ Object.keys(queries).forEach(queryName => {
   )
 })
 
-export const within = async sel => {
-  const sanitizedSel = sel.replace(/"/g, "'")
-  await ClientFunction(
-    new Function(
-      ` 
-
-      window.TestcafeTestingLibrary = window.TestcafeTestingLibrary || {}
-      const elem = document.querySelector("${sanitizedSel}");
-      window.TestcafeTestingLibrary["within_${sanitizedSel}"] = TestingLibraryDom.within(elem);
+export const within = selector => {
+  const sanitizedSelector = selector.replace(/"/g, "'")
 
-    `,
-    ),
-  )()
   const container = {}
 
   Object.keys(queries).forEach(queryName => {
     container[queryName] = Selector(
       new Function(
-        `return window.TestcafeTestingLibrary["within_${sanitizedSel}"].${queryName}(...arguments)`,
+        `
+        window.TestcafeTestingLibrary = window.TestcafeTestingLibrary || {}
+        window.TestcafeTestingLibrary["within_${sanitizedSelector}"] = window.TestcafeTestingLibrary["within_${sanitizedSelector}"] || TestingLibraryDom.within(document.querySelector("${sanitizedSelector}"))
+        return window.TestcafeTestingLibrary["within_${sanitizedSelector}"].${queryName}(...arguments)`,
       ),
     )
   })

tests/testcafe/within.js

@@ -26,4 +26,12 @@ test('quotes in selector', async t => {
     .click(getByText('Button Text'))
     .expect(Selector('button').withExactText('Button Clicked').exists)
     .ok()
+});
+
+test('still works after browser page reload', async t => {
+  const nested = await within('#nested');
+  await t.expect(nested.getByText('Button Text').exists).ok()
+
+  await t.eval(() => location.reload(true));
+  await t.expect(nested.getByText('Button Text').exists).ok()
 })