micro_rank/exchanges/tests/bitget_spot_test.rs

use std::collections::BTreeMap;
use std::sync::Arc;
use std::sync::atomic::AtomicBool;

use futures_util::StreamExt;
use tokio::sync::Mutex;
use tracing::trace;

use exchanges::bitget_spot_rest::BitgetSpotRest;
use exchanges::bitget_spot_ws::{BitgetSpotLogin, BitgetSpotSubscribeType, BitgetSpotWs, BitgetSpotWsType};

const ACCESS_KEY: &str = "";
const SECRET_KEY: &str = "";
const PASS_KEY: &str = "";

//ws-订阅公共频道信息
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
async fn ws_custom_subscribe_pu() {
    global::log_utils::init_log_with_trace();

    let (write_tx, write_rx) = futures_channel::mpsc::unbounded();
    let (read_tx, mut read_rx) = futures_channel::mpsc::unbounded();

    let  is_shutdown_arc = Arc::new(AtomicBool::new(true));
    let mut ws = get_ws(None, BitgetSpotWsType::Public).await;
    ws.set_symbols(vec!["BTC_USDT".to_string()]);
    ws.set_subscribe(vec![
        BitgetSpotSubscribeType::PuTicker,
        BitgetSpotSubscribeType::PuCandle1m,
        BitgetSpotSubscribeType::PuTrade,
        BitgetSpotSubscribeType::PuBooks5,
    ]);

    let write_tx_am = Arc::new(Mutex::new(write_tx));
    let is_shutdown_arc = Arc::new(AtomicBool::new(true));

    //读取
    let _is_shutdown_arc_clone = Arc::clone(&is_shutdown_arc);
    let _tr = tokio::spawn(async move {
        trace!("线程-数据读取-开启");
        loop {
            if let Some(data) = read_rx.next().await {
                trace!("读取数据data:{:?}",data)
            }
        }
        // trace!("线程-数据读取-结束");
    });

    //写数据
    // let bool_v2_clone = Arc::clone(&is_shutdown_arc);
    // let write_tx_clone = Arc::clone(&write_tx_am);
    // let su = ws.get_subscription();
    // let tw = tokio::spawn(async move {
    //     trace!("线程-数据写入-开始");
    //     loop {
    //         tokio::time::sleep(Duration::from_millis(20 * 1000)).await;
    //         // let close_frame = CloseFrame {
    //         //     code: CloseCode::Normal,
    //         //     reason: Cow::Borrowed("Bye bye"),
    //         // };
    //         // let message = Message::Close(Some(close_frame));
    //
    //
    //         let message = Message::Text(su.clone());
    //         AbstractWsMode::send_subscribe(write_tx_clone.clone(), message.clone()).await;
    //         trace!("发送指令成功");
    //     }
    //     trace!("线程-数据写入-结束");
    // });

    let t1 = tokio::spawn(async move {
        //链接
        let bool_v3_clone = Arc::clone(&is_shutdown_arc);
        ws.ws_connect_async(bool_v3_clone, &write_tx_am, write_rx, read_tx).await.expect("链接失败(内部一个心跳线程应该已经关闭了)");
        trace!("test 唯一线程结束--");
    });
    tokio::try_join!(t1).unwrap();
    trace!("当此结束");
    trace!("重启！");
    trace!("参考交易所关闭");
    return;
}

//ws-订阅私有频道信息
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
async fn ws_custom_subscribe_pr() {
    global::log_utils::init_log_with_trace();


    let (write_tx, write_rx) = futures_channel::mpsc::unbounded();
    let (read_tx, mut read_rx) = futures_channel::mpsc::unbounded();

    let login_param = BitgetSpotLogin {
        api_key: ACCESS_KEY.to_string(),
        secret_key: SECRET_KEY.to_string(),
        passphrase_key: PASS_KEY.to_string(),
    };
    let mut ws = get_ws(None, BitgetSpotWsType::Private).await;
    ws.set_symbols(vec!["BTC_USDT".to_string()]);
    ws.set_subscribe(vec![
        BitgetSpotSubscribeType::PuTicker,
        BitgetSpotSubscribeType::PuCandle1m,
        BitgetSpotSubscribeType::PuTrade,
        BitgetSpotSubscribeType::PuBooks5,
    ]);

    let write_tx_am = Arc::new(Mutex::new(write_tx));
    let is_shutdown_arc = Arc::new(AtomicBool::new(true));

//读取
    let _is_shutdown_arc_clone = Arc::clone(&is_shutdown_arc);
    let _tr = tokio::spawn(async move {
        trace!("线程-数据读取-开启");
        loop {
            if let Some(data) = read_rx.next().await {
                trace!("读取数据data:{:?}",data)
            }
        }
        // trace!("线程-数据读取-结束");
    });

    //写数据
    // let bool_v2_clone = Arc::clone(&is_shutdown_arc);
    // let write_tx_clone = Arc::clone(&write_tx_am);
    // let su = ws.get_subscription();
    // let tw = tokio::spawn(async move {
    //     trace!("线程-数据写入-开始");
    //     loop {
    //         tokio::time::sleep(Duration::from_millis(20 * 1000)).await;
    //         // let close_frame = CloseFrame {
    //         //     code: CloseCode::Normal,
    //         //     reason: Cow::Borrowed("Bye bye"),
    //         // };
    //         // let message = Message::Close(Some(close_frame));
    //
    //
    //         let message = Message::Text(su.clone());
    //         AbstractWsMode::send_subscribe(write_tx_clone.clone(), message.clone()).await;
    //         trace!("发送指令成功");
    //     }
    //     trace!("线程-数据写入-结束");
    // });

    let t1 = tokio::spawn(async move {
        //链接
        let bool_v3_clone = Arc::clone(&is_shutdown_arc);
        ws.ws_connect_async(bool_v3_clone, &write_tx_am, write_rx, read_tx).await.expect("链接失败(内部一个心跳线程应该已经关闭了)");
        trace!("test 唯一线程结束--");
    });
    tokio::try_join!(t1).unwrap();
    trace!("当此结束");
    trace!("重启！");
    trace!("参考交易所关闭");
    return;
}


