fix(DraftableState return types): Fix DraftableState return types

lifechurch · May 30, 2019 · b4d7acf · b4d7acf
1 parent 3646740
commit b4d7acf
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 10 deletions.
diff --git a/src/lib/DraftableState.js b/src/lib/DraftableState.js
@@ -11,12 +11,12 @@ export const FORMAT_MARKDOWN = 'markdown';
 type FormatType = typeof FORMAT_HTML | typeof FORMAT_MARKDOWN;
 
 export default class DraftableState {
-  static createFromString(markup: string, format: FormatType) {
+  static createFromString(markup: string, format: FormatType):EditorState {
     switch (format) {
       case FORMAT_HTML:
-        return stateFromHTML(markup);
+        return EditorState.createWithContent(stateFromHTML(markup));
       case FORMAT_MARKDOWN:
-        return stateFromMarkdown(markup);
+        return EditorState.createWithContent(stateFromMarkdown(markup));
       default:
         throw new Error(`Format not supported: ${format}`);
     }
@@ -25,9 +25,9 @@ export default class DraftableState {
   static toString(editorState:EditorState, format: FormatType):string {
     switch (format) {
       case FORMAT_HTML:
-        return stateToHTML(editorState);
+        return stateToHTML(editorState.getCurrentContent());
       case FORMAT_MARKDOWN:
-        return stateToMarkdown(editorState);
+        return stateToMarkdown(editorState.getCurrentContent());
       default:
         throw new Error(`Format not supported: ${format}`);
     }

diff --git a/stories/index.stories.js b/stories/index.stories.js
@@ -1,6 +1,6 @@
 import React from 'react';
 import { storiesOf } from '@storybook/react';
-import { Draftable } from '../src';
+import { Draftable, DraftableState } from '../src';
 import BoldIcon from '../src/icons/TextBold';
 
 storiesOf('Draftable', module)
@@ -27,4 +27,10 @@ storiesOf('Draftable', module)
     return (
       <Draftable toolbarConfig={toolbarConfig} />
     );
+  })
+  .add('with initial state', () => {
+    const initialState = DraftableState.createFromString('<p>Test <b>bolded</b></p>', 'html');
+    return (
+      <Draftable initialState={initialState} />
+    );
   });
diff --git a/test/DraftableState.test.js b/test/DraftableState.test.js
@@ -7,8 +7,8 @@ describe('DraftableState', () => {
 
   test('Converts HTML to DraftJS state', () => {
     const convertedHTML = DraftableState.createFromString(HTML, FORMAT_HTML);
-    expect(convertedHTML.hasText()).toBeTruthy();
-    expect(convertedHTML.getPlainText()).toEqual('Test Document\nThis is the text I would like to initialize state with.');
+    expect(convertedHTML.getCurrentContent().hasText()).toBeTruthy();
+    expect(convertedHTML.getCurrentContent().getPlainText()).toEqual('Test Document\nThis is the text I would like to initialize state with.');
   });
 
   test('Converts DraftJS state to HTML', () => {
@@ -18,8 +18,8 @@ describe('DraftableState', () => {
 
   test('Converts Markdown to DraftJS state', () => {
     const convertedHTML = DraftableState.createFromString(MARKDOWN, FORMAT_MARKDOWN);
-    expect(convertedHTML.hasText()).toBeTruthy();
-    expect(convertedHTML.getPlainText()).toEqual('Test Document\nThis is the text I would like to initialize state with.');
+    expect(convertedHTML.getCurrentContent().hasText()).toBeTruthy();
+    expect(convertedHTML.getCurrentContent().getPlainText()).toEqual('Test Document\nThis is the text I would like to initialize state with.');
   });
 
   test('Converts DraftJS state to Markdown', () => {