102 lines
4.2 KiB
TypeScript
102 lines
4.2 KiB
TypeScript
import { BlockchainLifecycle } from '@0xproject/dev-utils';
|
|
import { Web3Wrapper } from '@0xproject/web3-wrapper';
|
|
import * as chai from 'chai';
|
|
import * as Web3 from 'web3';
|
|
|
|
import { constants } from '../../util/constants';
|
|
import { ContractInstance } from '../../util/types';
|
|
import { chaiSetup } from '../utils/chai_setup';
|
|
|
|
chaiSetup.configure();
|
|
const expect = chai.expect;
|
|
const TokenTransferProxy = artifacts.require('./db/TokenTransferProxy.sol');
|
|
// In order to benefit from type-safety, we re-assign the global web3 instance injected by Truffle
|
|
// with type `any` to a variable of type `Web3`.
|
|
const web3: Web3 = (global as any).web3;
|
|
const blockchainLifecycle = new BlockchainLifecycle(constants.RPC_URL);
|
|
|
|
describe('TokenTransferProxy', () => {
|
|
const web3Wrapper = new Web3Wrapper(web3.currentProvider);
|
|
let owner: string;
|
|
let notOwner: string;
|
|
let address: string;
|
|
let tokenTransferProxy: ContractInstance;
|
|
before(async () => {
|
|
const accounts = await web3Wrapper.getAvailableAddressesAsync();
|
|
owner = address = accounts[0];
|
|
notOwner = accounts[1];
|
|
tokenTransferProxy = await TokenTransferProxy.deployed();
|
|
});
|
|
beforeEach(async () => {
|
|
await blockchainLifecycle.startAsync();
|
|
});
|
|
afterEach(async () => {
|
|
await blockchainLifecycle.revertAsync();
|
|
});
|
|
describe('addAuthorizedAddress', () => {
|
|
it('should throw if not called by owner', async () => {
|
|
return expect(tokenTransferProxy.addAuthorizedAddress(notOwner, { from: notOwner })).to.be.rejectedWith(
|
|
constants.REVERT,
|
|
);
|
|
});
|
|
it('should allow owner to add an authorized address', async () => {
|
|
await tokenTransferProxy.addAuthorizedAddress(address, { from: owner });
|
|
const isAuthorized = await tokenTransferProxy.authorized.call(address);
|
|
expect(isAuthorized).to.be.true();
|
|
});
|
|
it('should throw if owner attempts to authorize a duplicate address', async () => {
|
|
await tokenTransferProxy.addAuthorizedAddress(address, { from: owner });
|
|
return expect(tokenTransferProxy.addAuthorizedAddress(address, { from: owner })).to.be.rejectedWith(
|
|
constants.REVERT,
|
|
);
|
|
});
|
|
});
|
|
|
|
describe('removeAuthorizedAddress', () => {
|
|
it('should throw if not called by owner', async () => {
|
|
await tokenTransferProxy.addAuthorizedAddress(address, { from: owner });
|
|
return expect(
|
|
tokenTransferProxy.removeAuthorizedAddress(address, {
|
|
from: notOwner,
|
|
}),
|
|
).to.be.rejectedWith(constants.REVERT);
|
|
});
|
|
|
|
it('should allow owner to remove an authorized address', async () => {
|
|
await tokenTransferProxy.addAuthorizedAddress(address, { from: owner });
|
|
await tokenTransferProxy.removeAuthorizedAddress(address, {
|
|
from: owner,
|
|
});
|
|
const isAuthorized = await tokenTransferProxy.authorized.call(address);
|
|
expect(isAuthorized).to.be.false();
|
|
});
|
|
|
|
it('should throw if owner attempts to remove an address that is not authorized', async () => {
|
|
return expect(
|
|
tokenTransferProxy.removeAuthorizedAddress(address, {
|
|
from: owner,
|
|
}),
|
|
).to.be.rejectedWith(constants.REVERT);
|
|
});
|
|
});
|
|
|
|
describe('getAuthorizedAddresses', () => {
|
|
it('should return all authorized addresses', async () => {
|
|
const initial = await tokenTransferProxy.getAuthorizedAddresses();
|
|
expect(initial).to.have.length(1);
|
|
await tokenTransferProxy.addAuthorizedAddress(address, {
|
|
from: owner,
|
|
});
|
|
const afterAdd = await tokenTransferProxy.getAuthorizedAddresses();
|
|
expect(afterAdd).to.have.length(2);
|
|
expect(afterAdd).to.include(address);
|
|
|
|
await tokenTransferProxy.removeAuthorizedAddress(address, {
|
|
from: owner,
|
|
});
|
|
const afterRemove = await tokenTransferProxy.getAuthorizedAddresses();
|
|
expect(afterRemove).to.have.length(1);
|
|
});
|
|
});
|
|
});
|