modify iro allocation (#486)

Author: dharjeezy

modules/pallets/bridge-drop/src/lib.rs

@@ -267,6 +267,7 @@ pub mod pallet {
 			origin: OriginFor<T>,
 			beneficiary: T::AccountId,
 			amount: <<T as Config>::Currency as Currency<T::AccountId>>::Balance,
+			bonus_amount: <<T as Config>::Currency as Currency<T::AccountId>>::Balance,
 		) -> DispatchResult {
 			T::BridgeDropOrigin::ensure_origin(origin)?;
 
@@ -278,12 +279,13 @@ pub mod pallet {
 			let percent = Permill::from_parts(250_000);
 			let unlocked_balance = percent * u128::from(amount);
 
-			let locked = u128::from(amount).saturating_sub(unlocked_balance);
+			let total_amount = u128::from(amount).saturating_add(u128::from(bonus_amount));
+			let locked = total_amount.saturating_sub(unlocked_balance);
 
 			<<T as Config>::Currency as Currency<T::AccountId>>::transfer(
 				&Self::account_id(),
 				&beneficiary,
-				amount.into(),
+				total_amount.into(),
 				ExistenceRequirement::AllowDeath,
 			)?;
 
@@ -299,9 +301,9 @@ pub mod pallet {
 				starting_block,
 			)?;
 
-			IroAllocations::<T>::insert(&beneficiary, u128::from(amount));
+			IroAllocations::<T>::insert(&beneficiary, total_amount);
 
-			Self::deposit_event(Event::<T>::Allocated { beneficiary, amount });
+			Self::deposit_event(Event::<T>::Allocated { beneficiary, amount: total_amount.into() });
 
 			Ok(())
 		}

modules/pallets/testsuite/src/tests/pallet_bridge_airdrop.rs

@@ -138,20 +138,24 @@ fn should_allocate_iro_correctly() {
 
 		let beneficiary = AccountId32::new(H256::random().0);
 		let amount = 3500_000_000_000_000u128;
+		let bonus_amount = 3500_000_000_000u128;
 
 		pallet_bridge_airdrop::Pallet::<Test>::allocate_iro_tokens(
 			RuntimeOrigin::root(),
 			beneficiary.clone(),
 			amount,
+			bonus_amount
 		)
 		.unwrap();
 
 		let account_data = frame_system::Account::<Test>::get(beneficiary.clone());
 
 		let initial_unlocked = Permill::from_parts(250_000) * amount;
-		let locked = amount.saturating_sub(initial_unlocked);
+		let locked = amount
+			.saturating_sub(initial_unlocked)
+			.saturating_add(bonus_amount);
 
-		assert_eq!(account_data.data.free, amount);
+		assert_eq!(account_data.data.free, amount.saturating_add(bonus_amount));
 
 		// transfer above unlocked balance should fail
 		let res = Balances::transfer_keep_alive(
@@ -189,6 +193,7 @@ fn should_allocate_iro_correctly() {
 			RuntimeOrigin::root(),
 			beneficiary.clone(),
 			amount.clone(),
+			bonus_amount
 		);
 
 		assert_err!(res, pallet_bridge_airdrop::pallet::Error::<Test>::AlreadyAllocated);

parachain/runtimes/gargantua/src/lib.rs

@@ -239,7 +239,7 @@ pub const VERSION: RuntimeVersion = RuntimeVersion {
 	spec_name: Cow::Borrowed("gargantua"),
 	impl_name: Cow::Borrowed("gargantua"),
 	authoring_version: 1,
-	spec_version: 3_800,
+	spec_version: 3_900,
 	impl_version: 0,
 	apis: RUNTIME_API_VERSIONS,
 	transaction_version: 1,

parachain/simtests/allocations/iro_allocations.json

@@ -1,14 +1,17 @@
 [
   {
     "beneficiary": "16kZJGPJ37uYxjs7adswyEHbPYeHS9jQHSaSUJhkfvWPcoeF",
-    "amount": 3076000000000000
+    "amount": 3076000000000000,
+    "bonus_amount": 30760000000000
   },
   {
     "beneficiary": "1egYCubF1U5CGWiXjQnsXduiJYP49KTs8eX1jn1JrTqCYyQ",
-    "amount": 1661040000000000
+    "amount": 1661040000000000,
+    "bonus_amount": 1661040000000
   },
   {
     "beneficiary": "126TwBzBM4jUEK2gTphmW4oLoBWWnYvPp8hygmduTr4uds57",
-    "amount": 67130863680000000
+    "amount": 67130863680000000,
+    "bonus_amount": 1661040000
   }
 ]

parachain/simtests/src/token_allocation.rs

@@ -24,6 +24,8 @@ use subxt_utils::Hyperbridge;
 struct AllocationRecord {
 	beneficiary: String,
 	amount: u128,
+	#[serde(default)]
+	bonus_amount: Option<u128>,
 }
 
 #[tokio::test]
@@ -79,6 +81,7 @@ async fn should_perform_batch_allocations() -> Result<(), anyhow::Error> {
 			let call = RuntimeCall::BridgeDrop(BridgeDropCall::allocate_iro_tokens {
 				beneficiary: dest.into(),
 				amount: record.amount,
+				bonus_amount: record.bonus_amount.unwrap()
 			});
 			calls.push(call);
 			records_to_check.push(record);