fixing viewafterlist without a key

Author: Igor Canedo

sample/__tests__/getBlocks.test.js

@@ -59,7 +59,16 @@ describe('return specific component based on type', () => {
     const bodyData = { blocks: [{ type: 'atomic' }] };
     const atomicHandler = item => item;
     const result = getBlocks({ contentState: bodyData, atomicHandler });
-    expect(result[0][1].type).toBe('atomic');
+    expect(result[0].type).toBe('atomic');
+  });
+
+  it('atomicHandler function when type atomic between lists', () => {
+    const bodyData = { blocks: [
+      { type: 'ordered-list-item' }, { type: 'atomic' }, { type: 'ordered-list-item' },
+    ] };
+    const atomicHandler = item => item;
+    const result = getBlocks({ contentState: bodyData, atomicHandler });
+    expect(result[1][1].type).toBe('atomic');
   });
 
   it('array of null when type is invalid', () => {

src/getBlocks.js

@@ -65,7 +65,7 @@ const getBlocks = (params: ParamsType): ?Array<*> => {
     if (myCounter.count >= 0) {
       if (myCounter.count > 0) {
         myCounter.count = 0;
-        return <ViewAfterList />;
+        return <ViewAfterList key={generateKey()} />;
       }
       return null;
     }
@@ -74,7 +74,7 @@ const getBlocks = (params: ParamsType): ?Array<*> => {
     if (myCounter['unordered-list-item'].count > 0 || myCounter['ordered-list-item'].count > 0) {
       myCounter['unordered-list-item'].count = 0;
       myCounter['ordered-list-item'].count = 0;
-      return <ViewAfterList />;
+      return <ViewAfterList key={generateKey()} />;
     }
 
     return null;
@@ -115,10 +115,14 @@ const getBlocks = (params: ParamsType): ?Array<*> => {
         }
 
         case 'atomic': {
-          const atomicView = [];
-          atomicView.push(checkCounter(counters));
-          atomicView.push(atomicHandler(item));
-          return atomicView;
+          const separator = checkCounter(counters);
+          if (separator) {
+            const atomicView = [];
+            atomicView.push(separator);
+            atomicView.push(atomicHandler(item));
+            return atomicView;
+          }
+          return atomicHandler(item);
         }
 
         case 'blockquote': {