fix(types): Fixed last context types

fb55 · Apr 16, 2021 · e15cc53 · e15cc53
1 parent 600a774
commit e15cc53
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -49,7 +49,7 @@ function wrapCompile<Selector, Node, ElementNode extends Node>(
     return function addAdapter(
         selector: Selector,
         options?: Options<Node, ElementNode>,
-        context?: Node[]
+        context?: Node[] | Node
     ) {
         const opts = convertOptionFormats(options);
 

diff --git a/src/types.ts b/src/types.ts
@@ -116,7 +116,7 @@ export interface Options<Node, ElementNode extends Node> {
      * The context of the current query. Used to limit the scope of searches.
      * Can be matched directly using the `:scope` pseudo-selector.
      */
-    context?: ElementNode | ElementNode[];
+    context?: Node | Node[];
     /**
      * Allow css-select to cache results for some selectors, sometimes greatly
      * improving querying performance. Disable this if your document can
@@ -145,5 +145,5 @@ export type Query<ElementNode> =
 export type CompileToken<Node, ElementNode extends Node> = (
     token: InternalSelector[][],
     options: InternalOptions<Node, ElementNode>,
-    context?: Node[]
+    context?: Node[] | Node
 ) => CompiledQuery<ElementNode>;
diff --git a/test/qwery.ts b/test/qwery.ts
@@ -41,6 +41,8 @@ const doc = parseDOM(
 
 const el = DomUtils.getElementById("attr-child-boosh", document);
 
+if (!el) throw new Error("Couldn't find element");
+
 // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
 const pseudos = DomUtils.getElementById("pseudos", document)!.children.filter(
     DomUtils.isTag