Snowbridge westend migration(no need to merge)

bridges/snowbridge/pallets/system/src/migration.rs

@@ -2,9 +2,9 @@
 // SPDX-FileCopyrightText: 2023 Snowfork <hello@snowfork.com>
 //! Governance API for controlling the Ethereum side of the bridge
 use super::*;
-use frame_support::traits::OnRuntimeUpgrade;
+use frame_support::{traits::OnRuntimeUpgrade, StorageHasher};
 use log;
-
+use sp_core::storage::StorageKey;
 #[cfg(feature = "try-runtime")]
 use sp_runtime::TryRuntimeError;
 
@@ -72,3 +72,33 @@ pub mod v0 {
 		}
 	}
 }
+
+pub const PALLET_NAME: &str = "EthereumSystem";
+
+pub fn storage_map_final_key<H: StorageHasher>(
+	pallet_prefix: &str,
+	map_name: &str,
+	key: &[u8],
+) -> StorageKey {
+	let key_hashed = H::hash(key);
+	let pallet_prefix_hashed = frame_support::Twox128::hash(pallet_prefix.as_bytes());
+	let storage_prefix_hashed = frame_support::Twox128::hash(map_name.as_bytes());
+
+	let mut final_key = Vec::with_capacity(
+		pallet_prefix_hashed.len() + storage_prefix_hashed.len() + key_hashed.as_ref().len(),
+	);
+
+	final_key.extend_from_slice(&pallet_prefix_hashed[..]);
+	final_key.extend_from_slice(&storage_prefix_hashed[..]);
+	final_key.extend_from_slice(key_hashed.as_ref());
+
+	StorageKey(final_key)
+}
+
+pub fn agent_key(agent_id: AgentId) -> StorageKey {
+	storage_map_final_key::<Twox64Concat>(PALLET_NAME, "Agents", &agent_id.encode())
+}
+
+pub fn channel_key(channel_id: ChannelId) -> StorageKey {
+	storage_map_final_key::<Twox64Concat>(PALLET_NAME, "Channels", &channel_id.encode())
+}

bridges/snowbridge/pallets/system/src/tests.rs

@@ -1,10 +1,14 @@
 // SPDX-License-Identifier: Apache-2.0
 // SPDX-FileCopyrightText: 2023 Snowfork <hello@snowfork.com>
-use crate::{mock::*, *};
-use frame_support::{assert_noop, assert_ok};
+use crate::{
+	migration::{agent_key, channel_key},
+	mock::*,
+	*,
+};
+use frame_support::{assert_noop, assert_ok, storage::generator::StorageMap};
 use hex_literal::hex;
 use snowbridge_core::eth;
-use sp_core::H256;
+use sp_core::{bytes::to_hex, H256};
 use sp_runtime::{AccountId32, DispatchError::BadOrigin, TokenError};
 
 #[test]
@@ -744,3 +748,57 @@ fn register_ethereum_native_token_fails() {
 		);
 	});
 }
+
+#[test]
+fn initialize_bridge() {
+	new_test_ext(false).execute_with(|| {
+		let asset_hub_location: Location =
+			ParentThen(Parachain(AssetHubParaId::get().into()).into()).into();
+		let asset_hub_agent_id = agent_id_of::<Test>(&asset_hub_location).unwrap();
+		let asset_hub_channel_id: ChannelId = AssetHubParaId::get().into();
+		let bridge_hub_agent_id = agent_id_of::<Test>(&Location::here()).unwrap();
+
+		assert_eq!(
+			Agents::<Test>::storage_map_final_key(bridge_hub_agent_id),
+			agent_key(bridge_hub_agent_id).0
+		);
+
+		assert_eq!(Channels::<Test>::contains_key(PRIMARY_GOVERNANCE_CHANNEL), false);
+
+		let key_value_items = vec![
+			(agent_key(bridge_hub_agent_id).0, ().encode()),
+			(agent_key(asset_hub_agent_id).0, ().encode()),
+			(
+				channel_key(PRIMARY_GOVERNANCE_CHANNEL).0,
+				Channel { agent_id: bridge_hub_agent_id, para_id: BridgeHubParaId::get() }.encode(),
+			),
+			(
+				channel_key(SECONDARY_GOVERNANCE_CHANNEL).0,
+				Channel { agent_id: bridge_hub_agent_id, para_id: BridgeHubParaId::get() }.encode(),
+			),
+			(
+				channel_key(asset_hub_channel_id).0,
+				Channel { agent_id: asset_hub_agent_id, para_id: AssetHubParaId::get().into() }
+					.encode(),
+			),
+		];
+		assert_ok!(System::set_storage(RuntimeOrigin::root(), key_value_items.clone()));
+
+		assert_eq!(Channels::<Test>::contains_key(PRIMARY_GOVERNANCE_CHANNEL), true);
+
+		assert_eq!(Channels::<Test>::contains_key(asset_hub_channel_id), true);
+
+		let channel = Channels::<Test>::get(asset_hub_channel_id).unwrap();
+
+		assert_eq!(channel.agent_id, asset_hub_agent_id);
+
+		assert_eq!(EthereumSystem::is_initialized(), true);
+
+		let set_storage_call: <Test as frame_system::Config>::RuntimeCall =
+			frame_system::Call::<Test>::set_storage { items: key_value_items.clone() }.into();
+
+		assert_eq!(to_hex(&set_storage_call.encode(), false),
+		"0x0004142101ccee781f0b9380204db9882d1b1c771d0ec3e268c3abf5d87f15dde43fc56ec13a4d74bbd00efc4503170a2e7597b7b7e3d84c05391d139a62b157e78786d8c082f29dcf4c111314002101ccee781f0b9380204db9882d1b1c771d0ec3e268c3abf5d87f15dde43fc56ec14f4be588bc0653e381c5ab2571199e3188135178f3c2c8e2d268be1313d029b30f534fa579b69b79002101ccee781f0b9380204db9882d1b1c771d7759ccc18de172078306ca5973251a7b874cb762a9d6a66800000000000000000000000000000000000000000000000000000000000000019003170a2e7597b7b7e3d84c05391d139a62b157e78786d8c082f29dcf4c111314ea0300002101ccee781f0b9380204db9882d1b1c771d7759ccc18de172078306ca5973251a7bbf8a4d00acb08a4700000000000000000000000000000000000000000000000000000000000000029003170a2e7597b7b7e3d84c05391d139a62b157e78786d8c082f29dcf4c111314ea0300002101ccee781f0b9380204db9882d1b1c771d7759ccc18de172078306ca5973251a7b594aa8a9c557dabac173fac324158e77fb5840738a1a541f633cbec8884c6a601c567d2b376a05399081c5ab2571199e3188135178f3c2c8e2d268be1313d029b30f534fa579b69b79e8030000"
+		);
+	});
+}