refactor: match statement

r0gue-io · Jul 31, 2024 · 51c38cc · 51c38cc
1 parent 5aec185
commit 51c38cc
Showing 1 changed file with 22 additions and 44 deletions.
diff --git a/pop-api/integration-tests/src/local_fungibles.rs b/pop-api/integration-tests/src/local_fungibles.rs
@@ -441,21 +441,17 @@ fn total_supply(addr: AccountId32, asset_id: AssetId) -> Result<Balance, Error>
 	let function = function_selector("total_supply");
 	let params = [function, asset_id.encode()].concat();
 	let result = bare_call(addr, params, 0).expect("should work");
-	match decoded::<Result<Balance, Error>>(result) {
-		Ok(x) => x,
-		Err(result) => panic!("Contract reverted: {:?}", result),
-	}
+	decoded::<Result<Balance, Error>>(result.clone())
+		.unwrap_or_else(|_| panic!("Contract reverted: {:?}", result))
 }
 
 // Call balance_of contract message.
 fn balance_of(addr: AccountId32, asset_id: AssetId, owner: AccountId32) -> Result<Balance, Error> {
 	let function = function_selector("balance_of");
 	let params = [function, asset_id.encode(), owner.encode()].concat();
 	let result = bare_call(addr, params, 0).expect("should work");
-	match decoded::<Result<Balance, Error>>(result) {
-		Ok(x) => x,
-		Err(result) => panic!("Contract reverted: {:?}", result),
-	}
+	decoded::<Result<Balance, Error>>(result.clone())
+		.unwrap_or_else(|_| panic!("Contract reverted: {:?}", result))
 }
 
 // Call allowance contract message.
@@ -468,43 +464,35 @@ fn allowance(
 	let function = function_selector("allowance");
 	let params = [function, asset_id.encode(), owner.encode(), spender.encode()].concat();
 	let result = bare_call(addr, params, 0).expect("should work");
-	match decoded::<Result<Balance, Error>>(result) {
-		Ok(x) => x,
-		Err(result) => panic!("Contract reverted: {:?}", result),
-	}
+	decoded::<Result<Balance, Error>>(result.clone())
+		.unwrap_or_else(|_| panic!("Contract reverted: {:?}", result))
 }
 
 // Call token_name contract message.
 fn token_name(addr: AccountId32, asset_id: AssetId) -> Result<Vec<u8>, Error> {
 	let function = function_selector("token_name");
 	let params = [function, asset_id.encode()].concat();
 	let result = bare_call(addr, params, 0).expect("should work");
-	match decoded::<Result<Vec<u8>, Error>>(result) {
-		Ok(x) => x,
-		Err(result) => panic!("Contract reverted: {:?}", result),
-	}
+	decoded::<Result<Vec<u8>, Error>>(result.clone())
+		.unwrap_or_else(|_| panic!("Contract reverted: {:?}", result))
 }
 
 // Call token_symbol contract message.
 fn token_symbol(addr: AccountId32, asset_id: AssetId) -> Result<Vec<u8>, Error> {
 	let function = function_selector("token_symbol");
 	let params = [function, asset_id.encode()].concat();
 	let result = bare_call(addr, params, 0).expect("should work");
-	match decoded::<Result<Vec<u8>, Error>>(result) {
-		Ok(x) => x,
-		Err(result) => panic!("Contract reverted: {:?}", result),
-	}
+	decoded::<Result<Vec<u8>, Error>>(result.clone())
+		.unwrap_or_else(|_| panic!("Contract reverted: {:?}", result))
 }
 
 // Call token_decimals contract message.
 fn token_decimals(addr: AccountId32, asset_id: AssetId) -> Result<u8, Error> {
 	let function = function_selector("token_decimals");
 	let params = [function, asset_id.encode()].concat();
 	let result = bare_call(addr, params, 0).expect("should work");
-	match decoded::<Result<u8, Error>>(result) {
-		Ok(x) => x,
-		Err(result) => panic!("Contract reverted: {:?}", result),
-	}
+	decoded::<Result<u8, Error>>(result.clone())
+		.unwrap_or_else(|_| panic!("Contract reverted: {:?}", result))
 }
 
 fn transfer(
@@ -516,10 +504,8 @@ fn transfer(
 	let function = function_selector("transfer");
 	let params = [function, asset_id.encode(), to.encode(), value.encode()].concat();
 	let result = bare_call(addr, params, 0).expect("should work");
-	match decoded::<Result<(), Error>>(result) {
-		Ok(x) => x,
-		Err(result) => panic!("Contract reverted: {:?}", result),
-	}
+	decoded::<Result<(), Error>>(result.clone())
+		.unwrap_or_else(|_| panic!("Contract reverted: {:?}", result))
 }
 
 fn transfer_from(
@@ -535,10 +521,8 @@ fn transfer_from(
 		[function, asset_id.encode(), from.encode(), to.encode(), value.encode(), data.encode()]
 			.concat();
 	let result = bare_call(addr, params, 0).expect("should work");
-	match decoded::<Result<(), Error>>(result) {
-		Ok(x) => x,
-		Err(result) => panic!("Contract reverted: {:?}", result),
-	}
+	decoded::<Result<(), Error>>(result.clone())
+		.unwrap_or_else(|_| panic!("Contract reverted: {:?}", result))
 }
 
 fn approve(
@@ -550,10 +534,8 @@ fn approve(
 	let function = function_selector("approve");
 	let params = [function, asset_id.encode(), spender.encode(), value.encode()].concat();
 	let result = bare_call(addr, params, 0).expect("should work");
-	match decoded::<Result<(), Error>>(result) {
-		Ok(x) => x,
-		Err(result) => panic!("Contract reverted: {:?}", result),
-	}
+	decoded::<Result<(), Error>>(result.clone())
+		.unwrap_or_else(|_| panic!("Contract reverted: {:?}", result))
 }
 
 fn increase_allowance(
@@ -565,10 +547,8 @@ fn increase_allowance(
 	let function = function_selector("increase_allowance");
 	let params = [function, asset_id.encode(), spender.encode(), value.encode()].concat();
 	let result = bare_call(addr, params, 0).expect("should work");
-	match decoded::<Result<(), Error>>(result) {
-		Ok(x) => x,
-		Err(result) => panic!("Contract reverted: {:?}", result),
-	}
+	decoded::<Result<(), Error>>(result.clone())
+		.unwrap_or_else(|_| panic!("Contract reverted: {:?}", result))
 }
 
 fn decrease_allowance(
@@ -580,10 +560,8 @@ fn decrease_allowance(
 	let function = function_selector("decrease_allowance");
 	let params = [function, asset_id.encode(), spender.encode(), value.encode()].concat();
 	let result = bare_call(addr, params, 0).expect("should work");
-	match decoded::<Result<(), Error>>(result) {
-		Ok(x) => x,
-		Err(result) => panic!("Contract reverted: {:?}", result),
-	}
+	decoded::<Result<(), Error>>(result.clone())
+		.unwrap_or_else(|_| panic!("Contract reverted: {:?}", result))
 }
 
 // fn asset_exists(addr: AccountId32, asset_id: AssetId) -> bool {