PercentagePostingRuleTest.java

package org.multibit.mbm.core.accounting.rules;

import org.junit.Test;
import org.multibit.mbm.core.accounting.Account;
import org.multibit.mbm.core.accounting.DefaultEntry;
import org.multibit.mbm.core.accounting.EntryFactory;
import org.multibit.mbm.core.accounting.Transaction;

import static org.hamcrest.CoreMatchers.equalTo;
import static org.junit.Assert.assertThat;

public class PercentagePostingRuleTest {

  private PercentagePostingRule<DefaultEntry> testObject = null;

  @Test
  public void testApply() {

    // Configure supporting objects
    Account<DefaultEntry> sourceAccount = new Account<DefaultEntry>();
    Account<DefaultEntry> targetAccount = new Account<DefaultEntry>();
    // TODO Make this a real MemoAccount
    Account<DefaultEntry> memoAccount = new Account<DefaultEntry>();

    // Use the target account for triggering the memo
    testObject = new PercentagePostingRule<DefaultEntry>(memoAccount,0.9);
    targetAccount.addPostingRule(testObject);

    DefaultEntry sourceEntry = EntryFactory.INSTANCE.buildDefaultEntry(sourceAccount, -10L);
    DefaultEntry targetEntry = EntryFactory.INSTANCE.buildDefaultEntry(targetAccount, 10L);

    // Perform the test (Account will notify PostingRule)
    // Can only get an Entry into an Account through a Transaction
    Transaction<DefaultEntry> transaction = new Transaction<DefaultEntry>(sourceEntry, targetEntry);
    transaction.commit();

    // Verify the outcome
    assertThat(memoAccount.getBalance(), equalTo(9L));

  }

  @Test
  public void testReverse() {

    // Configure supporting objects
    Account<DefaultEntry> sourceAccount = new Account<DefaultEntry>();
    Account<DefaultEntry> targetAccount = new Account<DefaultEntry>();
    // TODO Make this a real MemoAccount
    Account<DefaultEntry> memoAccount = new Account<DefaultEntry>();

    // Use the target account for triggering the memo
    testObject = new PercentagePostingRule<DefaultEntry>(memoAccount,0.9);
    targetAccount.addPostingRule(testObject);

    DefaultEntry sourceEntry = EntryFactory.INSTANCE.buildDefaultEntry(sourceAccount, -10L);
    DefaultEntry targetEntry = EntryFactory.INSTANCE.buildDefaultEntry(targetAccount, 10L);

    // Perform the test (Account will notify PostingRule)
    // Can only get an Entry into an Account through a Transaction
    Transaction<DefaultEntry> transaction = new Transaction<DefaultEntry>(sourceEntry, targetEntry);
    transaction.commit();

    // Perform the test
    testObject.reverse(targetEntry);

    // Verify the outcome
    assertThat(memoAccount.getBalance(), equalTo(0L));

  }

}