Mailspring/internal_packages/composer-spellcheck/spec/spellcheck-composer-extension-spec.es6

import fs from 'fs';
import path from 'path';
import {Spellchecker, Message} from 'nylas-exports';

import SpellcheckComposerExtension from '../lib/spellcheck-composer-extension';

const initialPath = path.join(__dirname, 'fixtures', 'california-with-misspellings-before.html');
const initialHTML = fs.readFileSync(initialPath).toString();
const afterPath = path.join(__dirname, 'fixtures', 'california-with-misspellings-after.html');
const afterHTML = fs.readFileSync(afterPath).toString();

describe('SpellcheckComposerExtension', function spellcheckComposerExtension() {
  beforeEach(() => {
    // Avoid differences between node-spellcheck on different platforms
    const lookupPath = path.join(__dirname, 'fixtures', 'california-spelling-lookup.json');
    const spellings = JSON.parse(fs.readFileSync(lookupPath));
    spyOn(Spellchecker, 'isMisspelled').andCallFake(word => spellings[word])
    spyOn(Spellchecker.handler, 'provideHintText').andReturn({
      then(cb) {
        cb()
      },
    })
  });

  describe("onContentChanged", () => {
    it("correctly walks a DOM tree and surrounds mispelled words", () => {
      const node = document.createElement('div');
      node.innerHTML = initialHTML;

      const editor = {
        rootNode: node,
        whilePreservingSelection: (cb) => cb(),
      };

      SpellcheckComposerExtension.onContentChanged({editor});
      advanceClock(1000) // Wait for debounce
      advanceClock(1) // Wait for defer
      expect(node.innerHTML).toEqual(afterHTML);
    });

    it("does not mark misspelled words inside A, CODE and PRE tags", () => {
      const node = document.createElement('div');
      node.innerHTML = `
      <br>
      This is a testst! I have a few misspellled words.
      <code>myvariable</code>
      <pre>
         fragmen = document.applieed();
      </pre>
      <a href="apple.com">I like appples!</a>
      <br>
      This is back to normall.
      `;

      const editor = {
        rootNode: node,
        whilePreservingSelection: (cb) => cb(),
      };

      SpellcheckComposerExtension.onContentChanged({editor});
      advanceClock(1000) // Wait for debounce
      advanceClock(1) // Wait for defer
      expect(node.innerHTML).toEqual(`
      <br>
      This is a <spelling class="misspelled">testst</spelling>! I have a few <spelling class="misspelled">misspellled</spelling> words.
      <code>myvariable</code>
      <pre>         fragmen = document.applieed();
      </pre>
      <a href="apple.com">I like appples!</a>
      <br>
      This is back to <spelling class="misspelled">normall</spelling>.
      `);
    });
  });

  describe("applyTransformsForSending", () => {
    it("removes the spelling annotations it inserted", () => {
      const draft = new Message({ body: afterHTML });
      const fragment = document.createDocumentFragment();
      const draftBodyRootNode = document.createElement('root')
      fragment.appendChild(draftBodyRootNode)
      draftBodyRootNode.innerHTML = afterHTML
      SpellcheckComposerExtension.applyTransformsForSending({draftBodyRootNode, draft});
      expect(draftBodyRootNode.innerHTML).toEqual(initialHTML);
    });
  });
});