Blame view

node_modules/htmlparser2/test/test-helper.js 1.84 KB
aaac7fed   liuqimichale   add
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
  var htmlparser2 = require(".."),
      fs = require("fs"),
      path = require("path"),
      assert = require("assert"),
  	Parser = htmlparser2.Parser,
  	CollectingHandler = htmlparser2.CollectingHandler;
  
  exports.writeToParser = function(handler, options, data){
  	var parser = new Parser(handler, options);
  	//first, try to run the test via chunks
  	for(var i = 0; i < data.length; i++){
  		parser.write(data.charAt(i));
  	}
  	parser.end();
  	//then parse everything
  	parser.parseComplete(data);
  };
  
  //returns a tree structure
  exports.getEventCollector = function(cb){
  	var handler = new CollectingHandler({onerror: cb, onend: function(){
  		cb(null, handler.events.reduce(eventReducer, []));
  	}});
  
  	return handler;
  };
  
  function eventReducer(events, arr){
  	if(arr[0] === "onerror" || arr[0] === "onend");
  	else if(arr[0] === "ontext" && events.length && events[events.length-1].event === "text"){
  		events[events.length-1].data[0] += arr[1];
  	} else {
  		events.push({
  			event: arr[0].substr(2),
  			data: arr.slice(1)
  		});
  	}
  
  	return events;
  }
  
  function getCallback(expected, done){
  	var repeated = false;
  
  	return function(err, actual){
  		assert.ifError(err);
  		try {
  			assert.deepEqual(expected, actual, "didn't get expected output");
  		} catch(e){
  			e.expected = JSON.stringify(expected, null, 2);
  			e.actual = JSON.stringify(actual, null, 2);
  			throw e;
  		}
  
  		if(repeated) done();
  		else repeated = true;
  	};
  }
  
  exports.mochaTest = function(name, root, test){
  	describe(name, readDir);
  
  	function readDir(cb){
  		var dir = path.join(root, name);
  
  		fs
  		.readdirSync(dir)
  		.filter(RegExp.prototype.test, /^[^\._]/) //ignore all files with a leading dot or underscore
  		.map(function(name){
  			return path.join(dir, name);
  		})
  		.map(require)
  		.forEach(runTest);
  	}
  
  	function runTest(file){
  		it(file.name, function(done){
  			test(file, getCallback(file.expected, done));
  		});
  	}
  };