Asset registry genesis (#803)

* asset registry genesis

* fmt

* clone instead of copy

Author: LauraReesby

asset-registry/src/lib.rs

@@ -40,7 +40,7 @@ pub mod module {
 		type CustomMetadata: Parameter + Member + TypeInfo;
 
 		/// The type used as a unique asset id,
-		type AssetId: Parameter + Member + Default + TypeInfo;
+		type AssetId: Parameter + Member + Default + TypeInfo + MaybeSerializeDeserialize;
 
 		/// Checks that an origin has the authority to register/update an asset
 		type AuthorityOrigin: EnsureOriginWithArg<Self::Origin, Option<Self::AssetId>>;
@@ -103,21 +103,31 @@ pub mod module {
 
 	#[pallet::genesis_config]
 	pub struct GenesisConfig<T: Config> {
-		_phantom: PhantomData<T>,
+		pub assets: Vec<(T::AssetId, Vec<u8>)>,
+		pub last_asset_id: T::AssetId,
 	}
 
 	#[cfg(feature = "std")]
 	impl<T: Config> Default for GenesisConfig<T> {
 		fn default() -> Self {
 			Self {
-				_phantom: Default::default(),
+				assets: vec![],
+				last_asset_id: Default::default(),
 			}
 		}
 	}
 
 	#[pallet::genesis_build]
 	impl<T: Config> GenesisBuild<T> for GenesisConfig<T> {
-		fn build(&self) {}
+		fn build(&self) {
+			self.assets.iter().for_each(|(asset_id, metadata_encoded)| {
+				let metadata = AssetMetadata::decode(&mut &metadata_encoded[..]).expect("Error decoding AssetMetadata");
+				Pallet::<T>::do_register_asset_without_asset_processor(metadata, asset_id.clone())
+					.expect("Error registering Asset");
+			});
+
+			LastAssetId::<T>::set(self.last_asset_id.clone());
+		}
 	}
 
 	#[pallet::pallet]

asset-registry/src/mock/mod.rs

@@ -109,7 +109,7 @@ decl_test_parachain! {
 		Runtime = para::Runtime,
 		XcmpMessageHandler = para::XcmpQueue,
 		DmpMessageHandler = para::DmpQueue,
-		new_ext = para_ext(1),
+		new_ext = para_ext(1, None),
 	}
 }
 
@@ -118,7 +118,7 @@ decl_test_parachain! {
 		Runtime = para::Runtime,
 		XcmpMessageHandler = para::XcmpQueue,
 		DmpMessageHandler = para::DmpQueue,
-		new_ext = para_ext(2),
+		new_ext = para_ext(2, None),
 	}
 }
 
@@ -127,7 +127,41 @@ decl_test_parachain! {
 		Runtime = para::Runtime,
 		XcmpMessageHandler = para::XcmpQueue,
 		DmpMessageHandler = para::DmpQueue,
-		new_ext = para_ext(3),
+		new_ext = para_ext(3, None),
+	}
+}
+
+decl_test_parachain! {
+	pub struct ParaG {
+		Runtime = para::Runtime,
+		XcmpMessageHandler = para::XcmpQueue,
+		DmpMessageHandler = para::DmpQueue,
+		new_ext = para_ext(4, Some((
+			vec![(
+				4,
+				AssetMetadata::<Balance, para::CustomMetadata>::encode(&AssetMetadata {
+				decimals: 12,
+				name: "para G native token".as_bytes().to_vec(),
+				symbol: "paraG".as_bytes().to_vec(),
+				existential_deposit: 0,
+				location: None,
+				additional: para::CustomMetadata {
+					fee_per_second: 1_000_000_000_000,
+				},
+			})),
+			(
+				5,
+				AssetMetadata::<Balance, para::CustomMetadata>::encode(&AssetMetadata {
+				decimals: 12,
+				name: "para G foreign token".as_bytes().to_vec(),
+				symbol: "paraF".as_bytes().to_vec(),
+				existential_deposit: 0,
+				location: None,
+				additional: para::CustomMetadata {
+					fee_per_second: 1_000_000_000_000,
+				},
+			}))], 5
+		))),
 	}
 }
 
@@ -146,14 +180,15 @@ decl_test_network! {
 			(1, ParaA),
 			(2, ParaB),
 			(3, ParaC),
+			(4, ParaG),
 		],
 	}
 }
 
 pub type ParaTokens = orml_tokens::Pallet<para::Runtime>;
 pub type ParaXTokens = orml_xtokens::Pallet<para::Runtime>;
 
-pub fn para_ext(para_id: u32) -> TestExternalities {
+pub fn para_ext(para_id: u32, asset_data: Option<(Vec<(u32, Vec<u8>)>, u32)>) -> TestExternalities {
 	use para::{Runtime, System};
 
 	let mut t = frame_system::GenesisConfig::default()
@@ -172,6 +207,15 @@ pub fn para_ext(para_id: u32) -> TestExternalities {
 	.assimilate_storage(&mut t)
 	.unwrap();
 
+	if let Some((assets, last_asset_id)) = asset_data {
+		GenesisConfig::<Runtime> {
+			assets: assets,
+			last_asset_id: last_asset_id,
+		}
+		.assimilate_storage(&mut t)
+		.unwrap();
+	}
+
 	let mut ext = TestExternalities::new(t);
 	ext.execute_with(|| System::set_block_number(1));
 	ext

asset-registry/src/tests.rs

@@ -53,6 +53,37 @@ fn dummy_metadata() -> AssetMetadata<<para::Runtime as orml_asset_registry::Conf
 	}
 }
 
+#[test]
+fn genesis_issuance_should_work() {
+	TestNet::reset();
+
+	ParaG::execute_with(|| {
+		let metadata1 = AssetMetadata {
+			decimals: 12,
+			name: "para G native token".as_bytes().to_vec(),
+			symbol: "paraG".as_bytes().to_vec(),
+			existential_deposit: 0,
+			location: None,
+			additional: CustomMetadata {
+				fee_per_second: 1_000_000_000_000,
+			},
+		};
+		let metadata2 = AssetMetadata {
+			decimals: 12,
+			name: "para G foreign token".as_bytes().to_vec(),
+			symbol: "paraF".as_bytes().to_vec(),
+			existential_deposit: 0,
+			location: None,
+			additional: CustomMetadata {
+				fee_per_second: 1_000_000_000_000,
+			},
+		};
+		assert_eq!(AssetRegistry::metadata(4).unwrap(), metadata1);
+		assert_eq!(AssetRegistry::metadata(5).unwrap(), metadata2);
+		assert_eq!(LastAssetId::<para::Runtime>::get(), 5);
+	});
+}
+
 #[test]
 /// test that the asset registry can be used in xcm transfers
 fn send_self_parachain_asset_to_sibling() {