es6map.js 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586
  1. "use strict";
  2. var assert = require('chai').assert;
  3. require('es6-shim');
  4. var Multimap = require('..');
  5. var map = new Multimap([
  6. ['a', 'one'],
  7. ['b', 1],
  8. ['a', 'two'],
  9. ['b', 2]
  10. ]);
  11. assert.equal(map.size, 4);
  12. assert.equal(map.get('a').length, 2);
  13. assert.equal(map.get('a')[0], 'one'); // ['one', 'two']
  14. assert.equal(map.get('a')[1], 'two'); // ['one', 'two']
  15. assert.equal(map.get('b').length, 2);
  16. assert.equal(map.get('b')[0], 1); // [1, 2]
  17. assert.equal(map.get('b')[1], 2); // [1, 2]
  18. assert(map.has('a'), "map contains key 'a'");
  19. assert(!map.has('foo'), "map does not contain key 'foo'");
  20. assert(map.has('a', 'one'), "map contains entry 'a'=>'one'");
  21. assert(!map.has('b', 3), "map does not contain entry 'b'=>3");
  22. map.set('a', 'three');
  23. assert.equal(map.size, 5);
  24. assert.equal(map.get('a').length, 3); // ['one', 'two', 'three']
  25. map.set('b', 3, 4);
  26. assert.equal(map.size, 7);
  27. assert(map.delete('a', 'three'), "delete 'a'=>'three'");
  28. assert.equal(map.size, 6);
  29. assert(!map.delete('x'), "empty 'x' for delete");
  30. assert(!map.delete('a', 'four'), "no such entry 'a'=>'four'");
  31. assert(map.delete('b'), "delete all 'b'");
  32. assert.equal(map.size, 2);
  33. map.set('b', 1, 2);
  34. assert.equal(map.size, 4); // 4
  35. var cnt = 0;
  36. map.forEach(function(value, key) {
  37. // iterates { 'a', 'one' }, { 'a', 'two' }, { 'b', 1 }, { 'b', 2 }
  38. cnt++;
  39. assert(key == 'a' || key == 'b', "key must be either 'a' or 'b'");
  40. });
  41. assert.equal(cnt, 4);
  42. cnt = 0;
  43. map.forEachEntry(function(entry, key) {
  44. // iterates { 'a', ['one', 'two'] }, { 'b', [1, 2] }
  45. cnt++;
  46. assert(key == 'a' || key == 'b', "key must be either 'a' or 'b'");
  47. assert.equal(entry.length, 2);
  48. });
  49. assert.equal(cnt, 2);
  50. var keys = map.keys(); // ['a', 'b']
  51. assert.equal(keys.next().value, 'a');
  52. assert.equal(keys.next().value, 'b');
  53. assert(keys.next().done);
  54. var values = map.values(); // ['one', 'two', 1, 2]
  55. assert.equal(values.next().value, 'one');
  56. assert.equal(values.next().value, 'two');
  57. assert.equal(values.next().value, 1);
  58. assert.equal(values.next().value, 2);
  59. assert(values.next().done);
  60. map.clear();
  61. assert.equal(map.size, 0);