as-rust/derive/tests/export_excel_test.rs

use std::collections::{BTreeMap};
use tracing::trace;
use derive::export_excel::{ExportEnum, ExportExcel};
use derive::ExportConnector;
use exchanges::proxy;

// 创建实体
#[allow(dead_code)]
pub async fn test_new_export(export_enum: ExportEnum) -> Box<dyn ExportConnector> {
    // 检测是否走代理
    pub fn proxy_handle() {
        if proxy::ParsingDetail::http_enable_proxy() {
            trace!("检测有代理配置，配置走代理");
        }
    }

    let account_info = global::account_info::get_account_info("../test_account.toml");

    match export_enum {
        ExportEnum::BinanceSwap => {
            let mut params: BTreeMap<String, String> = BTreeMap::new();
            let access_key = account_info.binance_access_key;
            let secret_key = account_info.binance_secret_key;
            params.insert("access_key".to_string(), access_key);
            params.insert("secret_key".to_string(), secret_key);
            ExportExcel::new(ExportEnum::BinanceSwap, false, params).await
        }
        ExportEnum::KucoinSwap => {
            let mut params: BTreeMap<String, String> = BTreeMap::new();
            let access_key = account_info.kucoin_access_key;
            let secret_key = account_info.kucoin_secret_key;
            let pass_key = account_info.kucoin_pass;
            params.insert("access_key".to_string(), access_key);
            params.insert("secret_key".to_string(), secret_key);
            params.insert("pass_key".to_string(), pass_key);
            ExportExcel::new(ExportEnum::KucoinSwap, false, params).await
        }
        ExportEnum::KucoinSpot => {
            let mut params: BTreeMap<String, String> = BTreeMap::new();
            let access_key = account_info.kucoin_access_key;
            let secret_key = account_info.kucoin_secret_key;
            let pass_key = account_info.kucoin_pass;
            params.insert("access_key".to_string(), access_key);
            params.insert("secret_key".to_string(), secret_key);
            params.insert("pass_key".to_string(), pass_key);
            ExportExcel::new(ExportEnum::KucoinSpot, false, params).await
        }
        ExportEnum::GateSwap => {
            let mut params: BTreeMap<String, String> = BTreeMap::new();
            let access_key = account_info.gate_access_key;
            let secret_key = account_info.gate_secret_key;
            params.insert("access_key".to_string(), access_key);
            params.insert("secret_key".to_string(), secret_key);
            ExportExcel::new(ExportEnum::GateSwap, false, params).await
        }
        ExportEnum::BitgetSpot => {
            let mut params: BTreeMap<String, String> = BTreeMap::new();
            let access_key = account_info.bitget_access_key;
            let secret_key = account_info.bitget_secret_key;
            let pass_key = account_info.bitget_pass;
            params.insert("access_key".to_string(), access_key);
            params.insert("secret_key".to_string(), secret_key);
            params.insert("pass_key".to_string(), pass_key);
            ExportExcel::new(ExportEnum::BitgetSpot, false, params).await
        }
        ExportEnum::OkxSwap => {
            let mut params: BTreeMap<String, String> = BTreeMap::new();
            let access_key = account_info.okx_access_key;
            let secret_key = account_info.okx_secret_key;
            let pass_key = account_info.okx_pass;
            params.insert("access_key".to_string(), access_key);
            params.insert("secret_key".to_string(), secret_key);
            params.insert("pass_key".to_string(), pass_key);
            ExportExcel::new(ExportEnum::OkxSwap, false, params).await
        }
    }
}