87 lines
2.1 KiB
JavaScript
87 lines
2.1 KiB
JavaScript
|
"use strict";
|
||
|
|
|
||
|
|
var assert = require('chai').assert;
|
||
|
|
require('es6-shim');
|
||
|
|
|
||
|
|
var Multimap = require('..');
|
||
|
|
|
||
|
|
var map = new Multimap([
|
||
|
|
['a', 'one'],
|
||
|
|
['b', 1],
|
||
|
|
['a', 'two'],
|
||
|
|
['b', 2]
|
||
|
|
]);
|
||
|
|
|
||
|
|
assert.equal(map.size, 4);
|
||
|
|
|
||
|
|
assert.equal(map.get('a').length, 2);
|
||
|
|
assert.equal(map.get('a')[0], 'one'); // ['one', 'two']
|
||
|
|
assert.equal(map.get('a')[1], 'two'); // ['one', 'two']
|
||
|
|
|
||
|
|
assert.equal(map.get('b').length, 2);
|
||
|
|
assert.equal(map.get('b')[0], 1); // [1, 2]
|
||
|
|
assert.equal(map.get('b')[1], 2); // [1, 2]
|
||
|
|
|
||
|
|
|
||
|
|
assert(map.has('a'), "map contains key 'a'");
|
||
|
|
assert(!map.has('foo'), "map does not contain key 'foo'");
|
||
|
|
|
||
|
|
assert(map.has('a', 'one'), "map contains entry 'a'=>'one'");
|
||
|
|
assert(!map.has('b', 3), "map does not contain entry 'b'=>3");
|
||
|
|
|
||
|
|
map.set('a', 'three');
|
||
|
|
|
||
|
|
assert.equal(map.size, 5);
|
||
|
|
assert.equal(map.get('a').length, 3); // ['one', 'two', 'three']
|
||
|
|
|
||
|
|
map.set('b', 3, 4);
|
||
|
|
assert.equal(map.size, 7);
|
||
|
|
|
||
|
|
assert(map.delete('a', 'three'), "delete 'a'=>'three'");
|
||
|
|
assert.equal(map.size, 6);
|
||
|
|
assert(!map.delete('x'), "empty 'x' for delete");
|
||
|
|
assert(!map.delete('a', 'four'), "no such entry 'a'=>'four'");
|
||
|
|
assert(map.delete('b'), "delete all 'b'");
|
||
|
|
|
||
|
|
assert.equal(map.size, 2);
|
||
|
|
|
||
|
|
map.set('b', 1, 2);
|
||
|
|
assert.equal(map.size, 4); // 4
|
||
|
|
|
||
|
|
var cnt = 0;
|
||
|
|
map.forEach(function(value, key) {
|
||
|
|
// iterates { 'a', 'one' }, { 'a', 'two' }, { 'b', 1 }, { 'b', 2 }
|
||
|
|
cnt++;
|
||
|
|
assert(key == 'a' || key == 'b', "key must be either 'a' or 'b'");
|
||
|
|
});
|
||
|
|
|
||
|
|
assert.equal(cnt, 4);
|
||
|
|
|
||
|
|
cnt = 0;
|
||
|
|
map.forEachEntry(function(entry, key) {
|
||
|
|
// iterates { 'a', ['one', 'two'] }, { 'b', [1, 2] }
|
||
|
|
cnt++;
|
||
|
|
assert(key == 'a' || key == 'b', "key must be either 'a' or 'b'");
|
||
|
|
assert.equal(entry.length, 2);
|
||
|
|
});
|
||
|
|
|
||
|
|
assert.equal(cnt, 2);
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
var keys = map.keys(); // ['a', 'b']
|
||
|
|
assert.equal(keys.next().value, 'a');
|
||
|
|
assert.equal(keys.next().value, 'b');
|
||
|
|
assert(keys.next().done);
|
||
|
|
|
||
|
|
var values = map.values(); // ['one', 'two', 1, 2]
|
||
|
|
assert.equal(values.next().value, 'one');
|
||
|
|
assert.equal(values.next().value, 'two');
|
||
|
|
assert.equal(values.next().value, 1);
|
||
|
|
assert.equal(values.next().value, 2);
|
||
|
|
assert(values.next().done);
|
||
|
|
|
||
|
|
map.clear();
|
||
|
|
|
||
|
|
assert.equal(map.size, 0);
|