From 26263acee7bb734ec80538bffc67a69d906f6c3f Mon Sep 17 00:00:00 2001
From: ron <yrong1997@gmail.com>
Date: Tue, 3 Sep 2024 08:49:38 +0800
Subject: [PATCH] Fix test

---
 .../primitives/router/src/outbound/tests.rs   | 40 +++++++++++--------
 1 file changed, 24 insertions(+), 16 deletions(-)

diff --git a/bridges/snowbridge/primitives/router/src/outbound/tests.rs b/bridges/snowbridge/primitives/router/src/outbound/tests.rs
index 75dce864493a..6e4fd5946340 100644
--- 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])));