Fix test

paritytech · Sep 3, 2024 · 26263ac · 26263ac
1 parent ae0f674
commit 26263ac
Showing 1 changed file with 24 additions and 16 deletions.
diff --git a/bridges/snowbridge/primitives/router/src/outbound/tests.rs b/bridges/snowbridge/primitives/router/src/outbound/tests.rs
@@ -424,11 +424,13 @@ fn xcm_converter_convert_success() {
 	.into();
 	let mut converter =
 		XcmConverter::<MockTokenIdConvert, ()>::new(&message, network, Default::default());
-	let expected_payload = Command::TransferNativeToken {
+	let expected_payload = Command::AgentExecute {
 		agent_id: Default::default(),
-		token: token_address.into(),
-		recipient: beneficiary_address.into(),
-		amount: 1000,
+		command: AgentExecuteCommand::TransferToken {
+			token: token_address.into(),
+			recipient: beneficiary_address.into(),
+			amount: 1000,
+		},
 	};
 	let result = converter.convert();
 	assert_eq!(result, Ok((expected_payload, [0; 32])));
@@ -459,11 +461,13 @@ fn xcm_converter_convert_without_buy_execution_yields_success() {
 	.into();
 	let mut converter =
 		XcmConverter::<MockTokenIdConvert, ()>::new(&message, network, Default::default());
-	let expected_payload = Command::TransferNativeToken {
+	let expected_payload = Command::AgentExecute {
 		agent_id: Default::default(),
-		token: token_address.into(),
-		recipient: beneficiary_address.into(),
-		amount: 1000,
+		command: AgentExecuteCommand::TransferToken {
+			token: token_address.into(),
+			recipient: beneficiary_address.into(),
+			amount: 1000,
+		},
 	};
 	let result = converter.convert();
 	assert_eq!(result, Ok((expected_payload, [0; 32])));
@@ -496,11 +500,13 @@ fn xcm_converter_convert_with_wildcard_all_asset_filter_succeeds() {
 	.into();
 	let mut converter =
 		XcmConverter::<MockTokenIdConvert, ()>::new(&message, network, Default::default());
-	let expected_payload = Command::TransferNativeToken {
+	let expected_payload = Command::AgentExecute {
 		agent_id: Default::default(),
-		token: token_address.into(),
-		recipient: beneficiary_address.into(),
-		amount: 1000,
+		command: AgentExecuteCommand::TransferToken {
+			token: token_address.into(),
+			recipient: beneficiary_address.into(),
+			amount: 1000,
+		},
 	};
 	let result = converter.convert();
 	assert_eq!(result, Ok((expected_payload, [0; 32])));
@@ -533,11 +539,13 @@ fn xcm_converter_convert_with_fees_less_than_reserve_yields_success() {
 	.into();
 	let mut converter =
 		XcmConverter::<MockTokenIdConvert, ()>::new(&message, network, Default::default());
-	let expected_payload = Command::TransferNativeToken {
+	let expected_payload = Command::AgentExecute {
 		agent_id: Default::default(),
-		token: token_address.into(),
-		recipient: beneficiary_address.into(),
-		amount: 1000,
+		command: AgentExecuteCommand::TransferToken {
+			token: token_address.into(),
+			recipient: beneficiary_address.into(),
+			amount: 1000,
+		},
 	};
 	let result = converter.convert();
 	assert_eq!(result, Ok((expected_payload, [0; 32])));