309 lines
12 KiB
Rust
309 lines
12 KiB
Rust
// 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
|
|
);
|
|
});
|
|
}
|