feat: initialize Kurdistan SDK - independent fork of Polkadot SDK
This commit is contained in:
@@ -0,0 +1,308 @@
|
||||
// This file is part of Substrate.
|
||||
|
||||
// Copyright (C) Parity Technologies (UK) Ltd.
|
||||
// SPDX-License-Identifier: Apache-2.0
|
||||
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
//! Asset Conversion Ops pallet tests.
|
||||
|
||||
use crate::{mock::*, *};
|
||||
use frame_support::{
|
||||
assert_noop, assert_ok,
|
||||
traits::{
|
||||
fungible::{Inspect as FungibleInspect, NativeOrWithId},
|
||||
fungibles::{Create, Inspect},
|
||||
Incrementable,
|
||||
},
|
||||
};
|
||||
|
||||
#[test]
|
||||
fn migrate_pool_account_id_with_native() {
|
||||
new_test_ext().execute_with(|| {
|
||||
type PoolLocator = <Test as pallet_asset_conversion::Config>::PoolLocator;
|
||||
let user = 1;
|
||||
let token_1 = NativeOrWithId::Native;
|
||||
let token_2 = NativeOrWithId::WithId(2);
|
||||
let pool_id = PoolLocator::pool_id(&token_1, &token_2).unwrap();
|
||||
let lp_token =
|
||||
<Test as pallet_asset_conversion::Config>::PoolAssetId::initial_value().unwrap();
|
||||
|
||||
// setup pool and provide some liquidity.
|
||||
assert_ok!(NativeAndAssets::create(token_2.clone(), user, false, 1));
|
||||
|
||||
assert_ok!(AssetConversion::create_pool(
|
||||
RuntimeOrigin::signed(user),
|
||||
Box::new(token_1.clone()),
|
||||
Box::new(token_2.clone())
|
||||
));
|
||||
|
||||
let ed = Balances::minimum_balance();
|
||||
assert_ok!(Balances::force_set_balance(RuntimeOrigin::root(), user, 10000 * 2 + ed));
|
||||
assert_ok!(Assets::mint(RuntimeOrigin::signed(user), 2, user, 1000));
|
||||
|
||||
assert_ok!(AssetConversion::add_liquidity(
|
||||
RuntimeOrigin::signed(user),
|
||||
Box::new(token_1.clone()),
|
||||
Box::new(token_2.clone()),
|
||||
10000,
|
||||
10,
|
||||
10000,
|
||||
10,
|
||||
user,
|
||||
));
|
||||
|
||||
// assert user's balance.
|
||||
assert_eq!(NativeAndAssets::balance(token_1.clone(), &user), 10000 + ed);
|
||||
assert_eq!(NativeAndAssets::balance(token_2.clone(), &user), 1000 - 10);
|
||||
assert_eq!(PoolAssets::balance(lp_token, &user), 216);
|
||||
|
||||
// record total issuances before migration.
|
||||
let total_issuance_token1 = NativeAndAssets::total_issuance(token_1.clone());
|
||||
let total_issuance_token2 = NativeAndAssets::total_issuance(token_2.clone());
|
||||
let total_issuance_lp_token = PoolAssets::total_issuance(lp_token);
|
||||
|
||||
let pool_account = PoolLocator::address(&pool_id).unwrap();
|
||||
let (prior_pool_account, new_pool_account) =
|
||||
AssetConversionOps::addresses(&pool_id).unwrap();
|
||||
assert_eq!(pool_account, prior_pool_account);
|
||||
|
||||
// assert pool's balances before migration.
|
||||
assert_eq!(NativeAndAssets::balance(token_1.clone(), &prior_pool_account), 10000);
|
||||
assert_eq!(NativeAndAssets::balance(token_2.clone(), &prior_pool_account), 10);
|
||||
assert_eq!(PoolAssets::balance(lp_token, &prior_pool_account), 100);
|
||||
|
||||
// migrate.
|
||||
assert_ok!(AssetConversionOps::migrate_to_new_account(
|
||||
RuntimeOrigin::signed(user),
|
||||
Box::new(token_1.clone()),
|
||||
Box::new(token_2.clone()),
|
||||
));
|
||||
|
||||
// assert user's balance has not changed.
|
||||
assert_eq!(NativeAndAssets::balance(token_1.clone(), &user), 10000 + ed);
|
||||
assert_eq!(NativeAndAssets::balance(token_2.clone(), &user), 1000 - 10);
|
||||
assert_eq!(PoolAssets::balance(lp_token, &user), 216);
|
||||
|
||||
// assert pool's balance on new account id is same as on prior account id.
|
||||
assert_eq!(NativeAndAssets::balance(token_1.clone(), &new_pool_account), 10000);
|
||||
assert_eq!(NativeAndAssets::balance(token_2.clone(), &new_pool_account), 10);
|
||||
assert_eq!(PoolAssets::balance(lp_token, &new_pool_account), 100);
|
||||
|
||||
// assert pool's balance on prior account id is zero.
|
||||
assert_eq!(NativeAndAssets::balance(token_1.clone(), &prior_pool_account), 0);
|
||||
assert_eq!(NativeAndAssets::balance(token_2.clone(), &prior_pool_account), 0);
|
||||
assert_eq!(PoolAssets::balance(lp_token, &prior_pool_account), 0);
|
||||
|
||||
// assert total issuance has not changed.
|
||||
assert_eq!(total_issuance_token1, NativeAndAssets::total_issuance(token_1));
|
||||
assert_eq!(total_issuance_token2, NativeAndAssets::total_issuance(token_2));
|
||||
assert_eq!(total_issuance_lp_token, PoolAssets::total_issuance(lp_token));
|
||||
});
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn migrate_pool_account_id_with_insufficient_assets() {
|
||||
new_test_ext().execute_with(|| {
|
||||
type PoolLocator = <Test as pallet_asset_conversion::Config>::PoolLocator;
|
||||
let user = 1;
|
||||
let token_1 = NativeOrWithId::WithId(1);
|
||||
let token_2 = NativeOrWithId::WithId(2);
|
||||
let pool_id = PoolLocator::pool_id(&token_1, &token_2).unwrap();
|
||||
let lp_token =
|
||||
<Test as pallet_asset_conversion::Config>::PoolAssetId::initial_value().unwrap();
|
||||
|
||||
// setup pool and provide some liquidity.
|
||||
assert_ok!(NativeAndAssets::create(token_1.clone(), user, false, 1));
|
||||
assert_ok!(NativeAndAssets::create(token_2.clone(), user, false, 1));
|
||||
|
||||
assert_ok!(AssetConversion::create_pool(
|
||||
RuntimeOrigin::signed(user),
|
||||
Box::new(token_1.clone()),
|
||||
Box::new(token_2.clone())
|
||||
));
|
||||
|
||||
assert_ok!(Assets::mint(RuntimeOrigin::signed(user), 1, user, 20000));
|
||||
assert_ok!(Assets::mint(RuntimeOrigin::signed(user), 2, user, 1000));
|
||||
|
||||
assert_ok!(AssetConversion::add_liquidity(
|
||||
RuntimeOrigin::signed(user),
|
||||
Box::new(token_1.clone()),
|
||||
Box::new(token_2.clone()),
|
||||
10000,
|
||||
10,
|
||||
10000,
|
||||
10,
|
||||
user,
|
||||
));
|
||||
|
||||
// assert user's balance.
|
||||
assert_eq!(NativeAndAssets::balance(token_1.clone(), &user), 10000);
|
||||
assert_eq!(NativeAndAssets::balance(token_2.clone(), &user), 1000 - 10);
|
||||
assert_eq!(PoolAssets::balance(lp_token, &user), 216);
|
||||
|
||||
// record total issuances before migration.
|
||||
let total_issuance_token1 = NativeAndAssets::total_issuance(token_1.clone());
|
||||
let total_issuance_token2 = NativeAndAssets::total_issuance(token_2.clone());
|
||||
let total_issuance_lp_token = PoolAssets::total_issuance(lp_token);
|
||||
|
||||
let pool_account = PoolLocator::address(&pool_id).unwrap();
|
||||
let (prior_pool_account, new_pool_account) =
|
||||
AssetConversionOps::addresses(&pool_id).unwrap();
|
||||
assert_eq!(pool_account, prior_pool_account);
|
||||
|
||||
// assert pool's balances before migration.
|
||||
assert_eq!(NativeAndAssets::balance(token_1.clone(), &prior_pool_account), 10000);
|
||||
assert_eq!(NativeAndAssets::balance(token_2.clone(), &prior_pool_account), 10);
|
||||
assert_eq!(PoolAssets::balance(lp_token, &prior_pool_account), 100);
|
||||
|
||||
// migrate.
|
||||
assert_ok!(AssetConversionOps::migrate_to_new_account(
|
||||
RuntimeOrigin::signed(user),
|
||||
Box::new(token_1.clone()),
|
||||
Box::new(token_2.clone()),
|
||||
));
|
||||
|
||||
// assert user's balance has not changed.
|
||||
assert_eq!(NativeAndAssets::balance(token_1.clone(), &user), 10000);
|
||||
assert_eq!(NativeAndAssets::balance(token_2.clone(), &user), 1000 - 10);
|
||||
assert_eq!(PoolAssets::balance(lp_token, &user), 216);
|
||||
|
||||
// assert pool's balance on new account id is same as on prior account id.
|
||||
assert_eq!(NativeAndAssets::balance(token_1.clone(), &new_pool_account), 10000);
|
||||
assert_eq!(NativeAndAssets::balance(token_2.clone(), &new_pool_account), 10);
|
||||
assert_eq!(PoolAssets::balance(lp_token, &new_pool_account), 100);
|
||||
|
||||
// assert pool's balance on prior account id is zero.
|
||||
assert_eq!(NativeAndAssets::balance(token_1.clone(), &prior_pool_account), 0);
|
||||
assert_eq!(NativeAndAssets::balance(token_2.clone(), &prior_pool_account), 0);
|
||||
assert_eq!(PoolAssets::balance(lp_token, &prior_pool_account), 0);
|
||||
|
||||
// assert total issuance has not changed.
|
||||
assert_eq!(total_issuance_token1, NativeAndAssets::total_issuance(token_1));
|
||||
assert_eq!(total_issuance_token2, NativeAndAssets::total_issuance(token_2));
|
||||
assert_eq!(total_issuance_lp_token, PoolAssets::total_issuance(lp_token));
|
||||
});
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn migrate_pool_account_id_with_sufficient_assets() {
|
||||
new_test_ext().execute_with(|| {
|
||||
type PoolLocator = <Test as pallet_asset_conversion::Config>::PoolLocator;
|
||||
let user = 1;
|
||||
let token_1 = NativeOrWithId::WithId(1);
|
||||
let token_2 = NativeOrWithId::WithId(2);
|
||||
let pool_id = PoolLocator::pool_id(&token_1, &token_2).unwrap();
|
||||
let lp_token =
|
||||
<Test as pallet_asset_conversion::Config>::PoolAssetId::initial_value().unwrap();
|
||||
|
||||
// setup pool and provide some liquidity.
|
||||
assert_ok!(NativeAndAssets::create(token_1.clone(), user, true, 1));
|
||||
assert_ok!(NativeAndAssets::create(token_2.clone(), user, true, 1));
|
||||
|
||||
assert_ok!(AssetConversion::create_pool(
|
||||
RuntimeOrigin::signed(user),
|
||||
Box::new(token_1.clone()),
|
||||
Box::new(token_2.clone())
|
||||
));
|
||||
|
||||
assert_ok!(Assets::mint(RuntimeOrigin::signed(user), 1, user, 20000));
|
||||
assert_ok!(Assets::mint(RuntimeOrigin::signed(user), 2, user, 1000));
|
||||
|
||||
assert_ok!(AssetConversion::add_liquidity(
|
||||
RuntimeOrigin::signed(user),
|
||||
Box::new(token_1.clone()),
|
||||
Box::new(token_2.clone()),
|
||||
10000,
|
||||
10,
|
||||
10000,
|
||||
10,
|
||||
user,
|
||||
));
|
||||
|
||||
// assert user's balance.
|
||||
assert_eq!(NativeAndAssets::balance(token_1.clone(), &user), 10000);
|
||||
assert_eq!(NativeAndAssets::balance(token_2.clone(), &user), 1000 - 10);
|
||||
assert_eq!(PoolAssets::balance(lp_token, &user), 216);
|
||||
|
||||
// record total issuances before migration.
|
||||
let total_issuance_token1 = NativeAndAssets::total_issuance(token_1.clone());
|
||||
let total_issuance_token2 = NativeAndAssets::total_issuance(token_2.clone());
|
||||
let total_issuance_lp_token = PoolAssets::total_issuance(lp_token);
|
||||
|
||||
let pool_account = PoolLocator::address(&pool_id).unwrap();
|
||||
let (prior_pool_account, new_pool_account) =
|
||||
AssetConversionOps::addresses(&pool_id).unwrap();
|
||||
assert_eq!(pool_account, prior_pool_account);
|
||||
|
||||
// assert pool's balances before migration.
|
||||
assert_eq!(NativeAndAssets::balance(token_1.clone(), &prior_pool_account), 10000);
|
||||
assert_eq!(NativeAndAssets::balance(token_2.clone(), &prior_pool_account), 10);
|
||||
assert_eq!(PoolAssets::balance(lp_token, &prior_pool_account), 100);
|
||||
|
||||
// migrate.
|
||||
assert_ok!(AssetConversionOps::migrate_to_new_account(
|
||||
RuntimeOrigin::signed(user),
|
||||
Box::new(token_1.clone()),
|
||||
Box::new(token_2.clone()),
|
||||
));
|
||||
|
||||
// assert user's balance has not changed.
|
||||
assert_eq!(NativeAndAssets::balance(token_1.clone(), &user), 10000);
|
||||
assert_eq!(NativeAndAssets::balance(token_2.clone(), &user), 1000 - 10);
|
||||
assert_eq!(PoolAssets::balance(lp_token, &user), 216);
|
||||
|
||||
// assert pool's balance on new account id is same as on prior account id.
|
||||
assert_eq!(NativeAndAssets::balance(token_1.clone(), &new_pool_account), 10000);
|
||||
assert_eq!(NativeAndAssets::balance(token_2.clone(), &new_pool_account), 10);
|
||||
assert_eq!(PoolAssets::balance(lp_token, &new_pool_account), 100);
|
||||
|
||||
// assert pool's balance on prior account id is zero.
|
||||
assert_eq!(NativeAndAssets::balance(token_1.clone(), &prior_pool_account), 0);
|
||||
assert_eq!(NativeAndAssets::balance(token_2.clone(), &prior_pool_account), 0);
|
||||
assert_eq!(PoolAssets::balance(lp_token, &prior_pool_account), 0);
|
||||
|
||||
// assert total issuance has not changed.
|
||||
assert_eq!(total_issuance_token1, NativeAndAssets::total_issuance(token_1));
|
||||
assert_eq!(total_issuance_token2, NativeAndAssets::total_issuance(token_2));
|
||||
assert_eq!(total_issuance_lp_token, PoolAssets::total_issuance(lp_token));
|
||||
});
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn migrate_empty_pool_account_id() {
|
||||
new_test_ext().execute_with(|| {
|
||||
let user = 1;
|
||||
let token_1 = NativeOrWithId::Native;
|
||||
let token_2 = NativeOrWithId::WithId(2);
|
||||
|
||||
// setup pool and provide some liquidity.
|
||||
assert_ok!(NativeAndAssets::create(token_2.clone(), user, false, 1));
|
||||
|
||||
assert_ok!(AssetConversion::create_pool(
|
||||
RuntimeOrigin::signed(user),
|
||||
Box::new(token_1.clone()),
|
||||
Box::new(token_2.clone())
|
||||
));
|
||||
|
||||
// migrate.
|
||||
assert_noop!(
|
||||
AssetConversionOps::migrate_to_new_account(
|
||||
RuntimeOrigin::signed(user),
|
||||
Box::new(token_1.clone()),
|
||||
Box::new(token_2.clone()),
|
||||
),
|
||||
Error::<Test>::ZeroBalance
|
||||
);
|
||||
});
|
||||
}
|
||||
Reference in New Issue
Block a user