import 'mocha';
import * as chai from 'chai';
import {web3Factory} from './utils/web3_factory';
import {ZeroEx} from '../src/0x.js';
import promisify = require('es6-promisify');
import * as _ from 'lodash';
import {BlockchainLifecycle} from './utils/blockchain_lifecycle';
import * as BigNumber from 'bignumber.js';
import {createSignedOrder} from './utils/order';
import {Token} from '../src/types';
import * as Web3 from 'web3';

const expect = chai.expect;
const blockchainLifecycle = new BlockchainLifecycle();

describe('ExchangeWrapper', () => {
    let zeroEx: ZeroEx;
    let userAddresses: string[];
    let web3: Web3;
    before(async () => {
        web3 = web3Factory.create();
        zeroEx = new ZeroEx(web3);
        userAddresses = await promisify(web3.eth.getAccounts)();
        web3.eth.defaultAccount = userAddresses[0];
    });
    beforeEach(async () => {
        await blockchainLifecycle.startAsync();
    });
    afterEach(async () => {
        await blockchainLifecycle.revertAsync();
    });
    describe('#isValidSignatureAsync', () => {
        // The Exchange smart contract `isValidSignature` method only validates orderHashes and assumes
        // the length of the data is exactly 32 bytes. Thus for these tests, we use data of this size.
        const dataHex = '0x6927e990021d23b1eb7b8789f6a6feaf98fe104bb0cf8259421b79f9a34222b0';
        const signature = {
            v: 27,
            r: '0x61a3ed31b43c8780e905a260a35faefcc527be7516aa11c0256729b5b351bc33',
            s: '0x40349190569279751135161d22529dc25add4f6069af05be04cacbda2ace2254',
        };
        const address = '0x5409ed021d9299bf6814279a6a1411a7e866a631';
        describe('should throw if passed a malformed signature', () => {
            it('malformed v', async () => {
                const malformedSignature = {
                    v: 34,
                    r: signature.r,
                    s: signature.s,
                };
                expect(zeroEx.exchange.isValidSignatureAsync(dataHex, malformedSignature, address))
                    .to.be.rejected;
            });
            it('r lacks 0x prefix', () => {
                const malformedR = signature.r.replace('0x', '');
                const malformedSignature = {
                    v: signature.v,
                    r: malformedR,
                    s: signature.s,
                };
                expect(zeroEx.exchange.isValidSignatureAsync(dataHex, malformedSignature, address))
                    .to.be.rejected;
            });
            it('r is too short', () => {
                const malformedR = signature.r.substr(10);
                const malformedSignature = {
                    v: signature.v,
                    r: malformedR,
                    s: signature.s.replace('0', 'z'),
                };
                expect(zeroEx.exchange.isValidSignatureAsync(dataHex, malformedSignature, address))
                    .to.be.rejected;
            });
            it('s is not hex', () => {
                const malformedS = signature.s.replace('0', 'z');
                const malformedSignature = {
                    v: signature.v,
                    r: signature.r,
                    s: malformedS,
                };
                expect(zeroEx.exchange.isValidSignatureAsync(dataHex, malformedSignature, address))
                    .to.be.rejected;
            });
        });
        it('should return false if the data doesn\'t pertain to the signature & address', async () => {
            const isValid = await zeroEx.exchange.isValidSignatureAsync('0x0', signature, address);
            expect(isValid).to.be.false;
        });
        it('should return false if the address doesn\'t pertain to the signature & dataHex', async () => {
            const validUnrelatedAddress = '0x8b0292B11a196601eD2ce54B665CaFEca0347D42';
            const isValid = await zeroEx.exchange.isValidSignatureAsync(dataHex, signature, validUnrelatedAddress);
            expect(isValid).to.be.false;
        });
        it('should return false if the signature doesn\'t pertain to the dataHex & address', async () => {
            const wrongSignature = {...signature, v: 28};
            const isValid = await zeroEx.exchange.isValidSignatureAsync(dataHex, wrongSignature, address);
            expect(isValid).to.be.false;
        });
        it('should return true if the signature does pertain to the dataHex & address', async () => {
            const isValid = await zeroEx.exchange.isValidSignatureAsync(dataHex, signature, address);
            expect(isValid).to.be.true;
        });
    });
    describe('#fillOrderAsync', () => {
        let tokens: Token[];
        const setBalance = async (toAddress: string,
                                    amountInBaseUnits: BigNumber.BigNumber|number,
                                    symbol: string) => {
            const amount = _.isNumber(amountInBaseUnits) ? new BigNumber(amountInBaseUnits) : amountInBaseUnits;
            const token = _.find(tokens, {symbol});
            if (_.isUndefined(token)) {
                throw new Error(`Token ${symbol} not found`);
            } else {
                await zeroEx.token.transferAsync(token.address, userAddresses[0], toAddress, amount);
            }
        };
        const setAllowance = async (ownerAddress: string,
                                    amountInBaseUnits: BigNumber.BigNumber|number,
                                    symbol: string) => {
            const amount = _.isNumber(amountInBaseUnits) ? new BigNumber(amountInBaseUnits) : amountInBaseUnits;
            const token = _.find(tokens, {symbol});
            if (_.isUndefined(token)) {
                throw new Error(`Token ${symbol} not found`);
            } else {
                await zeroEx.token.setProxyAllowanceAsync(token.address, ownerAddress, amount);
            }
        };
        before('fetch tokens', async () => {
            tokens = await zeroEx.tokenRegistry.getTokensAsync();
        });
        describe('failed fills', () => {
            it('should throw when the fill amount is zero', async () => {
                const signedOrder = await createSignedOrder(zeroEx, tokens, 5, 'MLN', 5, 'GNT');
                const fillAmount = new BigNumber(0);
                expect(zeroEx.exchange.fillOrderAsync(signedOrder, fillAmount))
                    .to.be.rejectedWith('This order has already been filled or cancelled');
            });
        });
        describe('successful fills', () => {
            afterEach('reset default account', () => {
               web3.eth.defaultAccount = userAddresses[0];
            });
            it('should fill the valid order', async () => {
                await setAllowance(userAddresses[0], 5, 'MLN');
                await setBalance(userAddresses[1], 5, 'GNT');
                await setAllowance(userAddresses[1], 5, 'GNT');
                const signedOrder = await createSignedOrder(zeroEx, tokens, 5, 'MLN', 5, 'GNT');
                const fillAmount = new BigNumber(5);
                web3.eth.defaultAccount = userAddresses[1];
                await zeroEx.exchange.fillOrderAsync(signedOrder, fillAmount);
            });
        });
    });
});