ether_link/
lib.rs

1#[cfg(feature = "wallet")]
2pub mod wallet;
3#[cfg(feature = "wallet")]
4pub use wallet::Wallet;
5
6#[cfg(feature = "signature")]
7pub mod signature;
8#[cfg(feature = "signature")]
9pub use signature::EthereumSignature;
10
11#[cfg(test)]
12mod tests {
13    use super::*;
14
15    #[tokio::test]
16    async fn test_random_wallet_generation() {
17        let wallet = Wallet::new();
18        println!("Private Key: {}", wallet.privkey);
19        println!("Compressed Public Key: {}", wallet.pubkey_compressed);
20        println!("Uncompressed Public Key: {}", wallet.pubkey_uncompressed);
21        println!("Ethereum Address: {}", wallet.address);
22    }
23
24    #[tokio::test]
25    async fn test_wallet_from_private_key() {
26        let wallet = Wallet::from_private_key(Some("0x1b5667a5972dba94b9baf4026288e4b51fdbc0c9c896473d394b78c5bd2ee92e"));
27        println!("Private Key: {}", wallet.privkey);
28        println!("Compressed Public Key: {}", wallet.pubkey_compressed);
29        println!("Uncompressed Public Key: {}", wallet.pubkey_uncompressed);
30        println!("Ethereum Address: {}", wallet.address);
31    }
32    
33
34    #[tokio::test]
35    async fn test_sign_message() {
36        let wallet = Wallet::new();
37        let message = "Hello, world!";
38        let signature = wallet.sign_message(message).await.unwrap_or_else(|e| panic!("Failed to sign message: {}", e));
39        
40        // Parse the signature JSON string
41        let sig_json: serde_json::Value = serde_json::from_str(&signature).unwrap_or_else(|e| panic!("Failed to parse signature JSON: {}", e));
42        println!("Signature JSON: {}", serde_json::to_string_pretty(&sig_json).unwrap_or_else(|e| panic!("Failed to format JSON: {}", e)));
43        
44        // Verify the signature contains all expected fields
45        assert!(sig_json.get("address").is_some());
46        assert!(sig_json.get("signed_message").is_some());
47        assert!(sig_json.get("signature").is_some());
48    }
49
50    
51
52    #[tokio::test]
53    async fn test_check_signature_is_valid() {
54        let wallet = Wallet::new();
55        let message = "Hello, world!";
56        let signature_json = wallet.sign_message(&message).await.unwrap_or_else(|e| panic!("Failed to sign message: {}", e));
57        
58        // Parse the signature JSON to extract the components
59        let sig_json: serde_json::Value = serde_json::from_str(&signature_json).unwrap_or_else(|e| panic!("Failed to parse signature JSON: {}", e));
60        let address = sig_json["address"].as_str().unwrap().to_string();
61        let signed_message = sig_json["signed_message"].as_str().unwrap().to_string();
62        let signature = sig_json["signature"].as_str().unwrap().to_string();
63
64        println!("Address: {}", address);
65        println!("Signed Message: {}", signed_message);
66        println!("Signature: {}", signature);
67
68        let eth_signature = EthereumSignature::new(address, signed_message, signature);
69        let is_valid = eth_signature.verify().unwrap_or_else(|e| panic!("Failed to verify signature: {}", e));
70        println!("Signature is valid: {}", is_valid);
71        assert!(is_valid, "Signature verification failed");
72    }
73    #[tokio::test]
74    async fn test_encrypt_and_decrypt_message() {
75        let sender = Wallet::from_private_key(Some("0x1f87cca105d954f1ae230f6ccdf8ea2c7df5eec8bb5d6c706344b42b725bd07d"));
76        let receiver = Wallet::from_private_key(Some("0xa4df62cbf0d00a76c6bceb5c6606823b8f4edb69c8b68607621dcc0a2e87a766"));
77        let message = "message from d07d to a4df: hello friend...ggagagag.?";
78        let encrypted_message = sender.encrypt_message(message, &receiver.pubkey_compressed).unwrap_or_else(|e| panic!("Failed to encrypt message: {}", e));
79        println!("Encrypted Message from sender(d07d) to receiver(a4df): {}", encrypted_message);
80        println!("Length of encrypted message: {}", encrypted_message.len());
81        let decrypted_message = receiver.decrypt_message(&encrypted_message).unwrap_or_else(|e| panic!("Failed to decrypt message: {}", e));
82        println!("Decrypted Message from receiver(a4df) to sender(d07d): {}", decrypted_message);
83    }
84}
ether_link/lib.rs

ether_link/
lib.rs
