add test for getCurrentEntityKey

Author: freedomlang

packages/utils/src/__test__/index.js

@@ -5,56 +5,67 @@ import {
   convertFromRaw,
   SelectionState,
 } from 'draft-js'
-import { getCharCount, getWordCount, getLineCount, hasEntity } from '../index'
+import { getCharCount, getWordCount, getLineCount, hasEntity, getCurrentEntityKey } from '../index'
 
-describe('Has Entity', () => {
+describe('Entity', () => {
   const createEditorStateFromRaw = raw => {
     const contentState = convertFromRaw(raw)
     return EditorState.createWithContent(contentState)
   }
 
-  it('create editorState and check if has Entity', () => {
-    const rawContent = {
-      blocks: [
-        {
-          text: 'Hey there duder',
-        },
-        {
-          type: 'atomic',
-          text: ' ',
-          entityRanges: [
-            {
-              key: 0,
-              length: 1,
-              offset: 0,
-            },
-          ],
-        },
-      ],
-      entityMap: {
-        0: {
-          data: {
-            title: 'Kitten',
-            src: 'https://placekitten.com/200/200',
+  const rawContent = {
+    blocks: [
+      {
+        text: 'Hey there duder',
+      },
+      {
+        type: 'atomic',
+        text: ' ',
+        entityRanges: [
+          {
+            key: 0,
+            length: 1,
+            offset: 0,
           },
-          mutability: 'IMMUTABLE',
-          type: 'IMAGE',
+        ],
+      },
+    ],
+    entityMap: {
+      0: {
+        data: {
+          title: 'Kitten',
+          src: 'https://placekitten.com/200/200',
         },
+        mutability: 'IMMUTABLE',
+        type: 'IMAGE',
       },
-    }
-    const editorState = createEditorStateFromRaw(rawContent)
-    const blockKey = editorState
-      .getCurrentContent()
-      .getLastBlock()
-      .getKey()
-    const selectionState = SelectionState.createEmpty(blockKey)
-    const updatedEditorState = EditorState.forceSelection(
-      editorState,
-      selectionState
-    )
+    },
+  }
+
+  const editorState = createEditorStateFromRaw(rawContent)
+  const blockKey = editorState
+    .getCurrentContent()
+    .getLastBlock()
+    .getKey()
+  const selectionState = SelectionState.createEmpty(blockKey)
+  const updatedEditorState = EditorState.forceSelection(
+    editorState,
+    selectionState
+  )
+
+  it('create editorState and check if the selection has Entity', () => {
     const result = hasEntity(updatedEditorState, 'IMAGE')
     expect(result).to.equal(true)
   })
+
+  it('create editorState and get entity key at selection', () => {
+    const result = getCurrentEntityKey(updatedEditorState)
+    const entityKey = updatedEditorState
+      .getCurrentContent()
+      .getBlockForKey(blockKey)
+      .getEntityAt(0)
+    expect(result).to.equal(entityKey)
+  })
 })
 
 describe('Char Counter', () => {