test_StockManager.cpp

浏览该文件的文档.

/*
 * test_StockManager.cpp
 *
 *  Created on: 2011-11-10
 *      Author: fasiondog
 */

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

#include <boost/filesystem.hpp>
#include <hikyuu/StockManager.h>
#include <hikyuu/utilities/util.h>
#include <hikyuu/Log.h>

using namespace hku;
using namespace boost::filesystem;

BOOST_AUTO_TEST_CASE( test_StockManager_size ) {
    StockManager& sm = StockManager::instance();
    BOOST_CHECK(sm.size() == 4729);
}

BOOST_AUTO_TEST_CASE( test_StockManager_getStock ) {
    StockManager& sm = StockManager::instance();
    Stock stock;
    Stock null_stock = Null<Stock>();

    BOOST_CHECK(sm.getStock("z000001") == null_stock);

    BOOST_CHECK(sm.getStock("sh1000001") == null_stock);

    stock = sm.getStock("sh000001");
    BOOST_CHECK(stock.market() == "SH");
    BOOST_CHECK(stock.code() == "000001");
    BOOST_CHECK(stock.name() == HKU_STR("上证指数"));
    BOOST_CHECK(stock.type() == 2);
    BOOST_CHECK(stock.valid() == true);
    BOOST_CHECK(stock.startDatetime() == Datetime(199012190000));
    BOOST_CHECK(stock.lastDatetime() == Null<Datetime>());
    BOOST_CHECK(stock.tick() == 0.001);
    BOOST_CHECK(stock.tickValue() == 0.001);
    BOOST_CHECK(stock.unit() == 1.0);
    BOOST_CHECK(stock.precision() == 3);
    BOOST_CHECK(stock.atom() == 1);
    BOOST_CHECK(stock.minTradeNumber() == 1);
    BOOST_CHECK(stock.maxTradeNumber() == 1000000);
}

BOOST_AUTO_TEST_CASE( test_StockManager_getMarketInfo ) {
    StockManager& sm = StockManager::instance();

    BOOST_CHECK(Null<MarketInfo>() == sm.getMarketInfo("XXX"));
    BOOST_CHECK(Null<MarketInfo>() == sm.getMarketInfo(""));

    MarketInfo marketInfo = sm.getMarketInfo("SH");
    BOOST_CHECK(marketInfo != Null<MarketInfo>());
    BOOST_CHECK(marketInfo.market() == "SH");
    BOOST_CHECK(marketInfo.code() == "000001");
    BOOST_CHECK(marketInfo.name() == HKU_STR("上海证劵交易所"));
    BOOST_CHECK(marketInfo.description() == HKU_STR("上海市场"));
    BOOST_CHECK(marketInfo.lastDate() == Datetime(201112060000L));

    marketInfo = sm.getMarketInfo("sh");
    BOOST_CHECK(marketInfo != Null<MarketInfo>());
    BOOST_CHECK(marketInfo.market() == "SH");


    marketInfo = sm.getMarketInfo("Sh");
    BOOST_CHECK(marketInfo != Null<MarketInfo>());
    BOOST_CHECK(marketInfo.market() == "SH");
}

BOOST_AUTO_TEST_CASE( test_StockManager_getStockTypeInfo ) {
    StockManager& sm = StockManager::instance();
    StockTypeInfo stockTypeInfo;

    BOOST_CHECK(sm.getStockTypeInfo(999) == Null<StockTypeInfo>());

    stockTypeInfo = sm.getStockTypeInfo(0);
    BOOST_CHECK(stockTypeInfo != Null<StockTypeInfo>());
    BOOST_CHECK(stockTypeInfo.type() == 0);
    BOOST_CHECK(stockTypeInfo.description() == HKU_STR("Block"));
    BOOST_CHECK(stockTypeInfo.tick() == 100);
    BOOST_CHECK(stockTypeInfo.precision() == 2);
    BOOST_CHECK(stockTypeInfo.minTradeNumber() == 100);
    BOOST_CHECK(stockTypeInfo.maxTradeNumber() == 1000000);

    stockTypeInfo = sm.getStockTypeInfo(8);
    BOOST_CHECK(stockTypeInfo != Null<StockTypeInfo>());
    BOOST_CHECK(stockTypeInfo.type() == 8);
#if !defined(BOOST_MSVC)
    //msvc中文支持不好，此处无法编译
    BOOST_CHECK(stockTypeInfo.description() == HKU_STR("创业板"));
#endif
    BOOST_CHECK(stockTypeInfo.tick() == 0.01);
    BOOST_CHECK(stockTypeInfo.precision() == 2);
    BOOST_CHECK(stockTypeInfo.minTradeNumber() == 100);
    BOOST_CHECK(stockTypeInfo.maxTradeNumber() == 1000000);

    stockTypeInfo = sm.getStockTypeInfo(2);
    BOOST_CHECK(stockTypeInfo != Null<StockTypeInfo>());
    BOOST_CHECK(stockTypeInfo.type() == 2);
    BOOST_CHECK(stockTypeInfo.description() == HKU_STR("指数"));
    BOOST_CHECK(stockTypeInfo.tick() == 0.001);
    BOOST_CHECK(stockTypeInfo.precision() == 3);
    BOOST_CHECK(stockTypeInfo.minTradeNumber() == 1);
    BOOST_CHECK(stockTypeInfo.maxTradeNumber() == 1000000);
}

