test_ProfitGoal.cpp

浏览该文件的文档.

/*
 * test_ProfitGoal.cpp
 *
 *  Created on: 2013-3-21
 *      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_sys/profitgoal/ProfitGoalBase.h>

using namespace hku;

class ProfitGoalTest: public ProfitGoalBase {
public:
    ProfitGoalTest(): ProfitGoalBase("ProfitGoalTest"), m_x(0) {}
    virtual ~ProfitGoalTest() {}

    virtual price_t getGoal(const Datetime& datetime, price_t price) {
        return m_x < 10 ? 0.0 : 1.0;
    }

    virtual void _reset() {
        m_x = 0;
    }

    virtual ProfitGoalPtr _clone() {
        ProfitGoalTest *p = new ProfitGoalTest;
        p->m_x = m_x;
        return ProfitGoalPtr(p);
    }

    virtual void _calculate() {}

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

private:
    int m_x;
};

BOOST_AUTO_TEST_CASE( test_ProfitGoal) {
    ProfitGoalPtr p(new ProfitGoalTest);
    BOOST_CHECK(p->name() == "ProfitGoalTest");
    BOOST_CHECK(p->getGoal(Datetime(200101010000), 1.0) == 0.0);
    ProfitGoalTest *p_src = (ProfitGoalTest *)p.get();
    BOOST_CHECK(p_src->getX() == 0);

    p_src->setX(10);
    BOOST_CHECK(p->getGoal(Datetime(200101010000), 1.0) == 1.0);
    BOOST_CHECK(p_src->getX() == 10);
    p->reset();
    BOOST_CHECK(p_src->getX() == 0);

    p_src->setX(10);
    ProfitGoalPtr p_clone = p->clone();
    BOOST_CHECK(p_clone->name() == "ProfitGoalTest");
    p_src = (ProfitGoalTest *)p_clone.get();
    BOOST_CHECK(p_src->getX() == 10);
    BOOST_CHECK(p != p_clone);
}