node-redis/lib/multi-command.spec.ts

import { strict as assert } from 'assert';
import RedisMultiCommand from './multi-command';
import { encodeCommand } from './commander';
import { WatchError } from './errors';
import { spy } from 'sinon';
import { SQUARE_SCRIPT } from './client.spec';

describe('Multi Command', () => {
    describe('exec', () => {
        it('simple', async () => {
            const multi = RedisMultiCommand.create((queue, symbol) => {
                assert.deepEqual(
                    queue.map(({encodedCommand}) => encodedCommand),
                    [
                        encodeCommand(['MULTI']),
                        encodeCommand(['PING']),
                        encodeCommand(['EXEC']),
                    ]
                );

                assert.equal(
                    typeof symbol,
                    'symbol'
                );

                return Promise.resolve(['QUEUED', 'QUEUED', ['PONG']]);
            });

            multi.ping();

            assert.deepEqual(
                await multi.exec(),
                ['PONG']
            );
        });

        it('executing an empty queue should resolve without executing on the server', async () => {
            const executor = spy();

            assert.deepEqual(
                await RedisMultiCommand.create(executor).exec(),
                []
            );

            assert.ok(executor.notCalled);
        });

        it('WatchError', async () => {
            assert.rejects(
                RedisMultiCommand.create(() => Promise.resolve([null])).exec(),
                WatchError
            );
        });
    });

    describe('execAsPipeline', () => {
        it('simple', async () => {
            const multi = RedisMultiCommand.create(queue => {
                assert.deepEqual(
                    queue.map(({encodedCommand}) => encodedCommand),
                    [encodeCommand(['PING'])]
                );

                return Promise.resolve(['PONG']);
            });

            multi.ping();

            assert.deepEqual(
                await multi.execAsPipeline(),
                ['PONG']
            );
        });

        it('executing an empty queue should resolve without executing on the server', async () => {
            const executor = spy();

            assert.deepEqual(
                await RedisMultiCommand.create(executor).execAsPipeline(),
                []
            );

            assert.ok(executor.notCalled);
        });

        it('with scripts', async () => {
            const MultiWithScript = RedisMultiCommand.extend({
                scripts: {
                    square: SQUARE_SCRIPT
                }
            });

            assert.deepEqual(
                await new MultiWithScript(queue => {
                    assert.deepEqual(
                        queue.map(({encodedCommand}) => encodedCommand),
                        [
                            encodeCommand(['EVAL', SQUARE_SCRIPT.SCRIPT, '0', '2']),
                            encodeCommand(['EVALSHA', SQUARE_SCRIPT.SHA1, '0', '3']),
                        ]
                    );

                    return Promise.resolve([4, 9]);
                }).square(2).square(3).execAsPipeline(),
                [4, 9]
            );
        });
    });
});