as-rust/exchanges/tests/binance_swap_test.rs

use std::collections::BTreeMap;
use std::sync::Arc;
use std::sync::atomic::AtomicBool;
use tokio::sync::mpsc::{channel, Sender};
use tokio::try_join;
use tracing::trace;
use exchanges::binance_swap_rest::BinanceSwapRest;
use exchanges::binance_swap_ws::{BinanceSubscribeType, BinanceSwapWs, BinanceWsType};
use exchanges::response_base::ResponseData;

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


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

    let  bool_v1 = Arc::new(AtomicBool::new(true));
    let btree_map: BTreeMap<String, String> = BTreeMap::new();
    let (tx, mut rx) = channel(1024);
    let mut ws = get_ws(btree_map, tx);
    ws.set_subscribe(vec![
        // BinanceSubscribeType::PuBookTicker,
        // BinanceSubscribeType::PuAggTrade,
        BinanceSubscribeType::PuDepth20levels100ms,
    ]);
    let t1 = tokio::spawn(async move {
        ws.custom_subscribe(bool_v1, vec!["BTCUSDT".to_string()]).await;
    });
    let t2 = tokio::spawn(async move {
        loop {
            if let Ok(received) = rx.try_recv() {
                trace!( "age: {:?}", received);
            }
        }
    });
    try_join!(t1,t2).unwrap();
}


//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_exchange_info_test() {
    global::log_utils::init_log_with_trace();

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

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

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


//rest-根据币对 撤销全部订单
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
async fn rest_cancel_order_all_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();

    let rep_data1 = rest.get_server_time().await;
    trace!(?rep_data1);

    trace!("开始时间：{:?}",chrono::Utc::now().timestamp_millis().to_string());
    let rep_data = rest.cancel_order_all("BTCUSDT".to_string()).await;
    trace!(?rep_data);
    trace!("结束时间：{:?}",chrono::Utc::now().timestamp_millis().to_string());
}


fn get_ws(btree_map: BTreeMap<String, String>, tx: Sender<ResponseData>) -> BinanceSwapWs {
    let  binance_ws = BinanceSwapWs::new(false,
                                            btree_map,
                                            BinanceWsType::PublicAndPrivate,
                                            tx);
    binance_ws
}

fn get_rest() -> BinanceSwapRest {
    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());

    let  ba_exc = BinanceSwapRest::new(false, btree_map);
    ba_exc
}