michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 'use strict'
michael@0: 
michael@0: const array = require('sdk/util/array');
michael@0: 
michael@0: exports.testHas = function(assert) {
michael@0:   var testAry = [1, 2, 3];
michael@0:   assert.equal(array.has([1, 2, 3], 1), true);
michael@0:   assert.equal(testAry.length, 3);
michael@0:   assert.equal(testAry[0], 1);
michael@0:   assert.equal(testAry[1], 2);
michael@0:   assert.equal(testAry[2], 3);
michael@0:   assert.equal(array.has(testAry, 2), true);
michael@0:   assert.equal(array.has(testAry, 3), true);
michael@0:   assert.equal(array.has(testAry, 4), false);
michael@0:   assert.equal(array.has(testAry, '1'), false);
michael@0: };
michael@0: exports.testHasAny = function(assert) {
michael@0:   var testAry = [1, 2, 3];
michael@0:   assert.equal(array.hasAny([1, 2, 3], [1]), true);
michael@0:   assert.equal(array.hasAny([1, 2, 3], [1, 5]), true);
michael@0:   assert.equal(array.hasAny([1, 2, 3], [5, 1]), true);
michael@0:   assert.equal(array.hasAny([1, 2, 3], [5, 2]), true);
michael@0:   assert.equal(array.hasAny([1, 2, 3], [5, 3]), true);
michael@0:   assert.equal(array.hasAny([1, 2, 3], [5, 4]), false);
michael@0:   assert.equal(testAry.length, 3);
michael@0:   assert.equal(testAry[0], 1);
michael@0:   assert.equal(testAry[1], 2);
michael@0:   assert.equal(testAry[2], 3);
michael@0:   assert.equal(array.hasAny(testAry, [2]), true);
michael@0:   assert.equal(array.hasAny(testAry, [3]), true);
michael@0:   assert.equal(array.hasAny(testAry, [4]), false);
michael@0:   assert.equal(array.hasAny(testAry), false);
michael@0:   assert.equal(array.hasAny(testAry, '1'), false);
michael@0: };
michael@0: 
michael@0: exports.testAdd = function(assert) {
michael@0:   var testAry = [1];
michael@0:   assert.equal(array.add(testAry, 1), false);
michael@0:   assert.equal(testAry.length, 1);
michael@0:   assert.equal(testAry[0], 1);
michael@0:   assert.equal(array.add(testAry, 2), true);
michael@0:   assert.equal(testAry.length, 2);
michael@0:   assert.equal(testAry[0], 1);
michael@0:   assert.equal(testAry[1], 2);
michael@0: };
michael@0: 
michael@0: exports.testRemove = function(assert) {
michael@0:   var testAry = [1, 2];
michael@0:   assert.equal(array.remove(testAry, 3), false);
michael@0:   assert.equal(testAry.length, 2);
michael@0:   assert.equal(testAry[0], 1);
michael@0:   assert.equal(testAry[1], 2);
michael@0:   assert.equal(array.remove(testAry, 2), true);
michael@0:   assert.equal(testAry.length, 1);
michael@0:   assert.equal(testAry[0], 1);
michael@0: };
michael@0: 
michael@0: exports.testFlatten = function(assert) {
michael@0:   assert.equal(array.flatten([1, 2, 3]).length, 3);
michael@0:   assert.equal(array.flatten([1, [2, 3]]).length, 3);
michael@0:   assert.equal(array.flatten([1, [2, [3]]]).length, 3);
michael@0:   assert.equal(array.flatten([[1], [[2, [3]]]]).length, 3);
michael@0: };
michael@0: 
michael@0: exports.testUnique = function(assert) {
michael@0:   var Class = function () {};
michael@0:   var A = {};
michael@0:   var B = new Class();
michael@0:   var C = [ 1, 2, 3 ];
michael@0:   var D = {};
michael@0:   var E = new Class();
michael@0: 
michael@0:   assert.deepEqual(array.unique([1,2,3,1,2]), [1,2,3]);
michael@0:   assert.deepEqual(array.unique([1,1,1,4,9,5,5]), [1,4,9,5]);
michael@0:   assert.deepEqual(array.unique([A, A, A, B, B, D]), [A,B,D]);
michael@0:   assert.deepEqual(array.unique([A, D, A, E, E, D, A, A, C]), [A, D, E, C])
michael@0: };
michael@0: 
michael@0: exports.testUnion = function(assert) {
michael@0:   var Class = function () {};
michael@0:   var A = {};
michael@0:   var B = new Class();
michael@0:   var C = [ 1, 2, 3 ];
michael@0:   var D = {};
michael@0:   var E = new Class();
michael@0: 
michael@0:   assert.deepEqual(array.union([1, 2, 3],[7, 1, 2]), [1, 2, 3, 7]);
michael@0:   assert.deepEqual(array.union([1, 1, 1, 4, 9, 5, 5], [10, 1, 5]), [1, 4, 9, 5, 10]);
michael@0:   assert.deepEqual(array.union([A, B], [A, D]), [A, B, D]);
michael@0:   assert.deepEqual(array.union([A, D], [A, E], [E, D, A], [A, C]), [A, D, E, C]);
michael@0: };
michael@0: 
michael@0: exports.testFind = function(assert) {
michael@0:   let isOdd = (x) => x % 2;
michael@0:   assert.equal(array.find([2, 4, 5, 7, 8, 9], isOdd), 5);
michael@0:   assert.equal(array.find([2, 4, 6, 8], isOdd), undefined);
michael@0:   assert.equal(array.find([2, 4, 6, 8], isOdd, null), null);
michael@0: };
michael@0: 
michael@0: require('test').run(exports);