//rest-获取系统时间
#[tokio::test]
async fn rest_get_server_time_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_server_time().await;
    trace!(?rep_data)
}


//rest-获取账户信息
#[tokio::test]
async fn rest_get_account_info_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_account_info().await;
    trace!(?rep_data)
}

//rest-获取账户币种资产
#[tokio::test]
async fn rest_get_account_assets_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_account_assets().await;
    trace!(?rep_data)
}

//rest-获取币种信息
#[tokio::test]
async fn rest_get_coins_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_coins("USDT".to_string()).await;
    trace!(?rep_data)
}

//rest-获取交易对信息
#[tokio::test]
async fn rest_get_symbols_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_symbols("BTCUSDT".to_string()).await;
    trace!(?rep_data)
}

//rest-获取现货VIP费率
#[tokio::test]
async fn rest_get_vip_fee_rate_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_vip_fee_rate().await;
    trace!(?rep_data)
}

//rest-获取行情信息
#[tokio::test]
async fn rest_get_tickers_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_tickers("BTCUSDT".to_string()).await;
    trace!(?rep_data)
}

//rest-获取合并交易深度
#[tokio::test]
async fn rest_get_merge_depth_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_merge_depth("BTCUSDT".to_string()).await;
    trace!(?rep_data)
}

//rest-获取K线数据
#[tokio::test]
async fn rest_get_candles_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_candles("BTCUSDT".to_string(), "1min".to_string(), "1697701550192".to_string(), "1697701556192".to_string(), "100".to_string()).await;
    trace!(?rep_data)
}

//rest-获取历史K线数据
#[tokio::test]
async fn rest_get_history_candles_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_history_candles("BTCUSDT".to_string(), "1min".to_string(), "1697701556192".to_string(), "100".to_string()).await;
    trace!(?rep_data)
}

//rest-获取最近成交数据
#[tokio::test]
async fn rest_get_market_fills_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_market_fills("BTCUSDT".to_string(), "100".to_string()).await;
    trace!(?rep_data)
}

//rest-获取历史成交数据
#[tokio::test]
async fn rest_get_fills_history_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_market_fills_history("BTCUSDT".to_string(), "1697701550192".to_string(), "1697701556192".to_string(), "100".to_string()).await;
    trace!(?rep_data)
}

//rest-下单
#[tokio::test]
async fn rest_spot_order_test() {
    global::log_utils::init_log_with_trace();

    //市价单
    let mut rest = get_rest();
    let params = serde_json::json!({
            // "symbol":"CELRUSDT",
            // "side":"sell",
            // "orderType":"market",
            // "force":"fok",
            // "size":"887",
            // "clientOid":"7d8zd4d_3",
         });

    //限价单
    let params = serde_json::json!({
            // "symbol":"CELRUSDT",
            // "side":"buy",
            // "orderType":"limit",
            // "force":"gtc",
            // "price":"0.01001",
            // "size":"10",
            // "clientOid":"7d8zd4d_z1",
         });

    let rep_data = rest.spot_order(params).await;
    trace!(?rep_data)
}


//rest-撤单
#[tokio::test]
async fn rest_spot_cancel_order_test() {
    global::log_utils::init_log_with_trace();
    let mut rest = get_rest();
    let rep_data = rest.spot_cancel_order("CELRUSDT".to_string(), "".to_string(), "1".to_string()).await;
    trace!(?rep_data)
}

//rest-批量撤单
#[tokio::test]
async fn rest_spot_cancel_orders_test() {
    global::log_utils::init_log_with_trace();
    let mut rest = get_rest();
    let v = serde_json::json!({
            "orderId":"1073370944162058240",
            "clientOid":"1073370944162058240"
        });
    let rep_data = rest.spot_cancel_orders("CELRUSDT".to_string(), vec![v]).await;
    trace!(?rep_data)
}

