mod utils;
const EXCHANGE_NAME: &str = "okx";
#[cfg(test)]
mod trade {
use super::EXCHANGE_NAME;
use crypto_market_type::MarketType;
use crypto_message::TradeSide;
use crypto_msg_parser::{extract_symbol, extract_timestamp, parse_trade, round};
#[test]
fn spot() {
let raw_msg = r#"{"table":"spot/trade","data":[{"side":"sell","trade_id":"161659503","price":"56593.6","size":"0.00020621","instrument_id":"BTC-USDT","timestamp":"2021-03-22T01:16:28.687Z"}]}"#;
let trades = &parse_trade(EXCHANGE_NAME, MarketType::Spot, raw_msg).unwrap();
assert_eq!(trades.len(), 1);
let trade = &trades[0];
crate::utils::check_trade_fields(
EXCHANGE_NAME,
MarketType::Spot,
"BTC/USDT".to_string(),
extract_symbol(EXCHANGE_NAME, MarketType::Spot, raw_msg).unwrap(),
trade,
raw_msg,
);
assert_eq!(
1616375788687,
extract_timestamp(EXCHANGE_NAME, MarketType::Spot, raw_msg).unwrap().unwrap()
);
assert_eq!(trade.quantity_base, 0.00020621);
assert_eq!(trade.quantity_contract, None);
assert_eq!(trade.side, TradeSide::Sell);
}
#[test]
fn linear_future() {
let raw_msg = r#"{"table":"futures/trade","data":[{"side":"buy","trade_id":"5430565","price":"60059.7","qty":"20","instrument_id":"BTC-USDT-210625","timestamp":"2021-03-22T01:32:18.087Z"}]}"#;
let trades = &parse_trade(EXCHANGE_NAME, MarketType::LinearFuture, raw_msg).unwrap();
assert_eq!(trades.len(), 1);
let trade = &trades[0];
crate::utils::check_trade_fields(
EXCHANGE_NAME,
MarketType::LinearFuture,
"BTC/USDT".to_string(),
extract_symbol(EXCHANGE_NAME, MarketType::LinearFuture, raw_msg).unwrap(),
trade,
raw_msg,
);
assert_eq!(
1616376738087,
extract_timestamp(EXCHANGE_NAME, MarketType::LinearFuture, raw_msg).unwrap().unwrap()
);
assert_eq!(trade.quantity_base, 20.0 * 0.01);
assert_eq!(trade.quantity_quote, 20.0 * 0.01 * 60059.7);
assert_eq!(trade.quantity_contract, Some(20.0));
assert_eq!(trade.side, TradeSide::Buy);
}
#[test]
fn linear_swap() {
let raw_msg = r#"{"table":"swap/trade","data":[{"side":"buy","trade_id":"62257592","price":"56480.1","size":"3","instrument_id":"BTC-USDT-SWAP","timestamp":"2021-03-22T01:33:00.684Z"}]}"#;
let trades = &parse_trade(EXCHANGE_NAME, MarketType::LinearSwap, raw_msg).unwrap();
assert_eq!(trades.len(), 1);
let trade = &trades[0];
crate::utils::check_trade_fields(
EXCHANGE_NAME,
MarketType::LinearSwap,
"BTC/USDT".to_string(),
extract_symbol(EXCHANGE_NAME, MarketType::LinearSwap, raw_msg).unwrap(),
trade,
raw_msg,
);
assert_eq!(
1616376780684,
extract_timestamp(EXCHANGE_NAME, MarketType::LinearSwap, raw_msg).unwrap().unwrap()
);
assert_eq!(trade.quantity_base, 0.01 * 3.0);
assert_eq!(trade.quantity_quote, round(0.01 * 3.0 * 56480.1));
assert_eq!(trade.quantity_contract, Some(3.0));
assert_eq!(trade.side, TradeSide::Buy);
}
#[test]
fn inverse_future() {
let raw_msg = r#"{"table":"futures/trade","data":[{"side":"sell","trade_id":"16606935","price":"59999.7","qty":"7","instrument_id":"BTC-USD-210625","timestamp":"2021-03-22T01:32:41.377Z"}]}"#;
let trades = &parse_trade(EXCHANGE_NAME, MarketType::InverseFuture, raw_msg).unwrap();
assert_eq!(trades.len(), 1);
let trade = &trades[0];
crate::utils::check_trade_fields(
EXCHANGE_NAME,
MarketType::InverseFuture,
"BTC/USD".to_string(),
extract_symbol(EXCHANGE_NAME, MarketType::InverseFuture, raw_msg).unwrap(),
trade,
raw_msg,
);
assert_eq!(
1616376761377,
extract_timestamp(EXCHANGE_NAME, MarketType::InverseFuture, raw_msg).unwrap().unwrap()
);
assert_eq!(trade.quantity_base, 100.0 * 7.0 / 59999.7);
assert_eq!(trade.quantity_quote, 100.0 * 7.0);
assert_eq!(trade.quantity_contract, Some(7.0));
assert_eq!(trade.side, TradeSide::Sell);
}
#[test]
fn inverse_swap() {
let raw_msg = r#"{"table":"swap/trade","data":[{"side":"sell","trade_id":"102067670","price":"56535.9","size":"1","instrument_id":"BTC-USD-SWAP","timestamp":"2021-03-22T01:33:14.051Z"}]}"#;
let trades = &parse_trade(EXCHANGE_NAME, MarketType::InverseSwap, raw_msg).unwrap();
assert_eq!(trades.len(), 1);
let trade = &trades[0];
crate::utils::check_trade_fields(
EXCHANGE_NAME,
MarketType::InverseSwap,
"BTC/USD".to_string(),
extract_symbol(EXCHANGE_NAME, MarketType::InverseSwap, raw_msg).unwrap(),
trade,
raw_msg,
);
assert_eq!(
1616376794051,
extract_timestamp(EXCHANGE_NAME, MarketType::InverseSwap, raw_msg).unwrap().unwrap()
);
assert_eq!(trade.quantity_base, 100.0 * 1.0 / 56535.9);
assert_eq!(trade.quantity_quote, 100.0 * 1.0);
assert_eq!(trade.quantity_contract, Some(1.0));
assert_eq!(trade.side, TradeSide::Sell);
}
#[test]
fn option() {
let raw_msg = r#"{"table":"option/trade","data":[{"side":"buy","trade_id":"231","price":"0.1545","qty":"4","instrument_id":"BTC-USD-210625-72000-C","timestamp":"2021-03-20T12:01:16.947Z"}]}"#;
let trades = &parse_trade(EXCHANGE_NAME, MarketType::EuropeanOption, raw_msg).unwrap();
assert_eq!(trades.len(), 1);
let trade = &trades[0];
crate::utils::check_trade_fields(
EXCHANGE_NAME,
MarketType::EuropeanOption,
"BTC/USD".to_string(),
extract_symbol(EXCHANGE_NAME, MarketType::EuropeanOption, raw_msg).unwrap(),
trade,
raw_msg,
);
assert_eq!(
1616241676947,
extract_timestamp(EXCHANGE_NAME, MarketType::EuropeanOption, raw_msg).unwrap().unwrap()
);
assert_eq!(trade.quantity_base, 1.0 * 4.0);
assert_eq!(trade.quantity_quote, 1.0 * 4.0 * 0.1545);
assert_eq!(trade.quantity_contract, Some(4.0));
assert_eq!(trade.side, TradeSide::Buy);
let raw_msg = r#"{"table":"option/trade","data":[{"side":"sell","trade_id":"3","price":"0.0255","qty":"10","instrument_id":"BTC-USD-211124-55500-C","timestamp":"2021-11-23T10:55:56.522Z"}]}"#;
let trades = &parse_trade(EXCHANGE_NAME, MarketType::EuropeanOption, raw_msg).unwrap();
assert_eq!(trades.len(), 1);
let trade = &trades[0];
crate::utils::check_trade_fields(
EXCHANGE_NAME,
MarketType::EuropeanOption,
"BTC/USD".to_string(),
extract_symbol(EXCHANGE_NAME, MarketType::EuropeanOption, raw_msg).unwrap(),
trade,
raw_msg,
);
assert_eq!(
1637664956522,
extract_timestamp(EXCHANGE_NAME, MarketType::EuropeanOption, raw_msg).unwrap().unwrap()
);
assert_eq!(trade.quantity_base, 1.0 * 10.0);
assert_eq!(trade.quantity_quote, 1.0 * 10.0 * 0.0255);
assert_eq!(trade.quantity_contract, Some(10.0));
assert_eq!(trade.side, TradeSide::Sell);
}
}
#[cfg(test)]
mod funding_rate {
use super::EXCHANGE_NAME;
use crypto_market_type::MarketType;
use crypto_msg_parser::{extract_symbol, extract_timestamp, parse_funding_rate};
#[test]
fn inverse_swap() {
let raw_msg = r#"{"table":"swap/funding_rate","data":[{"estimated_rate":"0.00065","funding_rate":"0.00072933","funding_time":"2021-04-02T00:00:00.000Z","instrument_id":"BTC-USD-SWAP","interest_rate":"0","settlement_time":"2021-04-02T08:00:00.000Z"}]}"#;
let received_at = 1617321600123;
let funding_rates =
&parse_funding_rate(EXCHANGE_NAME, MarketType::InverseSwap, raw_msg, Some(received_at))
.unwrap();
assert_eq!(funding_rates.len(), 1);
for rate in funding_rates.iter() {
crate::utils::check_funding_rate_fields(
EXCHANGE_NAME,
MarketType::InverseSwap,
rate,
raw_msg,
);
}
assert_eq!(
"BTC-USD-SWAP",
extract_symbol(EXCHANGE_NAME, MarketType::InverseSwap, raw_msg).unwrap()
);
assert_eq!(
None,
extract_timestamp(EXCHANGE_NAME, MarketType::InverseSwap, raw_msg).unwrap()
);
assert_eq!(funding_rates[0].pair, "BTC/USD".to_string());
assert_eq!(funding_rates[0].funding_rate, 0.00072933);
assert_eq!(funding_rates[0].estimated_rate, Some(0.00065));
assert_eq!(funding_rates[0].funding_time, 1617321600000);
assert_eq!(funding_rates[0].timestamp, received_at);
}
#[test]
fn linear_swap() {
let raw_msg = r#"{"table":"swap/funding_rate","data":[{"estimated_rate":"0.00031","funding_rate":"0.00081859","funding_time":"2021-04-02T00:00:00.000Z","instrument_id":"BTC-USDT-SWAP","interest_rate":"0","settlement_time":"2021-04-02T08:00:00.000Z"}]}"#;
let received_at = 1617321600123;
let funding_rates =
&parse_funding_rate(EXCHANGE_NAME, MarketType::LinearSwap, raw_msg, Some(received_at))
.unwrap();
assert_eq!(funding_rates.len(), 1);
for rate in funding_rates.iter() {
crate::utils::check_funding_rate_fields(
EXCHANGE_NAME,
MarketType::LinearSwap,
rate,
raw_msg,
);
}
assert_eq!(
"BTC-USDT-SWAP",
extract_symbol(EXCHANGE_NAME, MarketType::InverseSwap, raw_msg).unwrap()
);
assert_eq!(
None,
extract_timestamp(EXCHANGE_NAME, MarketType::InverseSwap, raw_msg).unwrap()
);
assert_eq!(funding_rates[0].pair, "BTC/USDT".to_string());
assert_eq!(funding_rates[0].funding_rate, 0.00081859);
assert_eq!(funding_rates[0].estimated_rate, Some(0.00031));
assert_eq!(funding_rates[0].funding_time, 1617321600000);
assert_eq!(funding_rates[0].timestamp, received_at);
}
}
#[cfg(test)]
mod l2_event {
use super::EXCHANGE_NAME;
use crypto_market_type::MarketType;
use crypto_msg_parser::{extract_symbol, extract_timestamp, parse_l2, round};
use crypto_msg_type::MessageType;
#[test]
fn spot_snapshot() {
let raw_msg = r#"{"table":"spot/depth_l2_tbt","action":"partial","data":[{"instrument_id":"BTC-USDT","asks":[["38930","3.84264467","0","12"],["38932.4","0.00135697","0","3"],["38932.5","0.14401147","0","2"]],"bids":[["38929.9","0.05005381","0","4"],["38925.7","0.00062109","0","2"],["38925.6","0.21438503","0","1"]],"timestamp":"2021-06-03T12:39:11.253Z","checksum":1860980846}]}"#;
let orderbook = &parse_l2(EXCHANGE_NAME, MarketType::Spot, raw_msg, None).unwrap()[0];
assert_eq!(orderbook.asks.len(), 3);
assert_eq!(orderbook.bids.len(), 3);
assert!(orderbook.snapshot);
crate::utils::check_orderbook_fields(
EXCHANGE_NAME,
MarketType::Spot,
MessageType::L2Event,
"BTC/USDT".to_string(),
extract_symbol(EXCHANGE_NAME, MarketType::Spot, raw_msg).unwrap(),
orderbook,
raw_msg,
);
assert_eq!(
1622723951253,
extract_timestamp(EXCHANGE_NAME, MarketType::Spot, raw_msg).unwrap().unwrap()
);
assert_eq!(orderbook.timestamp, 1622723951253);
assert_eq!(orderbook.bids[0].price, 38929.9);
assert_eq!(orderbook.bids[0].quantity_base, 0.05005381);
assert_eq!(orderbook.bids[0].quantity_quote, round(38929.9 * 0.05005381));
assert_eq!(orderbook.asks[0].price, 38930.0);
assert_eq!(orderbook.asks[0].quantity_base, 3.84264467);
assert_eq!(orderbook.asks[0].quantity_quote, 38930.0 * 3.84264467);
}
#[test]
fn spot_update() {
let raw_msg = r#"{"table":"spot/depth_l2_tbt","action":"update","data":[{"instrument_id":"BTC-USDT","asks":[["38888.7","4.14263198","0","12"]],"bids":[["38886.3","0","0","0"]],"timestamp":"2021-06-03T12:40:09.962Z","checksum":976527820}]}"#;
let orderbook = &parse_l2(EXCHANGE_NAME, MarketType::Spot, raw_msg, None).unwrap()[0];
assert_eq!(orderbook.asks.len(), 1);
assert_eq!(orderbook.bids.len(), 1);
assert!(!orderbook.snapshot);
crate::utils::check_orderbook_fields(
EXCHANGE_NAME,
MarketType::Spot,
MessageType::L2Event,
"BTC/USDT".to_string(),
extract_symbol(EXCHANGE_NAME, MarketType::Spot, raw_msg).unwrap(),
orderbook,
raw_msg,
);
assert_eq!(
1622724009962,
extract_timestamp(EXCHANGE_NAME, MarketType::Spot, raw_msg).unwrap().unwrap()
);
assert_eq!(orderbook.timestamp, 1622724009962);
assert_eq!(orderbook.bids[0].price, 38886.3);
assert_eq!(orderbook.bids[0].quantity_base, 0.0);
assert_eq!(orderbook.bids[0].quantity_quote, 0.0);
assert_eq!(orderbook.asks[0].price, 38888.7);
assert_eq!(orderbook.asks[0].quantity_base, 4.14263198);
assert_eq!(orderbook.asks[0].quantity_quote, 38888.7 * 4.14263198);
}
#[test]
fn linear_future_snapshot() {
let raw_msg = r#"{"table":"futures/depth_l2_tbt","action":"partial","data":[{"instrument_id":"BTC-USDT-210625","asks":[["39302.5","1","0","1"],["39302.6","5","0","2"],["39304.3","21","0","1"]],"bids":[["39302.2","4","0","1"],["39300.7","5","0","1"],["39299","4","0","1"]],"timestamp":"2021-06-03T13:09:34.429Z","checksum":698961978}]}"#;
let orderbook =
&parse_l2(EXCHANGE_NAME, MarketType::LinearFuture, raw_msg, None).unwrap()[0];
assert_eq!(orderbook.asks.len(), 3);
assert_eq!(orderbook.bids.len(), 3);
assert!(orderbook.snapshot);
crate::utils::check_orderbook_fields(
EXCHANGE_NAME,
MarketType::LinearFuture,
MessageType::L2Event,
"BTC/USDT".to_string(),
extract_symbol(EXCHANGE_NAME, MarketType::LinearFuture, raw_msg).unwrap(),
orderbook,
raw_msg,
);
assert_eq!(
1622725774429,
extract_timestamp(EXCHANGE_NAME, MarketType::LinearFuture, raw_msg).unwrap().unwrap()
);
assert_eq!(orderbook.timestamp, 1622725774429);
assert_eq!(orderbook.asks[0].price, 39302.5);
assert_eq!(orderbook.asks[0].quantity_base, 0.01);
assert_eq!(orderbook.asks[0].quantity_quote, round(39302.5 * 0.01));
assert_eq!(orderbook.asks[0].quantity_contract.unwrap(), 1.0);
assert_eq!(orderbook.bids[0].price, 39302.2);
assert_eq!(orderbook.bids[0].quantity_base, 0.04);
assert_eq!(orderbook.bids[0].quantity_quote, 39302.2 * 0.04);
assert_eq!(orderbook.bids[0].quantity_contract.unwrap(), 4.0);
}
#[test]
fn inverse_swap_snapshot() {
let raw_msg = r#"{"table":"swap/depth_l2_tbt","action":"partial","data":[{"instrument_id":"BTC-USD-SWAP","asks":[["39167.2","130","0","3"],["39169.6","45","0","1"],["39173.1","1","0","1"]],"bids":[["39167.1","1536","0","8"],["39166.2","68","0","1"],["39165.9","47","0","1"]],"timestamp":"2021-06-03T13:14:24.831Z","checksum":-1582320415}]}"#;
let orderbook =
&parse_l2(EXCHANGE_NAME, MarketType::InverseSwap, raw_msg, None).unwrap()[0];
assert_eq!(orderbook.asks.len(), 3);
assert_eq!(orderbook.bids.len(), 3);
assert!(orderbook.snapshot);
crate::utils::check_orderbook_fields(
EXCHANGE_NAME,
MarketType::InverseSwap,
MessageType::L2Event,
"BTC/USD".to_string(),
extract_symbol(EXCHANGE_NAME, MarketType::InverseSwap, raw_msg).unwrap(),
orderbook,
raw_msg,
);
assert_eq!(
1622726064831,
extract_timestamp(EXCHANGE_NAME, MarketType::InverseSwap, raw_msg).unwrap().unwrap()
);
assert_eq!(orderbook.timestamp, 1622726064831);
assert_eq!(orderbook.asks[0].price, 39167.2);
assert_eq!(orderbook.asks[0].quantity_base, 13000.0 / 39167.2);
assert_eq!(orderbook.asks[0].quantity_quote, 13000.0);
assert_eq!(orderbook.asks[0].quantity_contract.unwrap(), 130.0);
assert_eq!(orderbook.bids[0].price, 39167.1);
assert_eq!(orderbook.bids[0].quantity_base, 153600.0 / 39167.1);
assert_eq!(orderbook.bids[0].quantity_quote, 153600.0);
assert_eq!(orderbook.bids[0].quantity_contract.unwrap(), 1536.0);
}
#[test]
fn option_snapshot() {
let raw_msg = r#"{"table":"option/depth_l2_tbt","action":"partial","data":[{"instrument_id":"BTC-USD-210604-30000-P","asks":[["0.0015","906","0","3"]],"bids":[],"timestamp":"2021-06-03T13:18:55.745Z","checksum":-288111842}]}"#;
let orderbook =
&parse_l2(EXCHANGE_NAME, MarketType::EuropeanOption, raw_msg, None).unwrap()[0];
assert_eq!(orderbook.asks.len(), 1);
assert_eq!(orderbook.bids.len(), 0);
assert!(orderbook.snapshot);
crate::utils::check_orderbook_fields(
EXCHANGE_NAME,
MarketType::EuropeanOption,
MessageType::L2Event,
"BTC/USD".to_string(),
extract_symbol(EXCHANGE_NAME, MarketType::EuropeanOption, raw_msg).unwrap(),
orderbook,
raw_msg,
);
assert_eq!(
1622726335745,
extract_timestamp(EXCHANGE_NAME, MarketType::EuropeanOption, raw_msg).unwrap().unwrap()
);
assert_eq!(orderbook.timestamp, 1622726335745);
assert_eq!(orderbook.asks[0].price, 0.0015);
assert_eq!(orderbook.asks[0].quantity_base, 1.0 * 906.0);
assert_eq!(orderbook.asks[0].quantity_quote, 1.0 * 906.0 * 0.0015);
assert_eq!(orderbook.asks[0].quantity_contract.unwrap(), 906.0);
}
}
#[cfg(test)]
mod l2_topk {
use super::EXCHANGE_NAME;
use crypto_market_type::MarketType;
use crypto_msg_parser::{extract_symbol, extract_timestamp, parse_l2_topk, round};
use crypto_msg_type::MessageType;
#[test]
fn linear_swap() {
let raw_msg = r#"{"table":"swap/depth5","data":[{"asks":[["38407.4","1502","0","36"],["38407.5","37","0","1"],["38407.8","152","0","5"],["38409.1","2","0","1"],["38409.2","30","0","1"]],"bids":[["38407.3","252","0","5"],["38406","1","0","1"],["38405.9","25","0","1"],["38405.8","18","0","1"],["38405.7","28","0","1"]],"instrument_id":"BTC-USDT-SWAP","timestamp":"2022-02-25T00:45:05.387Z"}]}"#;
let orderbook =
&parse_l2_topk(EXCHANGE_NAME, MarketType::LinearFuture, raw_msg, None).unwrap()[0];
assert_eq!(orderbook.asks.len(), 5);
assert_eq!(orderbook.bids.len(), 5);
assert!(orderbook.snapshot);
crate::utils::check_orderbook_fields(
EXCHANGE_NAME,
MarketType::LinearFuture,
MessageType::L2TopK,
"BTC/USDT".to_string(),
extract_symbol(EXCHANGE_NAME, MarketType::LinearFuture, raw_msg).unwrap(),
orderbook,
raw_msg,
);
assert_eq!(
"BTC-USDT-SWAP",
extract_symbol(EXCHANGE_NAME, MarketType::LinearFuture, raw_msg).unwrap()
);
assert_eq!(
1645749905387,
extract_timestamp(EXCHANGE_NAME, MarketType::LinearFuture, raw_msg).unwrap().unwrap()
);
assert_eq!(orderbook.timestamp, 1645749905387);
assert_eq!(orderbook.seq_id, None);
assert_eq!(orderbook.prev_seq_id, None);
assert_eq!(orderbook.asks[0].price, 38407.4);
assert_eq!(orderbook.asks[0].quantity_base, 0.01 * 1502.0);
assert_eq!(orderbook.asks[0].quantity_quote, 38407.4 * 0.01 * 1502.0);
assert_eq!(orderbook.asks[0].quantity_contract.unwrap(), 1502.0);
assert_eq!(orderbook.asks[4].price, 38409.2);
assert_eq!(orderbook.asks[4].quantity_base, 0.01 * 30.0);
assert_eq!(orderbook.asks[4].quantity_quote, 38409.2 * 0.01 * 30.0);
assert_eq!(orderbook.asks[4].quantity_contract.unwrap(), 30.0);
assert_eq!(orderbook.bids[0].price, 38407.3);
assert_eq!(orderbook.bids[0].quantity_base, 0.01 * 252.0);
assert_eq!(orderbook.bids[0].quantity_quote, round(38407.3 * 0.01 * 252.0));
assert_eq!(orderbook.bids[0].quantity_contract.unwrap(), 252.0);
assert_eq!(orderbook.bids[4].price, 38405.7);
assert_eq!(orderbook.bids[4].quantity_base, 0.01 * 28.0);
assert_eq!(orderbook.bids[4].quantity_quote, 38405.7 * 0.01 * 28.0);
assert_eq!(orderbook.bids[4].quantity_contract.unwrap(), 28.0);
}
}
#[cfg(test)]
mod candlestick {
use super::EXCHANGE_NAME;
use crypto_market_type::MarketType;
use crypto_msg_parser::{extract_symbol, extract_timestamp};
#[test]
fn spot() {
let raw_msg = r#"{"table":"spot/candle60s","data":[{"candle":["2022-02-01T00:30:00.000Z","0.050209","0.050209","0.050209","0.050209","722.597549"],"instrument_id":"CELR-USDT"}]}"#;
assert_eq!(
1643675400000,
extract_timestamp(EXCHANGE_NAME, MarketType::Spot, raw_msg).unwrap().unwrap()
);
assert_eq!("CELR-USDT", extract_symbol(EXCHANGE_NAME, MarketType::Spot, raw_msg).unwrap());
}
}