management/front/dkha-web-sz-main/node_modules/unique-stream/test/index.js

110 lines
2.3 KiB
JavaScript
Raw Normal View History

2023-12-18 13:12:25 +08:00
var expect = require('chai').expect
, unique = require('..')
, Stream = require('stream')
, after = require('after')
, setImmediate = global.setImmediate || process.nextTick;
describe('unique stream', function() {
function makeStream(type) {
var s = new Stream();
s.readable = true;
var n = 10;
var next = after(n, function () {
setImmediate(function () {
s.emit('end');
});
});
for (var i = 0; i < n; i++) {
var o = {
type: type,
name: 'name ' + i,
number: i * 10
};
(function (o) {
setImmediate(function () {
s.emit('data', o);
next();
});
})(o);
}
return s;
}
it('should be able to uniqueify objects based on JSON data', function(done) {
var aggregator = unique();
makeStream('a')
.pipe(aggregator);
makeStream('a')
.pipe(aggregator);
var n = 0;
aggregator
.on('data', function () {
n++;
})
.on('end', function () {
expect(n).to.equal(10);
done();
});
});
it('should be able to uniqueify objects based on a property', function(done) {
var aggregator = unique('number');
makeStream('a')
.pipe(aggregator);
makeStream('b')
.pipe(aggregator);
var n = 0;
aggregator
.on('data', function () {
n++;
})
.on('end', function () {
expect(n).to.equal(10);
done();
});
});
it('should be able to uniqueify objects based on a function', function(done) {
var aggregator = unique(function (data) {
return data.name;
});
makeStream('a')
.pipe(aggregator);
makeStream('b')
.pipe(aggregator);
var n = 0;
aggregator
.on('data', function () {
n++;
})
.on('end', function () {
expect(n).to.equal(10);
done();
});
});
it('should be able to handle uniqueness when not piped', function(done) {
var stream = unique();
var count = 0;
stream.on('data', function (data) {
expect(data).to.equal('hello');
count++;
});
stream.on('end', function() {
expect(count).to.equal(1);
done();
});
stream.write('hello');
stream.write('hello');
stream.end();
});
});