mirror of
https://github.com/serai-dex/serai.git
synced 2024-12-24 12:39:33 +00:00
1493f49416
Some checks failed
Full Stack Tests / build (push) Waiting to run
Lint / fmt (push) Waiting to run
Lint / machete (push) Waiting to run
Lint / clippy (macos-13) (push) Waiting to run
Lint / clippy (macos-14) (push) Waiting to run
Lint / clippy (ubuntu-latest) (push) Waiting to run
Lint / clippy (windows-latest) (push) Waiting to run
Lint / deny (push) Waiting to run
Reproducible Runtime / build (push) Has been cancelled
Tests / test-infra (push) Has been cancelled
Tests / test-substrate (push) Has been cancelled
Tests / test-serai-client (push) Has been cancelled
* add genesis liquidity implementation * add missing deposit event * fix CI issues * minor fixes * make math safer * fix fmt * make remove liquidity an authorized call * implement setting initial values for coins * add genesis liquidity test & misc fixes * updato develop latest * fix rotation test * Finish merging develop * Remove accidentally committed ETH files * fix pr comments * further bug fixes * fix last pr comments * tidy up * Misc --------- Co-authored-by: Luke Parker <lukeparker5132@gmail.com>
216 lines
7 KiB
Rust
216 lines
7 KiB
Rust
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, INITIAL_GENESIS_LP_SHARES},
|
|
SeraiGenesisLiquidity,
|
|
},
|
|
validator_sets::primitives::{musig_context, Session, ValidatorSet},
|
|
};
|
|
|
|
use serai_abi::{
|
|
genesis_liquidity::primitives::{oraclize_values_message, Values},
|
|
primitives::COINS,
|
|
};
|
|
|
|
use sp_core::{sr25519::Signature, Pair as PairTrait};
|
|
|
|
use serai_client::{
|
|
primitives::{
|
|
Amount, NetworkId, Coin, Balance, BlockHash, SeraiAddress, insecure_pair_from_name, GENESIS_SRI,
|
|
},
|
|
in_instructions::primitives::{InInstruction, InInstructionWithBalance, Batch},
|
|
Serai,
|
|
};
|
|
|
|
mod common;
|
|
use common::{in_instructions::provide_batch, tx::publish_tx};
|
|
|
|
serai_test_fast_epoch!(
|
|
genesis_liquidity: (|serai: Serai| async move {
|
|
test_genesis_liquidity(serai).await;
|
|
})
|
|
);
|
|
|
|
async fn test_genesis_liquidity(serai: Serai) {
|
|
// all coins except the native
|
|
let coins = COINS.into_iter().filter(|c| *c != Coin::native()).collect::<Vec<_>>();
|
|
|
|
// make accounts with amounts
|
|
let mut accounts = HashMap::new();
|
|
for coin in coins.clone() {
|
|
// make 5 accounts per coin
|
|
let mut values = vec![];
|
|
for _ in 0 .. 5 {
|
|
let mut address = SeraiAddress::new([0; 32]);
|
|
OsRng.fill_bytes(&mut address.0);
|
|
values.push((address, Amount(OsRng.next_u64() % 10u64.pow(coin.decimals()))));
|
|
}
|
|
accounts.insert(coin, values);
|
|
}
|
|
|
|
// send a batch per coin
|
|
let mut batch_ids: HashMap<NetworkId, u32> = HashMap::new();
|
|
for coin in coins.clone() {
|
|
// set up instructions
|
|
let instructions = accounts[&coin]
|
|
.iter()
|
|
.map(|(addr, amount)| InInstructionWithBalance {
|
|
instruction: InInstruction::GenesisLiquidity(*addr),
|
|
balance: Balance { coin, amount: *amount },
|
|
})
|
|
.collect::<Vec<_>>();
|
|
|
|
// set up bloch hash
|
|
let mut block = BlockHash([0; 32]);
|
|
OsRng.fill_bytes(&mut block.0);
|
|
|
|
// set up batch id
|
|
batch_ids
|
|
.entry(coin.network())
|
|
.and_modify(|v| {
|
|
*v += 1;
|
|
})
|
|
.or_insert(0);
|
|
|
|
let batch =
|
|
Batch { network: coin.network(), id: batch_ids[&coin.network()], block, instructions };
|
|
provide_batch(&serai, batch).await;
|
|
}
|
|
|
|
// wait until genesis ends
|
|
let genesis_blocks = 10; // TODO
|
|
let block_time = 6; // TODO
|
|
tokio::time::timeout(
|
|
tokio::time::Duration::from_secs(3 * (genesis_blocks * block_time)),
|
|
async {
|
|
while serai.latest_finalized_block().await.unwrap().number() < 10 {
|
|
tokio::time::sleep(Duration::from_secs(6)).await;
|
|
}
|
|
},
|
|
)
|
|
.await
|
|
.unwrap();
|
|
|
|
// set values relative to each other
|
|
// TODO: Random values here
|
|
let values = Values { monero: 184100, ether: 4785000, dai: 1500 };
|
|
set_values(&serai, &values).await;
|
|
let values_map = HashMap::from([
|
|
(Coin::Monero, values.monero),
|
|
(Coin::Ether, values.ether),
|
|
(Coin::Dai, values.dai),
|
|
]);
|
|
|
|
// wait a little bit..
|
|
tokio::time::sleep(Duration::from_secs(12)).await;
|
|
|
|
// check total SRI supply is +100M
|
|
// there are 6 endowed accounts in dev-net. Take this into consideration when checking
|
|
// for the total sri minted at this time.
|
|
let serai = serai.as_of_latest_finalized_block().await.unwrap();
|
|
let sri = serai.coins().coin_supply(Coin::Serai).await.unwrap();
|
|
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 mut pool_amounts = HashMap::new();
|
|
let mut total_value = 0u128;
|
|
for coin in coins.clone() {
|
|
let total_coin = accounts[&coin].iter().fold(0u128, |acc, value| acc + u128::from(value.1 .0));
|
|
let value = if coin != Coin::Bitcoin {
|
|
(total_coin * u128::from(values_map[&coin])) / 10u128.pow(coin.decimals())
|
|
} else {
|
|
total_coin
|
|
};
|
|
|
|
total_value += value;
|
|
pool_amounts.insert(coin, (total_coin, value));
|
|
}
|
|
|
|
// check distributed SRI per pool
|
|
let mut total_sri_distributed = 0u128;
|
|
for coin in coins.clone() {
|
|
let sri = if coin == *COINS.last().unwrap() {
|
|
u128::from(GENESIS_SRI).checked_sub(total_sri_distributed).unwrap()
|
|
} else {
|
|
(pool_amounts[&coin].1 * u128::from(GENESIS_SRI)) / total_value
|
|
};
|
|
total_sri_distributed += sri;
|
|
|
|
let reserves = serai.dex().get_reserves(coin).await.unwrap().unwrap();
|
|
assert_eq!(u128::from(reserves.0 .0), pool_amounts[&coin].0); // coin side
|
|
assert_eq!(u128::from(reserves.1 .0), sri); // SRI side
|
|
}
|
|
|
|
// check each liquidity provider got liquidity tokens proportional to their value
|
|
for coin in coins {
|
|
let liq_supply = serai.genesis_liquidity().supply(coin).await.unwrap();
|
|
for (acc, amount) in &accounts[&coin] {
|
|
let acc_liq_shares = serai.genesis_liquidity().liquidity(acc, coin).await.unwrap().shares;
|
|
|
|
// since we can't test the ratios directly(due to integer division giving 0)
|
|
// we test whether they give the same result when multiplied by another constant.
|
|
// Following test ensures the account in fact has the right amount of shares.
|
|
let mut shares_ratio = (INITIAL_GENESIS_LP_SHARES * acc_liq_shares) / liq_supply.shares;
|
|
let amounts_ratio =
|
|
(INITIAL_GENESIS_LP_SHARES * amount.0) / u64::try_from(pool_amounts[&coin].0).unwrap();
|
|
|
|
// we can tolerate 1 unit diff between them due to integer division.
|
|
if shares_ratio.abs_diff(amounts_ratio) == 1 {
|
|
shares_ratio = amounts_ratio;
|
|
}
|
|
|
|
assert_eq!(shares_ratio, amounts_ratio);
|
|
}
|
|
}
|
|
|
|
// TODO: test remove the liq before/after genesis ended.
|
|
}
|
|
|
|
async fn set_values(serai: &Serai, values: &Values) {
|
|
// prepare a Musig tx to oraclize the relative values
|
|
let pair = insecure_pair_from_name("Alice");
|
|
let public = pair.public();
|
|
// we publish the tx in set 4
|
|
let set = ValidatorSet { session: Session(4), 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())]),
|
|
),
|
|
&oraclize_values_message(&set, values),
|
|
);
|
|
|
|
// oraclize values
|
|
let _ =
|
|
publish_tx(serai, &SeraiGenesisLiquidity::oraclize_values(*values, Signature(sig.to_bytes())))
|
|
.await;
|
|
}
|