index.js 2.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091
  1. "use strict";
  2. var assert = require('chai').assert;
  3. var Multimap = require('..');
  4. var map = new Multimap([
  5. ['a', 'one'],
  6. ['b', 1],
  7. ['a', 'two'],
  8. ['b', 2]
  9. ]);
  10. assert.equal(map.size, 4);
  11. assert.equal(map.count, 2);
  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.count, 2);
  25. assert.equal(map.get('a').length, 3); // ['one', 'two', 'three']
  26. map.set('b', 3, 4);
  27. assert.equal(map.size, 7);
  28. assert.equal(map.count, 2);
  29. assert(map.delete('a', 'three'), "delete 'a'=>'three'");
  30. assert.equal(map.size, 6);
  31. assert.equal(map.count, 2);
  32. assert(!map.delete('x'), "empty 'x' for delete");
  33. assert(!map.delete('a', 'four'), "no such entry 'a'=>'four'");
  34. assert(map.delete('b'), "delete all 'b'");
  35. assert.equal(map.size, 2);
  36. assert.equal(map.count, 1);
  37. map.set('b', 1, 2);
  38. assert.equal(map.size, 4); // 4
  39. assert.equal(map.count, 2);
  40. var cnt = 0;
  41. map.forEach(function(value, key) {
  42. // iterates { 'a', 'one' }, { 'a', 'two' }, { 'b', 1 }, { 'b', 2 }
  43. cnt++;
  44. assert(key == 'a' || key == 'b', "key must be either 'a' or 'b'");
  45. });
  46. assert.equal(cnt, 4);
  47. cnt = 0;
  48. map.forEachEntry(function(entry, key) {
  49. // iterates { 'a', ['one', 'two'] }, { 'b', [1, 2] }
  50. cnt++;
  51. assert(key == 'a' || key == 'b', "key must be either 'a' or 'b'");
  52. assert.equal(entry.length, 2);
  53. });
  54. assert.equal(cnt, 2);
  55. var keys = map.keys(); // ['a', 'b']
  56. assert.equal(keys.next().value, 'a');
  57. assert.equal(keys.next().value, 'b');
  58. assert(keys.next().done);
  59. var values = map.values(); // ['one', 'two', 1, 2]
  60. assert.equal(values.next().value, 'one');
  61. assert.equal(values.next().value, 'two');
  62. assert.equal(values.next().value, 1);
  63. assert.equal(values.next().value, 2);
  64. assert(values.next().done);
  65. map.clear();
  66. assert.equal(map.size, 0);
  67. assert.equal(map.count, 0);