/** * Copyright (c) Facebook, Inc. and its affiliates. * * This source code is licensed under the MIT license found in the * LICENSE file in the root directory of this source tree. * * @format * @flow * @emails oncall+draft_js */ 'use strict'; const ContentBlockNode = require("./ContentBlockNode"); const ContentState = require("./ContentState"); const EditorState = require("./EditorState"); const Immutable = require("immutable"); const getSampleSelectionMocksForTestingNestedBlocks = (): Object => { const root = document.createElement('div'); const contents = document.createElement('div'); contents.setAttribute('data-contents', 'true'); root.appendChild(contents); const text = [null, 'beta', null, 'delta']; const offsetKeys = ['a-0-0', 'b-0-0', 'c-0-0', 'd-0-0']; const contentBlocks = [new ContentBlockNode({ key: 'a', nextSibling: 'c', children: Immutable.List.of('b') }), new ContentBlockNode({ key: 'b', parent: 'a', text: text[1] }), new ContentBlockNode({ key: 'c', prevSibling: 'a', children: Immutable.List.of('d') }), new ContentBlockNode({ key: 'd', parent: 'c', text: text[3] })]; const contentState = ContentState.createFromBlockArray(contentBlocks); const editorState = EditorState.createWithContent(contentState); const textNodes = text.map(text => { if (!text) { return null; } return document.createTextNode(text); }); const leafChildren = textNodes.map(textNode => { if (!textNode) { return null; } const span = document.createElement('span'); span.appendChild(textNode); return span; }); const leafs = leafChildren.map((leafChild, index) => { if (!leafChild) { return null; } const blockKey = offsetKeys[index]; const span = document.createElement('span'); span.setAttribute('data-offset-key', blockKey); span.appendChild(leafChild); return span; }); const decorators = leafs.map((leaf, index) => { if (!leaf) { return null; } const blockKey = offsetKeys[index]; const span = document.createElement('span'); span.setAttribute('data-offset-key', blockKey); span.appendChild(leaf); return span; }); const blocks = offsetKeys.map((blockKey, index) => { const outerBlockElement = document.createElement('div'); const innerBlockElement = document.createElement('div'); innerBlockElement.setAttribute('data-offset-key', blockKey); outerBlockElement.setAttribute('data-offset-key', blockKey); outerBlockElement.setAttribute('data-block', 'true'); const decorator = decorators[index]; // only leaf nodes can have text if (decorator) { innerBlockElement.appendChild(decorator); } outerBlockElement.appendChild(innerBlockElement); return outerBlockElement; }); const blockCacheRef = {}; blocks.forEach((blockElem, index) => { const currentBlock = contentBlocks[index]; const parentKey = currentBlock.getParentKey(); // add this block reference to the cache lookup ref blockCacheRef[currentBlock.getKey()] = blockElem; // root nodes get appended directly to the contents block if (!parentKey) { contents.appendChild(blockElem); return; } // append to to the innerBlockElement of the parent block blockCacheRef[parentKey].firstChild.appendChild(blockElem); }); return { editorState, root, contents, blocks, decorators, leafs, leafChildren, textNodes }; }; module.exports = getSampleSelectionMocksForTestingNestedBlocks;