/* ===========================================================
* TradeManager : a application to trade strategies for the Java(tm) platform
* ===========================================================
*
* (C) Copyright 2011-2011, by Simon Allen and Contributors.
*
* Project Info: org.trade
*
* This library is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation; either version 2.1 of the License, or
* (at your option) any later version.
*
* This library is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
* License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
* USA.
*
* [Java is a trademark or registered trademark of Oracle, Inc.
* in the United States and other countries.]
*
* (C) Copyright 2011-2011, by Simon Allen and Contributors.
*
* Original Author: Simon Allen;
* Contributor(s): -;
*
* Changes
* -------
*
*/
package org.trade.broker.request;
import static org.junit.Assert.*;
import org.junit.After;
import org.junit.AfterClass;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.TestName;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.trade.core.dao.Aspect;
import org.trade.core.dao.Aspects;
import org.trade.core.factory.ClassFactory;
import org.trade.core.util.TradingCalendar;
import org.trade.dictionary.valuetype.AccountType;
import org.trade.dictionary.valuetype.Currency;
import org.trade.persistent.PersistentModel;
import org.trade.persistent.dao.Account;
import org.trade.persistent.dao.Portfolio;
import org.trade.persistent.dao.PortfolioAccount;
import org.trade.ui.TradeAppLoadConfig;
/**
*/
public class TWSFinancialAccountRequestTest {
private final static Logger _log = LoggerFactory.getLogger(TWSFinancialAccountRequestTest.class);
@Rule
public TestName name = new TestName();
private PersistentModel m_tradePersistentModel = null;
/**
* Method setUpBeforeClass.
*
* @throws java.lang.Exception
*/
@BeforeClass
public static void setUpBeforeClass() throws Exception {
}
/**
* Method setUp.
*
* @throws java.lang.Exception
*/
@Before
public void setUp() throws Exception {
TradeAppLoadConfig.loadAppProperties();
m_tradePersistentModel = (PersistentModel) ClassFactory.getServiceForInterface(PersistentModel._persistentModel,
this);
}
/**
* Method tearDown.
*
* @throws java.lang.Exception
*/
@After
public void tearDown() throws Exception {
Aspects portfolioAccounts = m_tradePersistentModel.findAspectsByClassName(PortfolioAccount.class.getName());
for (Aspect aspect : portfolioAccounts.getAspect()) {
m_tradePersistentModel.removeAspect(aspect);
}
Aspects accounts = m_tradePersistentModel.findAspectsByClassName(Account.class.getName());
for (Aspect aspect : accounts.getAspect()) {
m_tradePersistentModel.removeAspect(aspect);
}
}
/**
* Method tearDownAfterClass.
*
* @throws java.lang.Exception
*/
@AfterClass
public static void tearDownAfterClass() throws Exception {
}
@Test
public void testAliasEmptyRequest() {
try {
final TWSAccountAliasRequest request = new TWSAccountAliasRequest();
// String xml =
// "<?xml version=\"1.0\"
// encoding=\"UTF-8\"?><ListOfAccountAliases/>";
// ByteArrayInputStream inputSource = new ByteArrayInputStream(
// xml.getBytes("utf-8"));
// final Aspects aspects = (Aspects) request.fromXML(inputSource);
final Aspects aspects = (Aspects) request.fromXML(Thread.currentThread().getContextClassLoader()
.getResourceAsStream("org/trade/broker/request/aliasesEmpty.xml"));
for (Aspect aspect : aspects.getAspect()) {
Account item = (Account) aspect;
Account account = m_tradePersistentModel.findAccountByNumber(item.getAccountNumber());
account.setAlias(item.getAlias());
m_tradePersistentModel.persistAspect(account);
}
} catch (Exception | AssertionError ex) {
String msg = "Error running " + name.getMethodName() + " msg: " + ex.getMessage();
_log.error(msg);
fail(msg);
}
}
@Test
public void testAliasRequest() {
try {
final TWSAccountAliasRequest request = new TWSAccountAliasRequest();
final Aspects aspects = (Aspects) request.fromXML(Thread.currentThread().getContextClassLoader()
.getResourceAsStream("org/trade/broker/request/aliases.xml"));
for (Aspect aspect : aspects.getAspect()) {
Account item = (Account) aspect;
Account account = m_tradePersistentModel.findAccountByNumber(item.getAccountNumber());
if (null == account) {
account = new Account(item.getAccountNumber(), item.getAccountNumber(), Currency.USD,
AccountType.INDIVIDUAL);
}
account.setAlias(item.getAlias());
account.setLastUpdateDate(TradingCalendar.getDateTimeNowMarketTimeZone());
m_tradePersistentModel.persistAspect(account);
}
} catch (Exception | AssertionError ex) {
String msg = "Error running " + name.getMethodName() + " msg: " + ex.getMessage();
_log.error(msg);
fail(msg);
}
}
@Test
public void testAliasRequest1() {
try {
final TWSAccountAliasRequest request = new TWSAccountAliasRequest();
final Aspects aspects = (Aspects) request.fromXML(Thread.currentThread().getContextClassLoader()
.getResourceAsStream("org/trade/broker/request/aliases1.xml"));
for (Aspect aspect : aspects.getAspect()) {
Account item = (Account) aspect;
Account account = m_tradePersistentModel.findAccountByNumber(item.getAccountNumber());
if (null == account) {
account = new Account(item.getAccountNumber(), item.getAccountNumber(), Currency.USD,
AccountType.INDIVIDUAL);
}
account.setAlias(item.getAlias());
account.setLastUpdateDate(TradingCalendar.getDateTimeNowMarketTimeZone());
m_tradePersistentModel.persistAspect(account);
}
} catch (Exception | AssertionError ex) {
String msg = "Error running " + name.getMethodName() + " msg: " + ex.getMessage();
_log.error(msg);
fail(msg);
}
}
@Test
public void testAllocationEmptyRequest() {
try {
final TWSAllocationRequest request = new TWSAllocationRequest();
final Aspects aspects = (Aspects) request.fromXML(Thread.currentThread().getContextClassLoader()
.getResourceAsStream("org/trade/broker/request/allocationEmpty.xml"));
for (Aspect aspect : aspects.getAspect()) {
m_tradePersistentModel.persistPortfolio((Portfolio) aspect);
}
} catch (Exception | AssertionError ex) {
String msg = "Error running " + name.getMethodName() + " msg: " + ex.getMessage();
_log.error(msg);
fail(msg);
}
}
@Test
public void testAllocationRequest() {
try {
final TWSAllocationRequest request = new TWSAllocationRequest();
final Aspects aspects = (Aspects) request.fromXML(Thread.currentThread().getContextClassLoader()
.getResourceAsStream("org/trade/broker/request/allocation.xml"));
for (Aspect aspect : aspects.getAspect()) {
m_tradePersistentModel.persistPortfolio((Portfolio) aspect);
}
} catch (Exception | AssertionError ex) {
String msg = "Error running " + name.getMethodName() + " msg: " + ex.getMessage();
_log.error(msg);
fail(msg);
}
}
@Test
public void testAllocationRequest1() {
try {
final TWSAllocationRequest request = new TWSAllocationRequest();
final Aspects aspects = (Aspects) request.fromXML(Thread.currentThread().getContextClassLoader()
.getResourceAsStream("org/trade/broker/request/allocation1.xml"));
for (Aspect aspect : aspects.getAspect()) {
m_tradePersistentModel.persistPortfolio((Portfolio) aspect);
}
} catch (Exception | AssertionError ex) {
String msg = "Error running " + name.getMethodName() + " msg: " + ex.getMessage();
_log.error(msg);
fail(msg);
}
}
@Test
public void testAllocationRequestNew() {
try {
Aspects aspects = new Aspects();
Portfolio portfolio = new Portfolio("pf_eq_daily", "pf_eq_daily");
Account account = new Account("DU12345", "DU12345", Currency.USD, AccountType.INDIVIDUAL);
PortfolioAccount portfolioAccount = new PortfolioAccount(portfolio, account);
portfolio.getPortfolioAccounts().add(portfolioAccount);
aspects.add(portfolio);
for (Aspect aspect : aspects.getAspect()) {
m_tradePersistentModel.persistPortfolio((Portfolio) aspect);
}
} catch (Exception | AssertionError ex) {
String msg = "Error running " + name.getMethodName() + " msg: " + ex.getMessage();
_log.error(msg);
fail(msg);
}
}
@Test
public void testGroupEmptyRequest() {
try {
final TWSGroupRequest request = new TWSGroupRequest();
final Aspects aspects = (Aspects) request.fromXML(Thread.currentThread().getContextClassLoader()
.getResourceAsStream("org/trade/broker/request/groupsEmpty.xml"));
for (Aspect aspect : aspects.getAspect()) {
m_tradePersistentModel.persistPortfolio((Portfolio) aspect);
}
} catch (Exception | AssertionError ex) {
String msg = "Error running " + name.getMethodName() + " msg: " + ex.getMessage();
_log.error(msg);
fail(msg);
}
}
@Test
public void testGroupRequest() {
try {
final TWSGroupRequest request = new TWSGroupRequest();
final Aspects aspects = (Aspects) request.fromXML(Thread.currentThread().getContextClassLoader()
.getResourceAsStream("org/trade/broker/request/groups.xml"));
for (Aspect aspect : aspects.getAspect()) {
m_tradePersistentModel.persistPortfolio((Portfolio) aspect);
}
} catch (Exception | AssertionError ex) {
String msg = "Error running " + name.getMethodName() + " msg: " + ex.getMessage();
_log.error(msg);
fail(msg);
}
}
@Test
public void testGroupRequest1() {
try {
final TWSGroupRequest request = new TWSGroupRequest();
final Aspects aspects = (Aspects) request.fromXML(Thread.currentThread().getContextClassLoader()
.getResourceAsStream("org/trade/broker/request/groups1.xml"));
for (Aspect aspect : aspects.getAspect()) {
m_tradePersistentModel.persistPortfolio((Portfolio) aspect);
}
} catch (Exception | AssertionError ex) {
String msg = "Error running " + name.getMethodName() + " msg: " + ex.getMessage();
_log.error(msg);
fail(msg);
}
}
}