embark/packages/plugins/ens/test/index.js

const ENS = require("../src/index");
const { fakeEmbark } = require('embark-testing');
const {Utils} = require('embarkjs');
const secureSend = Utils.secureSend;

describe('embark-ens', () => {
  let ens;

  const { embark } = fakeEmbark();

  beforeEach(() => {
    embark.events.setCommandHandler('namesystem:node:register', () => {});
    embark.config.blockchainConfig = { enabled: true };
    ens = new ENS(embark);
    ens.config = {
      embarkConfig: {
        generationDir: 'test-dir'
      },
      namesystemConfig: {
        register: {
          rootDomain: 'root.eth'
        },
        dappConnection: []
      },
      contractsConfig: {dappAutoEnable: true}
    };
  });

  afterEach(() => {
    embark.teardown();
  });

  it("should register the right artifact", () => {
    return new Promise(done => {
      const pipelineRegisterHandler = jest.fn((args, cb) => {
        ens.getEnsConfig((err, config) => {
          expect(args).toEqual({
            path: ['test-dir', 'config'],
            file: 'namesystem.json',
            format: 'json',
            dappAutoEnable: true,
            content: Object.assign({}, embark.config.namesystemConfig, config)
          });
          cb();
          done();
        });
      });

      embark.events.setCommandHandler('pipeline:register', pipelineRegisterHandler);

      ens.addArtifactFile({}, () => {});
    });
  });

  describe('safeRegisterSubDomain', () => {
    it('should register if the name is not registered', () => {
      return new Promise(done => {
        ens.ensResolve = jest.fn((name, cb) => { cb(null, null); });
        ens.registerSubDomain = jest.fn((defaultAccount, subDomainName, reverseNode, address, secureSend, callback) => callback());

        ens.safeRegisterSubDomain('test.eth', '0x0123', '0x4321',  () => {
          expect(ens.registerSubDomain).toHaveBeenCalledWith('0x4321', 'test.eth', '0xd523d7aaff8eefa323a17f2c79662ff1a8d952f6fa9cf53986347e99ada8098c', '0x0123', secureSend, expect.any(Function));
          done();
        });
      });
    });

    it('should not register if the name is already registered', () => {
      return new Promise(done => {
        ens.ensResolve = jest.fn((name, cb) => { cb(null, '0x0123'); });
        ens.registerSubDomain = jest.fn((defaultAccount, subDomainName, reverseNode, address, secureSend, callback) => callback());

        ens.safeRegisterSubDomain('test.eth', '0x0123', '0x4321', () => {
          expect(ens.registerSubDomain).not.toHaveBeenCalled();
          done();
        });
      });
    });
  });
});