fix: handle count > 1 for within

Author: Ben Monro

src/index.js

@@ -67,6 +67,10 @@ export const within = async selector => {
   }
 
   if (selector.constructor.name === SELECTOR_TYPE) {
+    const count = await selector.count;
+    if (count > 1) {
+      throw new Error(`within() requires a single element, found ${count}`);
+    }
     const el = await selector;
     const withinQueryName = el.getAttribute('data-tctl-queryname');
 

tests/testcafe/within.js

@@ -70,3 +70,20 @@ test('works with nested selector from "All" query with index - function', async
 
   await t.expect(nested.getByText('Button Text').exists).ok();
 });
+
+
+test('should throw error if count > 1', async t => {
+  const nestedDivs = getAllByTestId(/nested/);
+
+  await t.expect(nestedDivs.count).eql(2);
+  let didThrow = false;
+  try {
+    await within(nestedDivs);
+  } catch (e) {
+    didThrow = true;
+  }
+  await t.expect(didThrow).ok();
+});
+
+
+