rustao 0.2.0

Rust SDK for the AO protocol on Arweave
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

#[cfg(test)]
mod tests {
    use rsa::{RsaPrivateKey, RsaPublicKey};
    use rustao::encrypt::{decrypt_payload, encrypt_payload};

    fn generate_test_keypair() -> (RsaPrivateKey, RsaPublicKey) {
        let mut rng = rand::thread_rng();
        let priv_key = RsaPrivateKey::new(&mut rng, 2048).unwrap();
        let pub_key = priv_key.to_public_key();
        (priv_key, pub_key)
    }

    #[test]
    fn test_encrypt_decrypt_roundtrip() {
        let (priv_key, pub_key) = generate_test_keypair();
        let plaintext = b"secret message";
        let (encrypted_data, encrypted_key, nonce) = encrypt_payload(plaintext, &pub_key).unwrap();
        let decrypted = decrypt_payload(&encrypted_data, &encrypted_key, &nonce, &priv_key).unwrap();
        assert_eq!(decrypted, plaintext);
    }

    #[test]
    fn test_encrypt_empty() {
        let (_, pub_key) = generate_test_keypair();  // underscore for unused
        let plaintext = b"";
        let (_encrypted_data, _encrypted_key, _nonce) = encrypt_payload(plaintext, &pub_key).unwrap();
        // just ensure it doesn't panic
    }

    #[test]
    fn test_wrong_key() {
        let (priv_key, pub_key) = generate_test_keypair();
        let (other_priv, _) = generate_test_keypair();
        let plaintext = b"secret";
        let (encrypted_data, encrypted_key, nonce) = encrypt_payload(plaintext, &pub_key).unwrap();
        let result = decrypt_payload(&encrypted_data, &encrypted_key, &nonce, &other_priv);
        assert!(result.is_err());
    }
}