TransactionService.spec.ts

/*
 * Copyright 2020 NEM
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { ChronoUnit } from '@js-joda/core';
import { expect } from 'chai';
import { EMPTY, firstValueFrom, from, of as observableOf } from 'rxjs';
import { deepEqual, instance, mock, when } from 'ts-mockito';
import { IListener, ReceiptRepository, TransactionRepository } from '../../src/infrastructure';
import { UInt64 } from '../../src/model';
import { Account, Address } from '../../src/model/account';
import { PlainMessage } from '../../src/model/message';
import { NetworkType } from '../../src/model/network';
import {
    AggregateTransaction,
    Deadline,
    HashLockTransaction,
    TransactionAnnounceResponse,
    TransactionStatusError,
    TransferTransaction,
} from '../../src/model/transaction';
import { TransactionService } from '../../src/service';
import { NetworkCurrencyLocal } from '../model/mosaic/Currency.spec';

/**
 * Unit test of TransactionService
 */
describe('TransactionService', () => {
    const generationHash = '82DB2528834C9926F0FCCE042466B24A266F5B685CB66D2869AF6648C043E950';
    const account = Account.generateNewAccount(NetworkType.TEST_NET);
    const epochAdjustment = 1573430400;
    const transferTransaction = TransferTransaction.create(
        Deadline.create(epochAdjustment, 1, ChronoUnit.HOURS),
        Address.createFromRawAddress('TATNE7Q5BITMUTRRN6IB4I7FLSDRDWZA37JGO5Q'),
        [],
        PlainMessage.create('test-message'),
        NetworkType.TEST_NET,
    );

    const aggregateCompleteTransaction = AggregateTransaction.createComplete(
        Deadline.create(epochAdjustment),
        [transferTransaction.toAggregate(account.publicAccount)],
        NetworkType.TEST_NET,
        [],
    );

    const aggregateBondedTransaction = AggregateTransaction.createBonded(
        Deadline.create(epochAdjustment),
        [transferTransaction.toAggregate(account.publicAccount)],
        NetworkType.TEST_NET,
        [],
    );

    const hashLockTransaction = HashLockTransaction.create(
        Deadline.create(epochAdjustment),
        NetworkCurrencyLocal.createRelative(10),
        UInt64.fromUint(10000),
        account.sign(aggregateBondedTransaction, generationHash),
        NetworkType.TEST_NET,
    );

    let transactionRepositoryMock: TransactionRepository;
    let mockedReceiptRepository: ReceiptRepository;
    let listener: IListener;

    before(() => {
        transactionRepositoryMock = mock();
        mockedReceiptRepository = mock();
        listener = mock();
    });

    it('announce transaction when reject promise', async () => {
        const signedTransaction = account.sign(transferTransaction, generationHash);
        when(transactionRepositoryMock.announce(deepEqual(signedTransaction))).thenReturn(from(Promise.reject(Error('break this!'))));
        const service = new TransactionService(instance(transactionRepositoryMock), instance(mockedReceiptRepository));
        try {
            await firstValueFrom(service.announce(signedTransaction, instance(listener)));
            expect(false).true;
        } catch (e) {
            expect(e.message).eq('break this!');
        }
    });

    it('announce transaction when exception', async () => {
        const signedTransaction = account.sign(transferTransaction, generationHash);
        when(transactionRepositoryMock.announce(deepEqual(signedTransaction))).thenThrow(Error('break this!'));
        const service = new TransactionService(instance(transactionRepositoryMock), instance(mockedReceiptRepository));
        try {
            await firstValueFrom(service.announce(signedTransaction, instance(listener)));
            expect(false).true;
        } catch (e) {
            expect(e.message).eq('break this!');
        }
    });

    it('announce when valid transaction', async () => {
        const signedTransaction = account.sign(transferTransaction, generationHash);

        const transactionAnnounceResponse = new TransactionAnnounceResponse('Some Message');

        when(transactionRepositoryMock.announce(deepEqual(signedTransaction))).thenReturn(observableOf(transactionAnnounceResponse));

        when(listener.confirmed(deepEqual(account.address), deepEqual(signedTransaction.hash))).thenReturn(
            observableOf(transferTransaction),
        );

        when(listener.status(deepEqual(account.address), signedTransaction.hash)).thenReturn(EMPTY);

        const service = new TransactionService(instance(transactionRepositoryMock), instance(mockedReceiptRepository));

        const announcedTransaction = service.announce(signedTransaction, instance(listener));

        const transaction = await firstValueFrom(announcedTransaction);
        expect(transaction).to.be.equal(transferTransaction);
    });

    it('announce when status error', async () => {
        const signedTransaction = account.sign(transferTransaction, generationHash);

        const transactionAnnounceResponse = new TransactionAnnounceResponse('Some Message');

        when(transactionRepositoryMock.announce(deepEqual(signedTransaction))).thenReturn(observableOf(transactionAnnounceResponse));

        when(listener.confirmed(deepEqual(account.address), deepEqual(signedTransaction.hash))).thenReturn(EMPTY);
        const statusError = new TransactionStatusError(
            account.address,
            signedTransaction.hash,
            'Some Error',
            Deadline.create(epochAdjustment),
        );
        when(listener.status(deepEqual(account.address), signedTransaction.hash)).thenReturn(observableOf(statusError));

        const service = new TransactionService(instance(transactionRepositoryMock), instance(mockedReceiptRepository));

        const announcedTransaction = service.announce(signedTransaction, instance(listener));

        try {
            await firstValueFrom(announcedTransaction);
        } catch (e) {
            expect(e.message).to.be.equal('Some Error');
        }
    });

    it('Basic announceAggregateBonded when valid transaction', async () => {
        const signedTransaction = account.sign(aggregateCompleteTransaction, generationHash);

        const transactionAnnounceResponse = new TransactionAnnounceResponse('Some Message');

        when(transactionRepositoryMock.announceAggregateBonded(deepEqual(signedTransaction))).thenReturn(
            observableOf(transactionAnnounceResponse),
        );

        when(listener.aggregateBondedAdded(deepEqual(account.address), deepEqual(signedTransaction.hash))).thenReturn(
            observableOf(aggregateCompleteTransaction),
        );
        when(listener.status(deepEqual(account.address), signedTransaction.hash)).thenReturn(EMPTY);

        const service = new TransactionService(instance(transactionRepositoryMock), instance(mockedReceiptRepository));

        const announcedTransaction = service.announceAggregateBonded(signedTransaction, instance(listener));

        const transaction = await firstValueFrom(announcedTransaction);
        expect(transaction).to.be.equal(aggregateCompleteTransaction);
    });

    it('announceAggregateBonded when status error', async () => {
        const signedTransaction = account.sign(aggregateCompleteTransaction, generationHash);

        const transactionAnnounceResponse = new TransactionAnnounceResponse('Some Message');

        when(transactionRepositoryMock.announceAggregateBonded(deepEqual(signedTransaction))).thenReturn(
            observableOf(transactionAnnounceResponse),
        );

        when(listener.aggregateBondedAdded(deepEqual(account.address), deepEqual(signedTransaction.hash))).thenReturn(EMPTY);
        const statusError = new TransactionStatusError(
            account.address,
            signedTransaction.hash,
            'Some Error',
            Deadline.create(epochAdjustment),
        );
        when(listener.status(deepEqual(account.address), signedTransaction.hash)).thenReturn(observableOf(statusError));

        const service = new TransactionService(instance(transactionRepositoryMock), instance(mockedReceiptRepository));

        const announcedTransaction = service.announceAggregateBonded(signedTransaction, instance(listener));

        try {
            await firstValueFrom(announcedTransaction);
        } catch (e) {
            expect(e.message).to.be.equal('Some Error');
        }
    });

    it('announceHashLockAggregateBonded when ok', async () => {
        const aggregateBondedSignedTransaction = account.sign(aggregateBondedTransaction, generationHash);

        const hashLockSignedTransaction = account.sign(hashLockTransaction, generationHash);

        const transactionAnnounceResponse = new TransactionAnnounceResponse('Some Message');

        when(transactionRepositoryMock.announceAggregateBonded(deepEqual(aggregateBondedSignedTransaction))).thenReturn(
            observableOf(transactionAnnounceResponse),
        );

        when(transactionRepositoryMock.announce(deepEqual(hashLockSignedTransaction))).thenReturn(
            observableOf(transactionAnnounceResponse),
        );

        when(listener.confirmed(deepEqual(account.address), deepEqual(hashLockSignedTransaction.hash))).thenReturn(
            observableOf(hashLockTransaction),
        );

        when(listener.aggregateBondedAdded(deepEqual(account.address), deepEqual(aggregateBondedSignedTransaction.hash))).thenReturn(
            observableOf(aggregateBondedTransaction),
        );

        when(listener.status(deepEqual(account.address), hashLockSignedTransaction.hash)).thenReturn(EMPTY);

        const service = new TransactionService(instance(transactionRepositoryMock), instance(mockedReceiptRepository));

        const announcedTransaction = service.announceHashLockAggregateBonded(
            hashLockSignedTransaction,
            aggregateBondedSignedTransaction,
            instance(listener),
        );

        const transaction = await firstValueFrom(announcedTransaction);
        expect(transaction).to.be.equal(aggregateBondedTransaction);
    });
});