From 5e93f9dab98752a3b2f492091cc9d7c5461c7730 Mon Sep 17 00:00:00 2001
From: bendtherules <abhasbhattacharya2@gmail.com>
Date: Sun, 21 Jul 2019 09:19:46 +0530
Subject: [PATCH] Add testcase - traverse - edge cases like rendering divs
 directly as root

---
 test/traverse-edge.spec.tsx | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 test/traverse-edge.spec.tsx

diff --git a/test/traverse-edge.spec.tsx b/test/traverse-edge.spec.tsx
new file mode 100644
index 0000000..a4a9dce
--- /dev/null
+++ b/test/traverse-edge.spec.tsx
@@ -0,0 +1,32 @@
+// import * as React from "react";
+
+// Import stuff from src
+import { traverse } from "../src";
+
+// Import test helpers and sample components
+import { mountAndGetRootNode } from "./utils/mount-in-enzyme";
+
+describe("traverse", () => {
+  let container: HTMLDivElement;
+
+  beforeEach(() => {
+    container = document.body.appendChild(document.createElement("div"));
+  });
+
+  afterEach(() => {
+    document.body.removeChild(container);
+  });
+
+  describe("edge cases", () => {
+    it("should work for string nodes", () => {
+      const rootNode = mountAndGetRootNode('div', container);
+      const mockCallback = jest.fn();
+
+      traverse(rootNode, mockCallback);
+
+      // Can't be zero, in any case
+      expect(mockCallback.mock.calls.length).not.toBe(0);
+      expect(mockCallback.mock.calls.length).toBe(1);
+    });
+  });
+});