//rest-按币对撤单
#[tokio::test]
async fn rest_spot_cancel_symbol_orders_test() {
    global::log_utils::init_log_with_trace();
    let mut rest = get_rest();
    let rep_data = rest.spot_cancel_symbol_orders("CELRUSDT".to_string()).await;
    trace!(?rep_data)
}

//rest-获取订单详情
#[tokio::test]
async fn rest_get_order_test() {
    global::log_utils::init_log_with_trace();
    let mut rest = get_rest();
    let rep_data = rest.get_order("".to_string(), "1".to_string()).await;
    trace!(?rep_data)
}

//rest-获取当前委托列表
#[tokio::test]
async fn rest_get_unfilled_orders_test() {
    global::log_utils::init_log_with_trace();
    let mut rest = get_rest();
    let rep_data = rest.get_unfilled_orders("CELRUSDT".to_string(),
                                            "".to_string(),
                                            "".to_string(),
                                            "".to_string(),
                                            "".to_string(),
                                            "".to_string(),
    ).await;
    trace!(?rep_data)
}

//rest-获取历史委托列表
#[tokio::test]
async fn rest_get_history_orders_test() {
    global::log_utils::init_log_with_trace();
    let mut rest = get_rest();
    let rep_data = rest.get_history_orders("CELRUSDT".to_string(),
                                           "".to_string(),
                                           "".to_string(),
                                           "".to_string(),
                                           "".to_string(),
                                           "".to_string(),
    ).await;
    trace!(?rep_data)
}

//rest-获取成交明细
#[tokio::test]
async fn rest_get_fills_test() {
    global::log_utils::init_log_with_trace();
    let mut rest = get_rest();
    let rep_data = rest.get_fills("CELRUSDT".to_string(),
                                  "1".to_string(),
                                  "".to_string(),
                                  "".to_string(),
                                  "".to_string(),
                                  "".to_string(),
    ).await;
    trace!(?rep_data)
}

//rest-获取成交明细
#[tokio::test]
async fn rest_spot_place_plan_order_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    //限价-委托单
    let params = serde_json::json!({
         });

    let rep_data = rest.spot_place_plan_order(params).await;
    trace!(?rep_data)
}

//rest-修改计划委托
#[tokio::test]
async fn rest_update_place_plan_order_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    //限价-委托单
    let params = serde_json::json!({
         });

    let rep_data = rest.update_place_plan_order(params).await;
    trace!(?rep_data)
}

//rest-撤销计划委托
#[tokio::test]
async fn rest_cancel_plan_order_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.cancel_plan_order("32131".to_string(), "3211".to_string()).await;
    trace!(?rep_data)
}

//rest-获取当前计划委托
#[tokio::test]
async fn rest_get_current_plan_order_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_current_plan_order("CELRUSDT".to_string(),
                                               "1".to_string(),
                                               "".to_string(),
                                               "".to_string(),
                                               "".to_string(),
    ).await;
    trace!(?rep_data)
}

//rest-获取历史计划委托
#[tokio::test]
async fn rest_get_history_plan_order_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_history_plan_order("CELRUSDT".to_string(),
                                               "1697701550192".to_string(),
                                               "1697701580192".to_string(),
                                               "100".to_string(),
    ).await;
    trace!(?rep_data)
}

//rest-批量撤销计划委托
#[tokio::test]
async fn rest_cancel_plan_orders_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.cancel_plan_orders(vec![]).await;
    trace!(?rep_data)
}

//rest-划转
#[tokio::test]
async fn rest_wallet_transfer_test() {
    global::log_utils::init_log_with_trace();
    let mut rest = get_rest();
    let rep_data = rest.wallet_transfer("".to_string(),
                                        "".to_string(),
                                        "".to_string(),
                                        "".to_string(),
                                        "".to_string(),
                                        "".to_string()).await;
    trace!(?rep_data)
}

//rest-获取账单流水
#[tokio::test]
async fn rest_get_account_bills_test() {
    global::log_utils::init_log_with_trace();
    let mut rest = get_rest();
    let rep_data = rest.get_account_bills("".to_string(),
                                          "".to_string(),
                                          "".to_string(),
                                          "".to_string(), ).await;
    trace!(?rep_data)
}


async fn get_ws(btree_map: Option<BitgetSpotLogin>, type_v: BitgetSpotWsType) -> BitgetSpotWs {
    let mut ku_ws = BitgetSpotWs::new(false, btree_map.clone(), type_v);
    ku_ws
}

fn get_rest() -> BitgetSpotRest {
    let mut btree_map: BTreeMap<String, String> = BTreeMap::new();
    // btree_map.insert("access_key".to_string(), ACCESS_KEY.to_string());
    // btree_map.insert("secret_key".to_string(), SECRET_KEY.to_string());
    btree_map.insert("access_key".to_string(), ACCESS_KEY.to_string());
    btree_map.insert("secret_key".to_string(), SECRET_KEY.to_string());
    btree_map.insert("pass_key".to_string(), PASS_KEY.to_string());

    let mut ku_exc = BitgetSpotRest::new(false, btree_map);
    ku_exc
}