BOOST_AUTO_TEST_CASE( test_StockManager_getAllMarket ) {
    StockManager& sm = StockManager::instance();

    MarketList result(sm.getAllMarket());
    BOOST_CHECK(result.size() == 3);
#if defined(BOOST_WINDOWS)
    BOOST_CHECK(result[0] == "TMP");
    BOOST_CHECK(result[1] == "SH");
    BOOST_CHECK(result[2] == "SZ");
#else
    BOOST_CHECK(result[0] == "TMP");
    BOOST_CHECK(result[1] == "SZ");
    BOOST_CHECK(result[2] == "SH");
#endif
}

BOOST_AUTO_TEST_CASE( test_StockManager_getBlock ) {
    StockManager& sm = StockManager::instance();

    Block result = sm.getBlock("地域板块", "陕西");
    BOOST_CHECK(result.size() != 0);

    BlockList blk_list = sm.getBlockList("地域板块");
    blk_list = sm.getBlockList();
    BOOST_CHECK(blk_list.size() != 0);

}

BOOST_AUTO_TEST_CASE( test_StockManager_TempCsvStock) {
    StockManager& sm = StockManager::instance();

    path tmp_dir(sm.tmpdir());
    tmp_dir = tmp_dir.parent_path();

    string day_filename(tmp_dir.string() + "/test_day_data.csv");
    string min_filename(tmp_dir.string() + "/test_min_data.csv");

    Stock stk = sm.addTempCsvStock("test", day_filename, min_filename);
    BOOST_CHECK(stk.isNull() == false);
    BOOST_CHECK(stk.market() == "TMP");
    BOOST_CHECK(stk.code() == "TEST");
    BOOST_CHECK(stk.market_code() == "TMPTEST");
    BOOST_CHECK(stk.getCount(KQuery::DAY) == 100);
    BOOST_CHECK(stk.getCount(KQuery::MIN) == 24000);

    KRecord record;
    record = stk.getKRecord(0);
    BOOST_CHECK(record.datetime == Datetime(201703070000));
    BOOST_CHECK((record.openPrice-3233.09) < 0.00001);
    BOOST_CHECK((record.highPrice-3242.66) < 0.00001);
    BOOST_CHECK((record.lowPrice-3226.82) < 0.00001);
    BOOST_CHECK((record.closePrice-3242.41) < 0.00001);
    BOOST_CHECK((record.transAmount-20993120.6) < 0.00001);
    BOOST_CHECK((record.transCount-164064235.0) < 0.00001);

    record = stk.getKRecord(10);
    BOOST_CHECK(record.datetime == Datetime(201703210000));
    BOOST_CHECK((record.openPrice-3250.25) < 0.00001);
    BOOST_CHECK((record.highPrice-3262.22) < 0.00001);
    BOOST_CHECK((record.lowPrice-3246.70) < 0.00001);
    BOOST_CHECK((record.closePrice-3261.61) < 0.00001);
    BOOST_CHECK((record.transAmount-21912127.0) < 0.00001);
    BOOST_CHECK((record.transCount-162719306.0) < 0.00001);

    record = stk.getKRecord(99);
    BOOST_CHECK(record.datetime == Datetime(201707310000));
    BOOST_CHECK((record.openPrice-3252.75) < 0.00001);
    BOOST_CHECK((record.highPrice-3276.95) < 0.00001);
    BOOST_CHECK((record.lowPrice-3251.19) < 0.00001);
    BOOST_CHECK((record.closePrice-3273.03) < 0.00001);
    BOOST_CHECK((record.transAmount-25352591.70) < 0.00001);
    BOOST_CHECK((record.transCount-246039440.0) < 0.00001);

    stk = sm.getStock("tmptest");
    BOOST_CHECK(stk.isNull() == false);
    BOOST_CHECK(stk.market() == "TMP");
    BOOST_CHECK(stk.code() == "TEST");
    BOOST_CHECK(stk.market_code() == "TMPTEST");
    BOOST_CHECK(stk.getCount(KQuery::DAY) == 100);
    BOOST_CHECK(stk.getCount(KQuery::MIN) == 24000);

    record = stk.getKRecord(10);
    BOOST_CHECK(record.datetime == Datetime(201703210000));
    BOOST_CHECK((record.openPrice-3250.25) < 0.00001);
    BOOST_CHECK((record.highPrice-3262.22) < 0.00001);
    BOOST_CHECK((record.lowPrice-3246.70) < 0.00001);
    BOOST_CHECK((record.closePrice-3261.61) < 0.00001);
    BOOST_CHECK((record.transAmount-21912127.0) < 0.00001);
    BOOST_CHECK((record.transCount-162719306.0) < 0.00001);

    sm.removeTempCsvStock("test");
    stk = sm.getStock("tmptest");
    BOOST_CHECK(stk.isNull() == true);
}