Add generic to ElementAssertion

stackbuilders · Jun 20, 2024 · c2da004 · c2da004
1 parent 5a276d8
commit c2da004
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/packages/dom/src/lib/ElementAssertion.ts b/packages/dom/src/lib/ElementAssertion.ts
@@ -1,8 +1,8 @@
 import { Assertion, AssertionError } from "@assertive-ts/core";
 
-export class ElementAssertion extends Assertion<Element> {
+export class ElementAssertion<T extends Element> extends Assertion<T> {
 
-  public constructor(actual: Element) {
+  public constructor(actual: T) {
     super(actual);
   }
 

diff --git a/packages/dom/test/unit/lib/ElementAssertion.test.tsx b/packages/dom/test/unit/lib/ElementAssertion.test.tsx
@@ -30,14 +30,11 @@ describe("[Unit] ElementAssertion.test.ts", () => {
       });
     });
     context("when the element is not in the document", () => {
-      it("returns the assertion instance", () => {
+      it("Throws an assertion error", () => {
         const detachedElement = document.createElement("div");
 
         const test = new ElementAssertion(detachedElement);
 
-        expect(test.not.toBeInTheDocument())
-          .toBeEqual(test);
-
         expect(() => test.toBeInTheDocument())
           .toThrowError(AssertionError)
           .toHaveMessage("Expected the element to be in the document");