test_MoneyManager.cpp

浏览该文件的文档.

/*
 * test_MoneyManager.cpp
 *
 *  Created on: 2013-3-11
 *      Author: fasiondog
 */

#ifdef TEST_ALL_IN_ONE
    #include <boost/test/unit_test.hpp>
#else
    #define BOOST_TEST_MODULE test_hikyuu_trade_sys_suite
    #include <boost/test/unit_test.hpp>
#endif

#include <hikyuu/StockManager.h>
#include <hikyuu/trade_manage/crt/crtTM.h>
#include <hikyuu/trade_sys/moneymanager/MoneyManagerBase.h>

using namespace hku;

class MoneyManagerTest: public MoneyManagerBase {
public:
    MoneyManagerTest() : MoneyManagerBase("MoneyManagerTest") { m_x = 0; }
    virtual ~MoneyManagerTest() {}

    TradeManagerPtr getTM() const { return m_tm; }
    int getX() const { return m_x; }
    void setX(int x) { m_x = x; }

    virtual size_t _getBuyNumber(const Datetime& datetime, const Stock& stock,
                price_t price, price_t risk, SystemPart from) {
        return 0;
    }

    virtual void _reset() {
        m_x = 0;
    }

    virtual MoneyManagerPtr _clone() {
        MoneyManagerTest *p = new MoneyManagerTest;
        p->m_x = m_x;
        return MoneyManagerPtr(p);
    }

private:
    int m_x;

};


BOOST_AUTO_TEST_CASE( test_MoneyManager ) {
    StockManager& sm = StockManager::instance();
    Stock stock = sm["sh000001"];
    TradeManagerPtr tm = crtTM();

    MoneyManagerPtr p(new MoneyManagerTest);
    MoneyManagerTest *p_src = (MoneyManagerTest *)p.get();
    BOOST_CHECK(p->name() == "MoneyManagerTest");
    BOOST_CHECK(p_src->getTM() == TradeManagerPtr());
    p->setTM(tm);
    BOOST_CHECK(p_src->getTM() == tm);
    BOOST_CHECK(p->getBuyNumber(Datetime(200001010000), stock, 10.0, 10.0, PART_SIGNAL) == 0);
    BOOST_CHECK(p->getSellNumber(Datetime(200001010000), stock, 10.0, 10.0, PART_SIGNAL) == Null<size_t>());
    BOOST_CHECK(p_src->getX() == 0);
    p_src->setX(10);
    BOOST_CHECK(p_src->getX() == 10);
    p->reset();
    BOOST_CHECK(p_src->getX() == 0);

    p_src->setX(10);
    MoneyManagerPtr p_clone = p->clone();
    BOOST_CHECK(p != p_clone);
    p_src = (MoneyManagerTest *)p_clone.get();
    BOOST_CHECK(p->name() == "MoneyManagerTest");
    //BOOST_CHECK(p_src->getTM() == tm);
    BOOST_CHECK(p_src->getX() == 10);
}