58 lines
2.3 KiB
TypeScript
58 lines
2.3 KiB
TypeScript
import { chaiSetup, constants, provider, txDefaults, web3Wrapper } from '@0x/contracts-test-utils';
|
|
import { BlockchainLifecycle } from '@0x/dev-utils';
|
|
import { OwnableRevertErrors } from '@0x/utils';
|
|
import * as chai from 'chai';
|
|
import * as _ from 'lodash';
|
|
|
|
import { artifacts, TestOwnableContract } from '../src';
|
|
|
|
chaiSetup.configure();
|
|
const expect = chai.expect;
|
|
const blockchainLifecycle = new BlockchainLifecycle(web3Wrapper);
|
|
|
|
describe('Ownable', () => {
|
|
let ownable: TestOwnableContract;
|
|
let owner: string;
|
|
let nonOwner: string;
|
|
|
|
before(async () => {
|
|
const accounts = await web3Wrapper.getAvailableAddressesAsync();
|
|
owner = await accounts[0];
|
|
nonOwner = await accounts[1];
|
|
await blockchainLifecycle.startAsync();
|
|
// Deploy Ownable from the owner address
|
|
txDefaults.from = owner;
|
|
ownable = await TestOwnableContract.deployFrom0xArtifactAsync(artifacts.TestOwnable, provider, txDefaults, {});
|
|
});
|
|
|
|
after(async () => {
|
|
await blockchainLifecycle.revertAsync();
|
|
});
|
|
|
|
describe('onlyOwner', () => {
|
|
it('should revert if sender is not the owner', async () => {
|
|
const expectedError = new OwnableRevertErrors.OnlyOwnerError(nonOwner, owner);
|
|
return expect(ownable.externalOnlyOwner.callAsync({ from: nonOwner })).to.revertWith(expectedError);
|
|
});
|
|
|
|
it('should succeed if sender is the owner', async () => {
|
|
const isSuccessful = await ownable.externalOnlyOwner.callAsync({ from: owner });
|
|
expect(isSuccessful).to.be.true();
|
|
});
|
|
});
|
|
|
|
describe('transferOwnership', () => {
|
|
it('should revert if the specified new owner is the zero address', async () => {
|
|
const expectedError = new OwnableRevertErrors.TransferOwnerToZeroError();
|
|
const tx = ownable.transferOwnership.sendTransactionAsync(constants.NULL_ADDRESS, { from: owner });
|
|
return expect(tx).to.revertWith(expectedError);
|
|
});
|
|
|
|
it('should transfer ownership if the specified new owner is not the zero address', async () => {
|
|
expect(ownable.transferOwnership.sendTransactionAsync(nonOwner, { from: owner })).to.be.fulfilled('');
|
|
const updatedOwner = await ownable.owner.callAsync();
|
|
expect(updatedOwner).to.be.eq(nonOwner);
|
|
});
|
|
});
|
|
});
|