zulip/frontend_tests/node_tests/int_dict.js

set_global('blueslip', global.make_zblueslip());
const IntDict = zrequire('int_dict').IntDict;

run_test('basic', () => {
    const d = new IntDict();

    assert.equal(d.size, 0);

    assert.deepEqual(Array.from(d.keys()), []);

    d.set(101, 'bar');
    assert.equal(d.get(101), 'bar');
    assert.notEqual(d.size, 0);

    d.set(101, 'baz');
    assert.equal(d.get(101), 'baz');
    assert.equal(d.size, 1);

    d.set(102, 'qux');
    assert.equal(d.get(101), 'baz');
    assert.equal(d.get(102), 'qux');
    assert.equal(d.size, 2);

    assert.equal(d.has(102), true);
    assert.equal(d.has(999), false);

    assert.deepEqual(Array.from(d.keys()), [101, 102]);
    assert.deepEqual(Array.from(d.values()), ['baz', 'qux']);

    d.delete(102);
    assert.equal(d.has(102), false);
    assert.strictEqual(d.get(102), undefined);

    assert.deepEqual(Array.from(d.keys()), [101]);

    const val = ['fred'];
    const res = d.set(103, val);
    assert.strictEqual(res, d);
});


run_test('each', () => {
    const d = new IntDict();
    d.set(4, 40);
    d.set(5, 50);
    d.set(6, 60);

    let unseen_keys = Array.from(d.keys());

    let cnt = 0;
    for (const [k, v] of d) {
        assert.equal(v, d.get(k));
        unseen_keys = _.without(unseen_keys, k);
        cnt += 1;
    }

    assert.equal(cnt, d.size);
    assert.equal(unseen_keys.length, 0);
});

/*
run_test('benchmark', () => {
    const d = new IntDict();
    const n = 5000;
    const t1 = new Date().getTime();

    _.each(_.range(n), (i) => {
        d.set(i, i);
    });

    _.each(_.range(n), (i) => {
        d.get(i, i);
    });

    const t2 = new Date().getTime();
    const elapsed = t2 - t1;
    console.log('elapsed (milli)', elapsed);
    console.log('per (micro)', 1000 * elapsed / n);
});
*/

run_test('filter_values', () => {
    const d = new IntDict();

    d.set(1, "fred");
    d.set(2, "foo");
    d.set(3, "bar");
    d.set(4, "baz");
    d.set(4, "fay");

    const pred = (v) => {
        return v.startsWith('f');
    };

    assert.deepEqual(d.filter_values(pred).sort(), ['fay', 'foo', 'fred']);
});

run_test('undefined_keys', () => {
    blueslip.clear_test_data();
    blueslip.set_test_data('error', 'Tried to call a IntDict method with an undefined key.');

    const d = new IntDict();

    assert.equal(d.has(undefined), false);
    assert.strictEqual(d.get(undefined), undefined);
    assert.equal(blueslip.get_test_logs('error').length, 2);
});

run_test('non integers', () => {
    blueslip.clear_test_data();
    blueslip.set_test_data('error', 'Tried to call a IntDict method with a non-integer.');

    const d = new IntDict();

    assert.equal(d.has('some-string'), false);
    assert.equal(blueslip.get_test_logs('error').length, 1);

    // verify stringified ints still work
    blueslip.clear_test_data();
    blueslip.set_test_data('error', 'Tried to call a IntDict method with a non-integer.');

    d.set('5', 'five');
    assert.equal(d.has(5), true);
    assert.equal(d.has('5'), true);

    assert.equal(d.get(5), 'five');
    assert.equal(d.get('5'), 'five');
    assert.equal(blueslip.get_test_logs('error').length, 3);
});

run_test('num_items', () => {
    const d = new IntDict();
    assert.equal(d.size, 0);

    d.set(101, 1);
    assert.equal(d.size, 1);

    d.set(101, 2);
    assert.equal(d.size, 1);

    d.set(102, 1);
    assert.equal(d.size, 2);
    d.delete(101);
    assert.equal(d.size, 1);
});

run_test('clear', () => {
    const d = new IntDict();

    function populate() {
        d.set(101, 1);
        assert.equal(d.get(101), 1);
        d.set(102, 2);
        assert.equal(d.get(102), 2);
    }

    populate();
    assert.equal(d.size, 2);

    d.clear();
    assert.equal(d.get(101), undefined);
    assert.equal(d.get(102), undefined);
    assert.equal(d.size, 0);

    // make sure it still works after clearing
    populate();
    assert.equal(d.size, 2);
});