api/packages/rpc-provider/src/ws/onMessageSubscribe.spec.js

// Copyright 2017-2018 @polkadot/rpc-provider authors & contributors
// This software may be modified and distributed under the terms
// of the ISC license. See the LICENSE file for details.

import Ws from './index';

describe('onMessageSubscribe', () => {
  let ws;
  let errorSpy;

  beforeEach(() => {
    ws = new Ws('ws://127.0.0.1:1234', false);
    errorSpy = jest.spyOn(console, 'error');
  });

  afterEach(() => {
    errorSpy.mockRestore();
  });

  it('fails with log when handler not found', () => {
    ws.onSocketMessage({ data: '{"method":"test","params":{"subscription":2}}' });

    expect(errorSpy).toHaveBeenCalledWith(
      expect.anything(), expect.anything(), 'Unable to find handler for subscription=test::2'
    );
  });

  it('calls the subscriber with data', (done) => {
    ws.handlers[11] = {
      callback: (_, id) => {},
      method: 'test',
      subscription: {
        callback: (_, result) => {
          expect(result).toEqual('test');
          done();
        },
        type: 'test'
      }
    };

    ws.onSocketMessage({ data: '{"jsonrpc":"2.0","id":11,"result":22}' });
    ws.onSocketMessage({ data: '{"jsonrpc":"2.0","method":"test","params":{"subscription":22,"result":"test"}}' });
  });

  it('calls the subscriber with error', (done) => {
    ws.handlers[11] = {
      callback: (_, id) => {},
      method: 'test',
      subscription: {
        callback: (error) => {
          expect(error.message).toMatch(/test/);
          done();
        },
        type: 'test'
      }
    };

    ws.onSocketMessage({ data: '{"jsonrpc":"2.0","id":11,"result":22}' });
    ws.onSocketMessage({ data: '{"jsonrpc":"2.0","method":"test","params":{"subscription":22,"error":{"message":"test"}}}' });
  });
});