Add testcase - utils.doesElementContainRootFiberNode

test/utils.doesElementContainRootFiberNode.spec.tsx

@@ -0,0 +1,77 @@
+// import * as React from "react";
+
+// Import stuff from src
+import { doesElementContainRootFiberNode } from "../src/utils";
+
+// Import test helpers and sample components
+import { mountAndGetRootNode } from "./utils/mountInEnzyme";
+import { createFunctionComponent } from "./utils/createComponent";
+
+describe("traverse", () => {
+  let container: HTMLDivElement;
+
+  beforeEach(() => {
+    container = document.body.appendChild(document.createElement("div"));
+  });
+
+  afterEach(() => {
+    try {
+      document.body.removeChild(container);
+    } catch (error) {
+      if (!(error instanceof DOMException)) {
+        throw error;
+      }
+    }
+  });
+
+  describe("doesElementContainRootFiberNode", () => {
+    it("should work for direct element", () => {
+      const fn1 = createFunctionComponent("fn1");
+      const containerInner = container.appendChild(
+        document.createElement("span")
+      );
+      mountAndGetRootNode(fn1, containerInner);
+
+      const resultPositive = doesElementContainRootFiberNode(containerInner);
+
+      expect(resultPositive).not.toBeFalsy();
+      expect(resultPositive).toBe(true);
+    });
+
+    it("should not work for parent element", () => {
+      const fn1 = createFunctionComponent("fn1");
+      const containerInner = container.appendChild(
+        document.createElement("span")
+      );
+      mountAndGetRootNode(fn1, containerInner);
+
+      const resultNegative = doesElementContainRootFiberNode(container);
+
+      expect(resultNegative).toBeFalsy();
+      expect(resultNegative).toBe(false);
+    });
+
+    it("should not work for child element", () => {
+      const fn1 = createFunctionComponent("fn1");
+      const containerInner = container.appendChild(
+        document.createElement("span")
+      );
+      mountAndGetRootNode(fn1, container);
+
+      const resultNegative = doesElementContainRootFiberNode(containerInner);
+
+      expect(resultNegative).toBeFalsy();
+      expect(resultNegative).toBe(false);
+    });
+
+    it("should work for body", () => {
+      const fn1 = createFunctionComponent("fn1");
+      mountAndGetRootNode(fn1, document.body);
+
+      const resultPositive = doesElementContainRootFiberNode(document.body);
+
+      expect(resultPositive).not.toBeFalsy();
+      expect(resultPositive).toBe(true);
+    });
+  });
+});