fix: 🐛 simplify AfterDraf component

Author: streamich

src/AfterDraf/__tests__/createSingleRunDraf.server.test.tsx

@@ -1,3 +1,16 @@
 /** @jest-environment node */
+import {createElement as h} from 'react';
+import ReactDOMServer from 'react-dom/server';
+import {AfterDraf} from '..';
 
-require('./index.test');
+const sleep = (time) => new Promise((resolve) => setTimeout(resolve, time));
+
+describe('<AfterDraf>', () => {
+  it('Renders children', async () => {
+    const str = ReactDOMServer.renderToStaticMarkup(<AfterDraf>
+        <div>foobar</div>
+      </AfterDraf>);
+
+    expect(str).toBe('<div>foobar</div>');
+  });
+});

src/AfterDraf/__tests__/createSingleRunDraf.test.tsx

@@ -1,8 +1,7 @@
 import {createElement as h} from 'react';
-import {shallow} from 'enzyme';
+import {render} from 'react-dom';
 import {AfterDraf} from '..';
 
-const isClient = typeof window === 'object';
 const sleep = (time) => new Promise((resolve) => setTimeout(resolve, time));
 
 describe('<AfterDraf>', () => {
@@ -11,50 +10,20 @@ describe('<AfterDraf>', () => {
   });
 
   it('waits for DRAF on client before rendering', async () => {
-    const wrapper = shallow(
-      <AfterDraf>
-        <div>foobar</div>
-      </AfterDraf>
-    );
-
-    if (isClient) {
-      expect(wrapper.html()).toBe(null);
-    } else {
-      expect(wrapper.html()).toBe('<div>foobar</div>');
-    }
-
-    await sleep(100);
-
-    wrapper.update();
+    const div = document.createElement('div');
 
-    expect(wrapper.html()).toBe('<div>foobar</div>');
-  });
-
-  it('waits for DRAF every mount', async () => {
-    shallow(
-      <AfterDraf>
+    document.body.appendChild(div);
+    render(<AfterDraf>
         <div>foobar</div>
-      </AfterDraf>
-    );
-
-    await sleep(100);
-
-    const wrapper2 = shallow(
-      <AfterDraf>
-        <div>bazooka</div>
-      </AfterDraf>
-    );
+      </AfterDraf>,
+    div);
 
-    if (isClient) {
-      expect(wrapper2.html()).toBe(null);
-    } else {
-      expect(wrapper2.html()).toBe('<div>bazooka</div>');
-    }
+    expect(div.innerHTML).toBe('');
 
     await sleep(100);
 
-    wrapper2.update();
+    expect(div.innerHTML).toBe('<div>foobar</div>');
 
-    expect(wrapper2.html()).toBe('<div>bazooka</div>');
+    document.body.removeChild(div);
   });
 });