Search in sources :

Example 1 with SystemExits

use of com.hedera.services.utils.SystemExits in project hedera-services by hashgraph.

the class SignedStateBalancesExporterTest method setUp.

@BeforeEach
void setUp() throws ConstructableRegistryException {
    ConstructableRegistry.registerConstructable(new ClassConstructorPair(MerkleAccount.class, MerkleAccount::new));
    thisNodeAccount = MerkleAccountFactory.newAccount().balance(thisNodeBalance).get();
    anotherNodeAccount = MerkleAccountFactory.newAccount().balance(anotherNodeBalance).get();
    firstNonNodeAccount = MerkleAccountFactory.newAccount().balance(firstNonNodeAccountBalance).get();
    secondNonNodeAccount = MerkleAccountFactory.newAccount().balance(secondNonNodeAccountBalance).tokens(theToken, theDeletedToken, theMissingToken).get();
    deletedAccount = MerkleAccountFactory.newAccount().deleted(true).get();
    accounts.put(fromAccountId(thisNode), thisNodeAccount);
    accounts.put(fromAccountId(anotherNode), anotherNodeAccount);
    accounts.put(fromAccountId(firstNonNode), firstNonNodeAccount);
    accounts.put(fromAccountId(secondNonNode), secondNonNodeAccount);
    accounts.put(fromAccountId(deleted), deletedAccount);
    token = mock(MerkleToken.class);
    given(token.isDeleted()).willReturn(false);
    deletedToken = mock(MerkleToken.class);
    given(deletedToken.isDeleted()).willReturn(true);
    tokens.put(fromTokenId(theToken), token);
    tokens.put(fromTokenId(theDeletedToken), deletedToken);
    tokenRels.put(fromAccountTokenRel(secondNonNode, theToken), new MerkleTokenRelStatus(secondNonNodeTokenBalance, false, true, false));
    tokenRels.put(fromAccountTokenRel(secondNonNode, theDeletedToken), new MerkleTokenRelStatus(secondNonNodeDeletedTokenBalance, false, true, false));
    assurance = mock(DirectoryAssurance.class);
    properties = mock(PropertySource.class);
    given(properties.getLongProperty("ledger.totalTinyBarFloat")).willReturn(ledgerFloat);
    var firstNodeAddress = mock(Address.class);
    given(firstNodeAddress.getMemo()).willReturn("0.0.3");
    var secondNodeAddress = mock(Address.class);
    given(secondNodeAddress.getMemo()).willReturn("0.0.4");
    var book = mock(AddressBook.class);
    given(book.getSize()).willReturn(2);
    given(book.getAddress(0)).willReturn(firstNodeAddress);
    given(book.getAddress(1)).willReturn(secondNodeAddress);
    state = mock(ServicesState.class);
    given(state.getAccountFromNodeId(nodeId)).willReturn(thisNode);
    given(state.tokens()).willReturn(tokens);
    given(state.accounts()).willReturn(accounts);
    given(state.tokenAssociations()).willReturn(tokenRels);
    given(state.addressBook()).willReturn(book);
    signer = mock(UnaryOperator.class);
    given(signer.apply(fileHash)).willReturn(sig);
    systemExits = mock(SystemExits.class);
    subject = new SignedStateBalancesExporter(systemExits, properties, signer, dynamicProperties);
    sigFileWriter = mock(SigFileWriter.class);
    hashReader = mock(FileHashReader.class);
    subject.sigFileWriter = sigFileWriter;
    subject.hashReader = hashReader;
}
Also used : MerkleToken(com.hedera.services.state.merkle.MerkleToken) MerkleAccount(com.hedera.services.state.merkle.MerkleAccount) PropertySource(com.hedera.services.context.properties.PropertySource) MerkleTokenRelStatus(com.hedera.services.state.merkle.MerkleTokenRelStatus) ServicesState(com.hedera.services.ServicesState) SystemExits(com.hedera.services.utils.SystemExits) UnaryOperator(java.util.function.UnaryOperator) ClassConstructorPair(com.swirlds.common.constructable.ClassConstructorPair) BeforeEach(org.junit.jupiter.api.BeforeEach)

Aggregations

ServicesState (com.hedera.services.ServicesState)1 PropertySource (com.hedera.services.context.properties.PropertySource)1 MerkleAccount (com.hedera.services.state.merkle.MerkleAccount)1 MerkleToken (com.hedera.services.state.merkle.MerkleToken)1 MerkleTokenRelStatus (com.hedera.services.state.merkle.MerkleTokenRelStatus)1 SystemExits (com.hedera.services.utils.SystemExits)1 ClassConstructorPair (com.swirlds.common.constructable.ClassConstructorPair)1 UnaryOperator (java.util.function.UnaryOperator)1 BeforeEach (org.junit.jupiter.api.BeforeEach)1