mirror of
https://github.com/serai-dex/serai.git
synced 2025-12-12 05:59:23 +00:00
add initial reward era test
This commit is contained in:
akildemir
227
substrate/client/tests/common/genesis_liquidity.rs
Normal file
227
substrate/client/tests/common/genesis_liquidity.rs
Normal file
@@ -0,0 +1,227 @@
|
||||
use std::{time::Duration, collections::HashMap};
|
||||
|
||||
use rand_core::{RngCore, OsRng};
|
||||
use zeroize::Zeroizing;
|
||||
|
||||
use ciphersuite::{Ciphersuite, Ristretto};
|
||||
use frost::dkg::musig::musig;
|
||||
use schnorrkel::Schnorrkel;
|
||||
|
||||
use serai_client::{
|
||||
genesis_liquidity::{
|
||||
primitives::{GENESIS_LIQUIDITY_ACCOUNT, GENESIS_SRI},
|
||||
SeraiGenesisLiquidity,
|
||||
},
|
||||
validator_sets::primitives::{musig_context, Session, ValidatorSet},
|
||||
};
|
||||
|
||||
use serai_abi::{
|
||||
genesis_liquidity::primitives::{set_initial_price_message, Prices},
|
||||
primitives::COINS,
|
||||
};
|
||||
|
||||
use sp_core::{sr25519::Signature, Pair as PairTrait};
|
||||
|
||||
use serai_client::{
|
||||
primitives::{
|
||||
Amount, NetworkId, Coin, Balance, BlockHash, SeraiAddress, insecure_pair_from_name,
|
||||
},
|
||||
in_instructions::primitives::{InInstruction, InInstructionWithBalance, Batch},
|
||||
Serai,
|
||||
};
|
||||
|
||||
use crate::common::{in_instructions::provide_batch, tx::publish_tx};
|
||||
|
||||
#[allow(dead_code)]
|
||||
pub async fn test_genesis_liquidity(serai: Serai) {
|
||||
// amounts
|
||||
let amounts = vec![
|
||||
Amount(5_53246991),
|
||||
Amount(3_14562819),
|
||||
Amount(9_33648912),
|
||||
Amount(150_873639000000),
|
||||
Amount(248_665228000000),
|
||||
Amount(451_765529000000),
|
||||
];
|
||||
|
||||
// addresses
|
||||
let mut btc_addresses = vec![];
|
||||
let mut xmr_addresses = vec![];
|
||||
let addr_count = amounts.len();
|
||||
for (i, amount) in amounts.into_iter().enumerate() {
|
||||
let mut address = SeraiAddress::new([0; 32]);
|
||||
OsRng.fill_bytes(&mut address.0);
|
||||
if i < addr_count / 2 {
|
||||
btc_addresses.push((address, amount));
|
||||
} else {
|
||||
xmr_addresses.push((address, amount));
|
||||
}
|
||||
}
|
||||
btc_addresses.sort_by(|a1, a2| a1.0.cmp(&a2.0));
|
||||
xmr_addresses.sort_by(|a1, a2| a1.0.cmp(&a2.0));
|
||||
|
||||
// btc batch
|
||||
let mut block_hash = BlockHash([0; 32]);
|
||||
OsRng.fill_bytes(&mut block_hash.0);
|
||||
let btc_ins = btc_addresses
|
||||
.iter()
|
||||
.map(|(addr, amount)| InInstructionWithBalance {
|
||||
instruction: InInstruction::GenesisLiquidity(*addr),
|
||||
balance: Balance { coin: Coin::Bitcoin, amount: *amount },
|
||||
})
|
||||
.collect::<Vec<_>>();
|
||||
let batch =
|
||||
Batch { network: NetworkId::Bitcoin, id: 0, block: block_hash, instructions: btc_ins };
|
||||
provide_batch(&serai, batch).await;
|
||||
|
||||
// xmr batch
|
||||
let mut block_hash = BlockHash([0; 32]);
|
||||
OsRng.fill_bytes(&mut block_hash.0);
|
||||
let xmr_ins = xmr_addresses
|
||||
.iter()
|
||||
.map(|(addr, amount)| InInstructionWithBalance {
|
||||
instruction: InInstruction::GenesisLiquidity(*addr),
|
||||
balance: Balance { coin: Coin::Monero, amount: *amount },
|
||||
})
|
||||
.collect::<Vec<_>>();
|
||||
let batch = Batch { network: NetworkId::Monero, id: 0, block: block_hash, instructions: xmr_ins };
|
||||
provide_batch(&serai, batch).await;
|
||||
|
||||
// set prices
|
||||
let prices = Prices { bitcoin: 10u64.pow(8), monero: 184100, ethereum: 4785000, dai: 1500 };
|
||||
set_prices(&serai, &prices).await;
|
||||
|
||||
// wait until genesis ends..
|
||||
tokio::time::timeout(tokio::time::Duration::from_secs(300), async {
|
||||
while serai.latest_finalized_block().await.unwrap().number() < 25 {
|
||||
tokio::time::sleep(Duration::from_secs(6)).await;
|
||||
}
|
||||
})
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
// check total SRI supply is +100M
|
||||
let last_block = serai.latest_finalized_block().await.unwrap().hash();
|
||||
let serai = serai.as_of(last_block);
|
||||
// Check balance instead of supply
|
||||
let sri = serai.coins().coin_supply(Coin::Serai).await.unwrap();
|
||||
// there are 6 endowed accounts in dev-net. Take this into consideration when checking
|
||||
// for the total sri minted at this time.
|
||||
let endowed_amount: u64 = 1 << 60;
|
||||
let total_sri = (6 * endowed_amount) + GENESIS_SRI;
|
||||
assert_eq!(sri, Amount(total_sri));
|
||||
|
||||
// check genesis account has no coins, all transferred to pools.
|
||||
for coin in COINS {
|
||||
let amount = serai.coins().coin_balance(coin, GENESIS_LIQUIDITY_ACCOUNT).await.unwrap();
|
||||
assert_eq!(amount.0, 0);
|
||||
}
|
||||
|
||||
// check pools has proper liquidity
|
||||
let pool_btc = btc_addresses.iter().fold(0u128, |acc, value| acc + u128::from(value.1 .0));
|
||||
let pool_xmr = xmr_addresses.iter().fold(0u128, |acc, value| acc + u128::from(value.1 .0));
|
||||
|
||||
let pool_btc_value = (pool_btc * u128::from(prices.bitcoin)) / 10u128.pow(8);
|
||||
let pool_xmr_value = (pool_xmr * u128::from(prices.monero)) / 10u128.pow(12);
|
||||
let total_value = pool_btc_value + pool_xmr_value;
|
||||
|
||||
// calculated distributed SRI. We know that xmr is at the end of COINS array
|
||||
// so it will be approximated to roof instead of floor after integer division.
|
||||
let btc_sri = (pool_btc_value * u128::from(GENESIS_SRI)) / total_value;
|
||||
let xmr_sri = ((pool_xmr_value * u128::from(GENESIS_SRI)) / total_value) + 1;
|
||||
|
||||
let btc_reserves = serai.dex().get_reserves(Coin::Bitcoin, Coin::Serai).await.unwrap().unwrap();
|
||||
assert_eq!(u128::from(btc_reserves.0 .0), pool_btc);
|
||||
assert_eq!(u128::from(btc_reserves.1 .0), btc_sri);
|
||||
|
||||
let xmr_reserves = serai.dex().get_reserves(Coin::Monero, Coin::Serai).await.unwrap().unwrap();
|
||||
assert_eq!(u128::from(xmr_reserves.0 .0), pool_xmr);
|
||||
assert_eq!(u128::from(xmr_reserves.1 .0), xmr_sri);
|
||||
|
||||
// check each btc liq provider got liq tokens proportional to their value
|
||||
let btc_liq_token_supply = u128::from(
|
||||
serai
|
||||
.liquidity_tokens()
|
||||
.token_balance(Coin::Bitcoin, GENESIS_LIQUIDITY_ACCOUNT)
|
||||
.await
|
||||
.unwrap()
|
||||
.0,
|
||||
);
|
||||
let mut total_tokens_this_coin: u128 = 0;
|
||||
for (i, (addr, amount)) in btc_addresses.iter().enumerate() {
|
||||
let addr_value = (u128::from(amount.0) * u128::from(prices.bitcoin)) / 10u128.pow(8);
|
||||
let addr_liq_tokens = if i == btc_addresses.len() - 1 {
|
||||
btc_liq_token_supply - total_tokens_this_coin
|
||||
} else {
|
||||
(addr_value * btc_liq_token_supply) / pool_btc_value
|
||||
};
|
||||
|
||||
let addr_actual_token_amount =
|
||||
serai.genesis_liquidity().liquidity_tokens(addr, Coin::Bitcoin).await.unwrap();
|
||||
|
||||
assert_eq!(addr_liq_tokens, addr_actual_token_amount.0.into());
|
||||
total_tokens_this_coin += addr_liq_tokens;
|
||||
}
|
||||
|
||||
// check each xmr liq provider got liq tokens proportional to their value
|
||||
let xmr_liq_token_supply = u128::from(
|
||||
serai
|
||||
.liquidity_tokens()
|
||||
.token_balance(Coin::Monero, GENESIS_LIQUIDITY_ACCOUNT)
|
||||
.await
|
||||
.unwrap()
|
||||
.0,
|
||||
);
|
||||
total_tokens_this_coin = 0;
|
||||
for (i, (addr, amount)) in xmr_addresses.iter().enumerate() {
|
||||
let addr_value = (u128::from(amount.0) * u128::from(prices.monero)) / 10u128.pow(12);
|
||||
let addr_liq_tokens = if i == xmr_addresses.len() - 1 {
|
||||
xmr_liq_token_supply - total_tokens_this_coin
|
||||
} else {
|
||||
(addr_value * xmr_liq_token_supply) / pool_xmr_value
|
||||
};
|
||||
|
||||
let addr_actual_token_amount =
|
||||
serai.genesis_liquidity().liquidity_tokens(addr, Coin::Monero).await.unwrap();
|
||||
|
||||
assert_eq!(addr_liq_tokens, addr_actual_token_amount.0.into());
|
||||
total_tokens_this_coin += addr_liq_tokens;
|
||||
}
|
||||
|
||||
// TODO: remove the liq before/after genesis ended.
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
async fn set_prices(serai: &Serai, prices: &Prices) {
|
||||
// prepare a Musig tx to set the initial prices
|
||||
let pair = insecure_pair_from_name("Alice");
|
||||
let public = pair.public();
|
||||
let set = ValidatorSet { session: Session(0), network: NetworkId::Serai };
|
||||
|
||||
let public_key = <Ristretto as Ciphersuite>::read_G::<&[u8]>(&mut public.0.as_ref()).unwrap();
|
||||
let secret_key = <Ristretto as Ciphersuite>::read_F::<&[u8]>(
|
||||
&mut pair.as_ref().secret.to_bytes()[.. 32].as_ref(),
|
||||
)
|
||||
.unwrap();
|
||||
|
||||
assert_eq!(Ristretto::generator() * secret_key, public_key);
|
||||
let threshold_keys =
|
||||
musig::<Ristretto>(&musig_context(set), &Zeroizing::new(secret_key), &[public_key]).unwrap();
|
||||
|
||||
let sig = frost::tests::sign_without_caching(
|
||||
&mut OsRng,
|
||||
frost::tests::algorithm_machines(
|
||||
&mut OsRng,
|
||||
&Schnorrkel::new(b"substrate"),
|
||||
&HashMap::from([(threshold_keys.params().i(), threshold_keys.into())]),
|
||||
),
|
||||
&set_initial_price_message(&set, prices),
|
||||
);
|
||||
|
||||
// set initial prices
|
||||
let _ = publish_tx(
|
||||
serai,
|
||||
&SeraiGenesisLiquidity::set_initial_price(*prices, Signature(sig.to_bytes())),
|
||||
)
|
||||
.await;
|
||||
}
|
||||
@@ -2,6 +2,7 @@ pub mod tx;
|
||||
pub mod validator_sets;
|
||||
pub mod in_instructions;
|
||||
pub mod dex;
|
||||
pub mod genesis_liquidity;
|
||||
|
||||
#[macro_export]
|
||||
macro_rules! serai_test {
|
||||
|
||||
141
substrate/client/tests/emissions.rs
Normal file
141
substrate/client/tests/emissions.rs
Normal file
@@ -0,0 +1,141 @@
|
||||
use std::{time::Duration, collections::HashMap};
|
||||
|
||||
use serai_client::TemporalSerai;
|
||||
|
||||
use serai_abi::{
|
||||
emissions::primitives::INITIAL_REWARD_PER_BLOCK,
|
||||
primitives::{Coin, COINS, NETWORKS},
|
||||
};
|
||||
|
||||
use serai_client::{
|
||||
primitives::{Amount, NetworkId, Balance},
|
||||
Serai,
|
||||
};
|
||||
|
||||
mod common;
|
||||
use common::genesis_liquidity::test_genesis_liquidity;
|
||||
|
||||
serai_test_fast_epoch!(
|
||||
emissions: (|serai: Serai| async move {
|
||||
test_emissions(serai).await;
|
||||
})
|
||||
);
|
||||
|
||||
async fn test_emissions(serai: Serai) {
|
||||
// provide some genesis liquidity
|
||||
test_genesis_liquidity(serai.clone()).await;
|
||||
|
||||
let mut current_stake = HashMap::new();
|
||||
for n in NETWORKS {
|
||||
let stake = serai
|
||||
.as_of_latest_finalized_block()
|
||||
.await
|
||||
.unwrap()
|
||||
.validator_sets()
|
||||
.total_allocated_stake(n)
|
||||
.await
|
||||
.unwrap()
|
||||
.unwrap_or(Amount(0))
|
||||
.0;
|
||||
current_stake.insert(n, stake);
|
||||
}
|
||||
|
||||
// wait until we have at least 1 session, epoch time is half an hour with the fast epoch
|
||||
// feature, so lets wait double that.
|
||||
tokio::time::timeout(tokio::time::Duration::from_secs(60 * 3), async {
|
||||
while serai
|
||||
.as_of_latest_finalized_block()
|
||||
.await
|
||||
.unwrap()
|
||||
.validator_sets()
|
||||
.session(NetworkId::Serai)
|
||||
.await
|
||||
.unwrap()
|
||||
.unwrap()
|
||||
.0 <
|
||||
1
|
||||
{
|
||||
tokio::time::sleep(Duration::from_secs(6)).await;
|
||||
}
|
||||
})
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
let last_block = serai.latest_finalized_block().await.unwrap();
|
||||
let serai_latest = serai.as_of(last_block.hash());
|
||||
|
||||
// we should be in the initial period, so calculate how much each network supposedly get..
|
||||
// we can check the supply to see how much coin hence liability we have.
|
||||
let mut distances: HashMap<NetworkId, u64> = HashMap::new();
|
||||
let mut total_distance = 0;
|
||||
for coin in COINS {
|
||||
if coin == Coin::Serai {
|
||||
continue;
|
||||
}
|
||||
|
||||
let amount = serai_latest.coins().coin_supply(coin).await.unwrap();
|
||||
let required = required_stake(&serai_latest, Balance { coin, amount }).await;
|
||||
let mut current = *current_stake.get(&coin.network()).unwrap();
|
||||
if current > required {
|
||||
current = required;
|
||||
}
|
||||
|
||||
let distance = required - current;
|
||||
total_distance += distance;
|
||||
|
||||
distances.insert(
|
||||
coin.network(),
|
||||
distances.get(&coin.network()).unwrap_or(&0).saturating_add(distance),
|
||||
);
|
||||
}
|
||||
|
||||
// add serai network portion(20%)
|
||||
let new_total_distance = total_distance.saturating_mul(10) / 8;
|
||||
distances.insert(NetworkId::Serai, new_total_distance - total_distance);
|
||||
total_distance = new_total_distance;
|
||||
|
||||
// since we should be in the first block after the first epoch, block number should also
|
||||
// give us the block count.
|
||||
let block_count = last_block.number();
|
||||
let reward_this_epoch = block_count * INITIAL_REWARD_PER_BLOCK;
|
||||
|
||||
let reward_per_network = distances
|
||||
.into_iter()
|
||||
.map(|(n, distance)| {
|
||||
let reward = u64::try_from(
|
||||
u128::from(reward_this_epoch).saturating_mul(u128::from(distance)) /
|
||||
u128::from(total_distance),
|
||||
)
|
||||
.unwrap();
|
||||
(n, reward)
|
||||
})
|
||||
.collect::<HashMap<NetworkId, u64>>();
|
||||
|
||||
for (n, reward) in reward_per_network {
|
||||
let stake =
|
||||
serai_latest.validator_sets().total_allocated_stake(n).await.unwrap().unwrap_or(Amount(0)).0;
|
||||
|
||||
// the reward should have been automatically staked for the network
|
||||
assert_eq!(stake, *current_stake.get(&n).unwrap() + reward);
|
||||
}
|
||||
|
||||
// TODO: check stake per address
|
||||
}
|
||||
|
||||
/// Returns the required stake in terms SRI for a given `Balance`.
|
||||
async fn required_stake(serai: &TemporalSerai<'_>, balance: Balance) -> u64 {
|
||||
// This is inclusive to an increase in accuracy
|
||||
let sri_per_coin = serai.dex().oracle_value(balance.coin).await.unwrap().unwrap_or(Amount(0));
|
||||
|
||||
// See dex-pallet for the reasoning on these
|
||||
let coin_decimals = balance.coin.decimals().max(5);
|
||||
let accuracy_increase = u128::from(u64::pow(10, coin_decimals));
|
||||
|
||||
let total_coin_value =
|
||||
u64::try_from(u128::from(balance.amount.0) * u128::from(sri_per_coin.0) / accuracy_increase)
|
||||
.unwrap_or(u64::MAX);
|
||||
|
||||
// required stake formula (COIN_VALUE * 1.5) + margin(20%)
|
||||
let required_stake = total_coin_value.saturating_mul(3).saturating_div(2);
|
||||
required_stake.saturating_add(total_coin_value.saturating_div(5))
|
||||
}
|
||||
@@ -1,232 +1,10 @@
|
||||
use std::{time::Duration, collections::HashMap};
|
||||
|
||||
use rand_core::{RngCore, OsRng};
|
||||
use zeroize::Zeroizing;
|
||||
|
||||
use ciphersuite::{Ciphersuite, Ristretto};
|
||||
use frost::dkg::musig::musig;
|
||||
use schnorrkel::Schnorrkel;
|
||||
|
||||
use serai_client::{
|
||||
genesis_liquidity::{
|
||||
primitives::{GENESIS_LIQUIDITY_ACCOUNT, GENESIS_SRI},
|
||||
SeraiGenesisLiquidity,
|
||||
},
|
||||
validator_sets::primitives::{musig_context, Session, ValidatorSet},
|
||||
};
|
||||
|
||||
use serai_abi::{
|
||||
genesis_liquidity::primitives::{set_initial_price_message, Prices},
|
||||
primitives::COINS,
|
||||
};
|
||||
|
||||
use sp_core::{sr25519::Signature, Pair as PairTrait};
|
||||
|
||||
use serai_client::{
|
||||
primitives::{
|
||||
Amount, NetworkId, Coin, Balance, BlockHash, SeraiAddress, insecure_pair_from_name,
|
||||
},
|
||||
in_instructions::primitives::{InInstruction, InInstructionWithBalance, Batch},
|
||||
Serai,
|
||||
};
|
||||
use serai_client::Serai;
|
||||
|
||||
mod common;
|
||||
use common::{in_instructions::provide_batch, tx::publish_tx};
|
||||
use common::genesis_liquidity::test_genesis_liquidity;
|
||||
|
||||
serai_test_fast_epoch!(
|
||||
genesis_liquidity: (|serai: Serai| async move {
|
||||
test_genesis_liquidity(serai).await;
|
||||
})
|
||||
);
|
||||
|
||||
async fn test_genesis_liquidity(serai: Serai) {
|
||||
// amounts
|
||||
let amounts = vec![
|
||||
Amount(5_53246991),
|
||||
Amount(3_14562819),
|
||||
Amount(9_33648912),
|
||||
Amount(150_873639000000),
|
||||
Amount(248_665228000000),
|
||||
Amount(451_765529000000),
|
||||
];
|
||||
|
||||
// addresses
|
||||
let mut btc_addresses = vec![];
|
||||
let mut xmr_addresses = vec![];
|
||||
let addr_count = amounts.len();
|
||||
for (i, amount) in amounts.into_iter().enumerate() {
|
||||
let mut address = SeraiAddress::new([0; 32]);
|
||||
OsRng.fill_bytes(&mut address.0);
|
||||
if i < addr_count / 2 {
|
||||
btc_addresses.push((address, amount));
|
||||
} else {
|
||||
xmr_addresses.push((address, amount));
|
||||
}
|
||||
}
|
||||
btc_addresses.sort_by(|a1, a2| a1.0.cmp(&a2.0));
|
||||
xmr_addresses.sort_by(|a1, a2| a1.0.cmp(&a2.0));
|
||||
|
||||
// btc batch
|
||||
let mut block_hash = BlockHash([0; 32]);
|
||||
OsRng.fill_bytes(&mut block_hash.0);
|
||||
let btc_ins = btc_addresses
|
||||
.iter()
|
||||
.map(|(addr, amount)| InInstructionWithBalance {
|
||||
instruction: InInstruction::GenesisLiquidity(*addr),
|
||||
balance: Balance { coin: Coin::Bitcoin, amount: *amount },
|
||||
})
|
||||
.collect::<Vec<_>>();
|
||||
let batch =
|
||||
Batch { network: NetworkId::Bitcoin, id: 0, block: block_hash, instructions: btc_ins };
|
||||
provide_batch(&serai, batch).await;
|
||||
|
||||
// xmr batch
|
||||
let mut block_hash = BlockHash([0; 32]);
|
||||
OsRng.fill_bytes(&mut block_hash.0);
|
||||
let xmr_ins = xmr_addresses
|
||||
.iter()
|
||||
.map(|(addr, amount)| InInstructionWithBalance {
|
||||
instruction: InInstruction::GenesisLiquidity(*addr),
|
||||
balance: Balance { coin: Coin::Monero, amount: *amount },
|
||||
})
|
||||
.collect::<Vec<_>>();
|
||||
let batch = Batch { network: NetworkId::Monero, id: 0, block: block_hash, instructions: xmr_ins };
|
||||
provide_batch(&serai, batch).await;
|
||||
|
||||
// set prices
|
||||
let prices = Prices { bitcoin: 10u64.pow(8), monero: 184100, ethereum: 4785000, dai: 1500 };
|
||||
set_prices(&serai, &prices).await;
|
||||
|
||||
// wait until genesis ends..
|
||||
tokio::time::timeout(tokio::time::Duration::from_secs(300), async {
|
||||
while serai.latest_finalized_block().await.unwrap().number() < 25 {
|
||||
tokio::time::sleep(Duration::from_secs(6)).await;
|
||||
}
|
||||
})
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
// check total SRI supply is +100M
|
||||
let last_block = serai.latest_finalized_block().await.unwrap().hash();
|
||||
let serai = serai.as_of(last_block);
|
||||
// Check balance instead of supply
|
||||
let sri = serai.coins().coin_supply(Coin::Serai).await.unwrap();
|
||||
// there are 6 endowed accounts in dev-net. Take this into consideration when checking
|
||||
// for the total sri minted at this time.
|
||||
let endowed_amount: u64 = 1 << 60;
|
||||
let total_sri = (6 * endowed_amount) + GENESIS_SRI;
|
||||
assert_eq!(sri, Amount(total_sri));
|
||||
|
||||
// check genesis account has no coins, all transferred to pools.
|
||||
for coin in COINS {
|
||||
let amount = serai.coins().coin_balance(coin, GENESIS_LIQUIDITY_ACCOUNT).await.unwrap();
|
||||
assert_eq!(amount.0, 0);
|
||||
}
|
||||
|
||||
// check pools has proper liquidity
|
||||
let pool_btc = btc_addresses.iter().fold(0u128, |acc, value| acc + u128::from(value.1 .0));
|
||||
let pool_xmr = xmr_addresses.iter().fold(0u128, |acc, value| acc + u128::from(value.1 .0));
|
||||
|
||||
let pool_btc_value = (pool_btc * u128::from(prices.bitcoin)) / 10u128.pow(8);
|
||||
let pool_xmr_value = (pool_xmr * u128::from(prices.monero)) / 10u128.pow(12);
|
||||
let total_value = pool_btc_value + pool_xmr_value;
|
||||
|
||||
// calculated distributed SRI. We know that xmr is at the end of COINS array
|
||||
// so it will be approximated to roof instead of floor after integer division.
|
||||
let btc_sri = (pool_btc_value * u128::from(GENESIS_SRI)) / total_value;
|
||||
let xmr_sri = ((pool_xmr_value * u128::from(GENESIS_SRI)) / total_value) + 1;
|
||||
|
||||
let btc_reserves = serai.dex().get_reserves(Coin::Bitcoin, Coin::Serai).await.unwrap().unwrap();
|
||||
assert_eq!(u128::from(btc_reserves.0 .0), pool_btc);
|
||||
assert_eq!(u128::from(btc_reserves.1 .0), btc_sri);
|
||||
|
||||
let xmr_reserves = serai.dex().get_reserves(Coin::Monero, Coin::Serai).await.unwrap().unwrap();
|
||||
assert_eq!(u128::from(xmr_reserves.0 .0), pool_xmr);
|
||||
assert_eq!(u128::from(xmr_reserves.1 .0), xmr_sri);
|
||||
|
||||
// check each btc liq provider got liq tokens proportional to their value
|
||||
let btc_liq_token_supply = u128::from(
|
||||
serai
|
||||
.liquidity_tokens()
|
||||
.token_balance(Coin::Bitcoin, GENESIS_LIQUIDITY_ACCOUNT)
|
||||
.await
|
||||
.unwrap()
|
||||
.0,
|
||||
);
|
||||
let mut total_tokens_this_coin: u128 = 0;
|
||||
for (i, (addr, amount)) in btc_addresses.iter().enumerate() {
|
||||
let addr_value = (u128::from(amount.0) * u128::from(prices.bitcoin)) / 10u128.pow(8);
|
||||
let addr_liq_tokens = if i == btc_addresses.len() - 1 {
|
||||
btc_liq_token_supply - total_tokens_this_coin
|
||||
} else {
|
||||
(addr_value * btc_liq_token_supply) / pool_btc_value
|
||||
};
|
||||
|
||||
let addr_actual_token_amount =
|
||||
serai.genesis_liquidity().liquidity_tokens(addr, Coin::Bitcoin).await.unwrap();
|
||||
|
||||
assert_eq!(addr_liq_tokens, addr_actual_token_amount.0.into());
|
||||
total_tokens_this_coin += addr_liq_tokens;
|
||||
}
|
||||
|
||||
// check each xmr liq provider got liq tokens proportional to their value
|
||||
let xmr_liq_token_supply = u128::from(
|
||||
serai
|
||||
.liquidity_tokens()
|
||||
.token_balance(Coin::Monero, GENESIS_LIQUIDITY_ACCOUNT)
|
||||
.await
|
||||
.unwrap()
|
||||
.0,
|
||||
);
|
||||
total_tokens_this_coin = 0;
|
||||
for (i, (addr, amount)) in xmr_addresses.iter().enumerate() {
|
||||
let addr_value = (u128::from(amount.0) * u128::from(prices.monero)) / 10u128.pow(12);
|
||||
let addr_liq_tokens = if i == xmr_addresses.len() - 1 {
|
||||
xmr_liq_token_supply - total_tokens_this_coin
|
||||
} else {
|
||||
(addr_value * xmr_liq_token_supply) / pool_xmr_value
|
||||
};
|
||||
|
||||
let addr_actual_token_amount =
|
||||
serai.genesis_liquidity().liquidity_tokens(addr, Coin::Monero).await.unwrap();
|
||||
|
||||
assert_eq!(addr_liq_tokens, addr_actual_token_amount.0.into());
|
||||
total_tokens_this_coin += addr_liq_tokens;
|
||||
}
|
||||
|
||||
// TODO: remove the liq before/after genesis ended.
|
||||
}
|
||||
|
||||
async fn set_prices(serai: &Serai, prices: &Prices) {
|
||||
// prepare a Musig tx to set the initial prices
|
||||
let pair = insecure_pair_from_name("Alice");
|
||||
let public = pair.public();
|
||||
let set = ValidatorSet { session: Session(0), network: NetworkId::Serai };
|
||||
|
||||
let public_key = <Ristretto as Ciphersuite>::read_G::<&[u8]>(&mut public.0.as_ref()).unwrap();
|
||||
let secret_key = <Ristretto as Ciphersuite>::read_F::<&[u8]>(
|
||||
&mut pair.as_ref().secret.to_bytes()[.. 32].as_ref(),
|
||||
)
|
||||
.unwrap();
|
||||
|
||||
assert_eq!(Ristretto::generator() * secret_key, public_key);
|
||||
let threshold_keys =
|
||||
musig::<Ristretto>(&musig_context(set), &Zeroizing::new(secret_key), &[public_key]).unwrap();
|
||||
|
||||
let sig = frost::tests::sign_without_caching(
|
||||
&mut OsRng,
|
||||
frost::tests::algorithm_machines(
|
||||
&mut OsRng,
|
||||
&Schnorrkel::new(b"substrate"),
|
||||
&HashMap::from([(threshold_keys.params().i(), threshold_keys.into())]),
|
||||
),
|
||||
&set_initial_price_message(&set, prices),
|
||||
);
|
||||
|
||||
// set initial prices
|
||||
let _ = publish_tx(
|
||||
serai,
|
||||
&SeraiGenesisLiquidity::set_initial_price(*prices, Signature(sig.to_bytes())),
|
||||
)
|
||||
.await;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user