129 lines
		
	
	
		
			6.3 KiB
		
	
	
	
		
			TypeScript
		
	
	
	
	
	
			
		
		
	
	
			129 lines
		
	
	
		
			6.3 KiB
		
	
	
	
		
			TypeScript
		
	
	
	
	
	
import { blockchainTests, expect, provider, txDefaults, web3Wrapper } from '@0x/contracts-test-utils';
 | 
						|
import { RevertReason } from '@0x/types';
 | 
						|
import { BigNumber } from '@0x/utils';
 | 
						|
 | 
						|
import { artifacts } from './artifacts';
 | 
						|
 | 
						|
import { MixinAuthorizableContract } from './wrappers';
 | 
						|
 | 
						|
blockchainTests.resets('Authorizable', () => {
 | 
						|
    let owner: string;
 | 
						|
    let notOwner: string;
 | 
						|
    let address: string;
 | 
						|
    let authorizable: MixinAuthorizableContract;
 | 
						|
 | 
						|
    before(async () => {
 | 
						|
        const accounts = await web3Wrapper.getAvailableAddressesAsync();
 | 
						|
        [owner, address, notOwner] = accounts.slice(0, 3);
 | 
						|
        authorizable = await MixinAuthorizableContract.deployFrom0xArtifactAsync(
 | 
						|
            artifacts.MixinAuthorizable,
 | 
						|
            provider,
 | 
						|
            txDefaults,
 | 
						|
            artifacts,
 | 
						|
        );
 | 
						|
    });
 | 
						|
 | 
						|
    describe('addAuthorizedAddress', () => {
 | 
						|
        it('should revert if not called by owner', async () => {
 | 
						|
            const tx = authorizable.addAuthorizedAddress(notOwner).sendTransactionAsync({ from: notOwner });
 | 
						|
            return expect(tx).to.revertWith(RevertReason.OnlyContractOwner);
 | 
						|
        });
 | 
						|
 | 
						|
        it('should allow owner to add an authorized address', async () => {
 | 
						|
            await authorizable.addAuthorizedAddress(address).awaitTransactionSuccessAsync({ from: owner });
 | 
						|
            const isAuthorized = await authorizable.authorized(address).callAsync();
 | 
						|
            expect(isAuthorized).to.be.true();
 | 
						|
        });
 | 
						|
 | 
						|
        it('should revert if owner attempts to authorize a duplicate address', async () => {
 | 
						|
            await authorizable.addAuthorizedAddress(address).awaitTransactionSuccessAsync({ from: owner });
 | 
						|
            const tx = authorizable.addAuthorizedAddress(address).sendTransactionAsync({ from: owner });
 | 
						|
            return expect(tx).to.revertWith(RevertReason.TargetAlreadyAuthorized);
 | 
						|
        });
 | 
						|
    });
 | 
						|
 | 
						|
    describe('removeAuthorizedAddress', () => {
 | 
						|
        it('should revert if not called by owner', async () => {
 | 
						|
            await authorizable.addAuthorizedAddress(address).awaitTransactionSuccessAsync({ from: owner });
 | 
						|
            const tx = authorizable.removeAuthorizedAddress(address).sendTransactionAsync({ from: notOwner });
 | 
						|
            return expect(tx).to.revertWith(RevertReason.OnlyContractOwner);
 | 
						|
        });
 | 
						|
 | 
						|
        it('should allow owner to remove an authorized address', async () => {
 | 
						|
            await authorizable.addAuthorizedAddress(address).awaitTransactionSuccessAsync({ from: owner });
 | 
						|
            await authorizable.removeAuthorizedAddress(address).awaitTransactionSuccessAsync({ from: owner });
 | 
						|
            const isAuthorized = await authorizable.authorized(address).callAsync();
 | 
						|
            expect(isAuthorized).to.be.false();
 | 
						|
        });
 | 
						|
 | 
						|
        it('should revert if owner attempts to remove an address that is not authorized', async () => {
 | 
						|
            const tx = authorizable.removeAuthorizedAddress(address).sendTransactionAsync({ from: owner });
 | 
						|
            return expect(tx).to.revertWith(RevertReason.TargetNotAuthorized);
 | 
						|
        });
 | 
						|
    });
 | 
						|
 | 
						|
    describe('removeAuthorizedAddressAtIndex', () => {
 | 
						|
        it('should revert if not called by owner', async () => {
 | 
						|
            await authorizable.addAuthorizedAddress(address).awaitTransactionSuccessAsync({ from: owner });
 | 
						|
            const index = new BigNumber(0);
 | 
						|
            const tx = authorizable
 | 
						|
                .removeAuthorizedAddressAtIndex(address, index)
 | 
						|
                .sendTransactionAsync({ from: notOwner });
 | 
						|
            return expect(tx).to.revertWith(RevertReason.OnlyContractOwner);
 | 
						|
        });
 | 
						|
 | 
						|
        it('should revert if index is >= authorities.length', async () => {
 | 
						|
            await authorizable.addAuthorizedAddress(address).awaitTransactionSuccessAsync({ from: owner });
 | 
						|
            const index = new BigNumber(1);
 | 
						|
            const tx = authorizable
 | 
						|
                .removeAuthorizedAddressAtIndex(address, index)
 | 
						|
                .sendTransactionAsync({ from: owner });
 | 
						|
            return expect(tx).to.revertWith(RevertReason.IndexOutOfBounds);
 | 
						|
        });
 | 
						|
 | 
						|
        it('should revert if owner attempts to remove an address that is not authorized', async () => {
 | 
						|
            const index = new BigNumber(0);
 | 
						|
            const tx = authorizable
 | 
						|
                .removeAuthorizedAddressAtIndex(address, index)
 | 
						|
                .sendTransactionAsync({ from: owner });
 | 
						|
            return expect(tx).to.revertWith(RevertReason.TargetNotAuthorized);
 | 
						|
        });
 | 
						|
 | 
						|
        it('should revert if address at index does not match target', async () => {
 | 
						|
            const address1 = address;
 | 
						|
            const address2 = notOwner;
 | 
						|
            await authorizable.addAuthorizedAddress(address1).awaitTransactionSuccessAsync({ from: owner });
 | 
						|
            await authorizable.addAuthorizedAddress(address2).awaitTransactionSuccessAsync({ from: owner });
 | 
						|
            const address1Index = new BigNumber(0);
 | 
						|
            const tx = authorizable
 | 
						|
                .removeAuthorizedAddressAtIndex(address2, address1Index)
 | 
						|
                .sendTransactionAsync({ from: owner });
 | 
						|
            return expect(tx).to.revertWith(RevertReason.AuthorizedAddressMismatch);
 | 
						|
        });
 | 
						|
 | 
						|
        it('should allow owner to remove an authorized address', async () => {
 | 
						|
            await authorizable.addAuthorizedAddress(address).awaitTransactionSuccessAsync({ from: owner });
 | 
						|
            const index = new BigNumber(0);
 | 
						|
            await authorizable.removeAuthorizedAddressAtIndex(address, index).awaitTransactionSuccessAsync({
 | 
						|
                from: owner,
 | 
						|
            });
 | 
						|
            const isAuthorized = await authorizable.authorized(address).callAsync();
 | 
						|
            expect(isAuthorized).to.be.false();
 | 
						|
        });
 | 
						|
    });
 | 
						|
 | 
						|
    describe('getAuthorizedAddresses', () => {
 | 
						|
        it('should return all authorized addresses', async () => {
 | 
						|
            const initial = await authorizable.getAuthorizedAddresses().callAsync();
 | 
						|
            expect(initial).to.have.length(0);
 | 
						|
            await authorizable.addAuthorizedAddress(address).awaitTransactionSuccessAsync({ from: owner });
 | 
						|
            const afterAdd = await authorizable.getAuthorizedAddresses().callAsync();
 | 
						|
            expect(afterAdd).to.have.length(1);
 | 
						|
            expect(afterAdd).to.include(address);
 | 
						|
            await authorizable.removeAuthorizedAddress(address).awaitTransactionSuccessAsync({ from: owner });
 | 
						|
            const afterRemove = await authorizable.getAuthorizedAddresses().callAsync();
 | 
						|
            expect(afterRemove).to.have.length(0);
 | 
						|
        });
 | 
						|
    });
 | 
						|
});
 |