js-xss/test/test_html_parser.js

/**
 * 测试 html parser
 */

var assert = require('assert');
var parser = require('../lib/parser');
var parseTag = parser.parseTag;
var parseAttr = parser.parseAttr;

describe('test HTML parser', function () {

  function escapeHtml (html) {
    return html.replace(/</g, '&lt;').replace(/>/g, '&gt;');
  }

  it('#parseTag', function () {
    var i = 0;
    var html = parseTag('hello<A href="#">www</A>ccc<b>', function (originPosition, position, tag, html, isClosing) {
      i++;
      console.log(arguments);
      if (i === 1) {
        // 第1个标签
        assert.equal(originPosition, 5);
        assert.equal(position, 5);
        assert.equal(tag, 'a');
        assert.equal(html, '<A href="#">');
        assert.equal(isClosing, false);
        return '[link]';
      } else if (i === 2) {
        // 第2个标签
        assert.equal(originPosition, 20);
        assert.equal(position, 14);
        assert.equal(tag, 'a');
        assert.equal(html, '</A>');
        assert.equal(isClosing, true);
        return '[/link]';
      } else if (i === 3) {
        // 第3个标签
        assert.equal(originPosition, 27);
        assert.equal(position, 24);
        assert.equal(tag, 'b');
        assert.equal(html, '<b>');
        assert.equal(isClosing, false);
        return '[B]';
      } else {
        throw new Error();
      }
    }, escapeHtml);
    console.log(html);
    assert.equal(html, 'hello[link]www[/link]ccc[B]');
  });

  it('#parseAttr', function () {
    var i = 0;
    function attr (n, v) {
      if (v) {
        return n + '="' + v.replace(/"/g, '&quote;') + '"';
      } else {
        return n;
      }
    }
    var html = parseAttr('href="#"attr1=b attr2=c attr3 attr4=\'value4"\'attr5/', function (name, value) {
      i++;
      console.log(arguments);
      if (i === 1) {
        assert.equal(name, 'href');
        assert.equal(value, '#');
        return attr(name, value);
      } else if (i === 2) {
        assert.equal(name, 'attr1');
        assert.equal(value, 'b');
        return attr(name, value);
      } else if (i === 3) {
        assert.equal(name, 'attr2');
        assert.equal(value, 'c');
        return attr(name, value);
      } else if (i === 4) {
        assert.equal(name, 'attr3');
        assert.equal(value, '');
        return attr(name, value);
      } else if (i === 5) {
        assert.equal(name, 'attr4');
        assert.equal(value, 'value4"');
        return attr(name, value);
      } else if (i === 6) {
        assert.equal(name, 'attr5');
        assert.equal(value, '');
        return attr(name, value);
      } else {
        throw new Error();
      }
    });
    console.log(html);
    assert.equal(html, 'href="#" attr1="b" attr2="c" attr3 attr4="value4&quote;" attr5');
  });

});