test: add sig test case (#5969)

Author: mattsse

crates/cast/bin/main.rs

@@ -7,7 +7,7 @@ use ethers::{
     types::Address,
     utils::keccak256,
 };
-use eyre::Result;
+use eyre::{Result, WrapErr};
 use foundry_cli::{handler, prompt, stdin, utils};
 use foundry_common::{
     abi::{format_tokens, get_event},
@@ -174,7 +174,7 @@ async fn main() -> Result<()> {
             println!("{}", pretty_calldata(&calldata, offline).await?);
         }
         Subcommands::Sig { sig, optimize } => {
-            let sig = stdin::unwrap_line(sig)?;
+            let sig = stdin::unwrap_line(sig).wrap_err("Failed to read signature")?;
             if optimize.is_none() {
                 println!("{}", SimpleCast::get_selector(&sig, None)?.0);
             } else {

crates/cast/bin/opts.rs

@@ -886,6 +886,7 @@ pub fn parse_slot(s: &str) -> Result<H256> {
 #[cfg(test)]
 mod tests {
     use super::*;
+    use cast::SimpleCast;
     use ethers::types::BlockNumber;
 
     #[test]
@@ -908,6 +909,29 @@ mod tests {
         };
     }
 
+    // <https://github.com/foundry-rs/book/issues/1019>
+    #[test]
+    fn parse_signature() {
+        let args: Opts = Opts::parse_from([
+            "foundry-cli",
+            "sig",
+            "__$_$__$$$$$__$$_$$$_$$__$$___$$(address,address,uint256)",
+        ]);
+        match args.sub {
+            Subcommands::Sig { sig, .. } => {
+                let sig = sig.unwrap();
+                assert_eq!(
+                    sig,
+                    "__$_$__$$$$$__$$_$$$_$$__$$___$$(address,address,uint256)".to_string()
+                );
+
+                let selector = SimpleCast::get_selector(&sig, None).unwrap();
+                assert_eq!(selector.0, "0x23b872dd".to_string());
+            }
+            _ => unreachable!(),
+        };
+    }
+
     #[test]
     fn parse_block_ids() {
         struct TestCase {