pen/test/test-markdown-socket.js

A better Markdown previewer

import assert from "assert";
import fs from "fs";
import helper from "./lib/helper";
import http from "http";
import MarkdownSocket from "../src/markdown-socket";
import { w3cwebsocket as WebSocket } from "websocket";

describe("MarkdownSocket", () => {
  let server;
  let mdSocket;

  beforeEach(done => {
    helper.makeDirectory("md-root");
    helper.createFile("md-root/test.md", "# hello");
    server = http.createServer((req, res) => {
      res.end("hello");
    });
    mdSocket = new MarkdownSocket(helper.path("md-root"));
    mdSocket.listenTo(server);
    server.listen(1234, done);
  });

  afterEach(done => {
    helper.clean();
    mdSocket.close();
    server.close(done);
  });

  it("handles a websocket connection", done => {
    let client = new WebSocket("ws://localhost:1234/test.md");

    client.onopen = () => {
      done();
    };
  });

  it("cannot handle a non markdown connection", done => {
    let client = new WebSocket("ws://localhost:1234");

    client.onerror = () => {
      done();
    };
  });

  it("opens a Markdown file and sends the parsed HTML", done => {
    let client = new WebSocket("ws://localhost:1234/test.md");

    client.onmessage = message => {
      assert.equal(message.data, '<h1 id="hello">hello</h1>\n');
      done();
    };
  });

  it("sends parsed HTML data again when the file is updated", done => {
    const callback = err => {
      if (err) {
        done(err);
      }
    };

    let called = 0;
    let client = new WebSocket("ws://localhost:1234/test.md");
    client.onmessage = message => {
      switch (called) {
        case 0:
          assert.equal(message.data, '<h1 id="hello">hello</h1>\n');
          fs.writeFile(
            helper.path("md-root/test.md"),
            "```js\nvar a=10;\n```",
            callback
          );
          break;
        case 1:
          assert.equal(
            message.data,
            '<pre><code class="hljs language-js"><span class="hljs-keyword">var</span> a=<span class="hljs-number">10</span>;\n</code></pre>\n'
          );
          fs.writeFile(
            helper.path("md-root/test.md"),
            "* nested\n  * nnested\n    * nnnested",
            callback
          );
          break;
        case 2:
          assert.equal(
            message.data,
            "<ul>\n<li>nested\n<ul>\n<li>nnested\n<ul>\n<li>nnnested</li>\n</ul>\n</li>\n</ul>\n</li>\n</ul>\n"
          );
          done();
          break;
      }
      called += 1;
    };
  });

  it("ignores when there is no file for the path", done => {
    let client = new WebSocket("ws://localhost:1234/no-file.md");
    client.onmessage = message => {
      assert.equal(message.data, "Not found");
      done();
    };
  });
});