Improve MockApi canonicalize error handling

Author: chipshort

contracts/reflect/src/contract.rs

@@ -353,7 +353,7 @@ mod tests {
         let err = execute(deps.as_mut(), mock_env(), info, msg).unwrap_err();
         match err {
             ReflectError::Std(StdError::GenericErr { msg, .. }) => {
-                assert!(msg.contains("Invalid input"))
+                assert!(msg.contains("Error decoding bech32"))
             }
             e => panic!("Unexpected error: {e:?}"),
         }

packages/std/src/testing/mock.rs

@@ -126,15 +126,20 @@ impl Api for MockApi {
     }
 
     fn addr_canonicalize(&self, input: &str) -> StdResult<CanonicalAddr> {
-        if let Ok((prefix, decoded, Variant::Bech32)) = decode(input) {
-            if prefix == self.bech32_prefix {
-                if let Ok(bytes) = Vec::<u8>::from_base32(&decoded) {
+        match decode(input) {
+            Ok((prefix, _, _)) if prefix != self.bech32_prefix => {
+                Err(StdError::generic_err("Wrong bech32 prefix"))
+            }
+            Ok((_, _, Variant::Bech32m)) => Err(StdError::generic_err("Wrong bech32 variant")),
+            Err(_) => Err(StdError::generic_err("Error decoding bech32")),
+            Ok((_, decoded, Variant::Bech32)) => match Vec::<u8>::from_base32(&decoded) {
+                Ok(bytes) => {
                     validate_length(&bytes)?;
-                    return Ok(bytes.into());
+                    Ok(bytes.into())
                 }
-            }
+                Err(_) => Err(StdError::generic_err("Invalid bech32 data")),
+            },
         }
-        Err(StdError::generic_err("Invalid input"))
     }
 
     fn addr_humanize(&self, canonical: &CanonicalAddr) -> StdResult<Addr> {
@@ -1235,9 +1240,9 @@ mod tests {
     fn addr_canonicalize_long_input() {
         let api = MockApi::default();
         let human =
-            "some-extremely-long-address-some-extremely-long-address-some-extremely-long-address-some-extremely-long-address";
+            "cosmwasm1qqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqehqqkz";
         let err = api.addr_canonicalize(human).unwrap_err();
-        assert!(err.to_string().contains("Invalid input"));
+        assert!(err.to_string().contains("Invalid canonical address length"));
     }
 
     #[test]

packages/vm/src/testing/mock.rs

@@ -131,15 +131,26 @@ impl BackendApi for MockApi {
             MockApi::Bech32 { bech32_prefix } => *bech32_prefix,
         };
 
-        if let Ok((prefix, decoded, Variant::Bech32)) = decode(input) {
-            if prefix == bech32_prefix {
-                if let Ok(bytes) = Vec::<u8>::from_base32(&decoded) {
+        match decode(input) {
+            Ok((prefix, _, _)) if prefix != bech32_prefix => {
+                (Err(BackendError::user_err("Wrong bech32 prefix")), gas_info)
+            }
+            Ok((_, _, Variant::Bech32m)) => (
+                Err(BackendError::user_err("Wrong bech32 variant")),
+                gas_info,
+            ),
+            Err(_) => (
+                Err(BackendError::user_err("Error decoding bech32")),
+                gas_info,
+            ),
+            Ok((_, decoded, Variant::Bech32)) => match Vec::<u8>::from_base32(&decoded) {
+                Ok(bytes) => {
                     try_br!((validate_length(&bytes), gas_info));
-                    return (Ok(bytes), gas_info);
+                    (Ok(bytes), gas_info)
                 }
-            }
+                Err(_) => (Err(BackendError::user_err("Invalid bech32 data")), gas_info),
+            },
         }
-        (Err(BackendError::user_err("Invalid input")), gas_info)
     }
 
     fn human_address(&self, canonical: &[u8]) -> BackendResult<String> {