sequoia-sq/tests/sq-packet-dump.rs

#[cfg(test)]
mod sq_packet_dump {
    use assert_cmd::Command;
    use predicates::prelude::*;

    use openpgp::Result;
    use sequoia_openpgp as openpgp;

    fn artifact(filename: &str) -> String {
        format!("tests/data/{}", filename)
    }

    #[test]
    fn session_key_without_prefix() -> Result<()> {
        Command::cargo_bin("sq")
        .unwrap()
        .arg("--no-cert-store")
        .arg("--no-key-store")
        .arg("toolbox")
        .arg("packet")
        .arg("dump")
        .args(["--session-key", "1FE820EC21FB5D7E33D83367106D1D3747DCD48E6320C1AEC57EE7D18FC437D4"])
        .arg(artifact("messages/rsa.msg.pgp"))
        .assert()
        .success()
        .stdout(predicate::str::contains("Decryption failed").not());
        Ok(())
    }

    #[test]
    fn session_key_with_prefix() -> Result<()> {
        Command::cargo_bin("sq")
        .unwrap()
        .arg("--no-cert-store")
        .arg("--no-key-store")
        .arg("toolbox")
        .arg("packet")
        .arg("dump")
        .args(["--session-key", "9:1FE820EC21FB5D7E33D83367106D1D3747DCD48E6320C1AEC57EE7D18FC437D4"])
        .arg(artifact("messages/rsa.msg.pgp"))
        .assert()
        .success()
        .stdout(predicate::str::contains("Decryption failed").not());
        Ok(())
    }

    #[test]
    fn session_key_with_bad_prefix() -> Result<()> {
        Command::cargo_bin("sq")
        .unwrap()
        .arg("--no-cert-store")
        .arg("--no-key-store")
        .arg("toolbox")
        .arg("packet")
        .arg("dump")
        .args(["--session-key", "1:1FE820EC21FB5D7E33D83367106D1D3747DCD48E6320C1AEC57EE7D18FC437D4"])
        .arg(artifact("messages/rsa.msg.pgp"))
        .assert()
        .success()
        .stdout(predicate::str::contains("Indicated Symmetric algo: IDEA"))
        .stdout(predicate::str::contains("Decryption failed"));
        Ok(())
    }

    #[test]
    fn session_key_wrong_length_without_prefix() -> Result<()> {
        // too short
        Command::cargo_bin("sq")
        .unwrap()
        .arg("--no-cert-store")
        .arg("--no-key-store")
        .arg("toolbox")
        .arg("packet")
        .arg("dump")
        .args(["--session-key", "1FE820EC21FB5D7E33D83367106D1D3747DCD48E6320C1AEC57EE7D18FC437"])
        .arg(artifact("messages/rsa.msg.pgp"))
        .assert()
        .success()
        .stdout(predicate::str::contains("Decryption failed"));

        // too long
        Command::cargo_bin("sq")
        .unwrap()
        .arg("--no-cert-store")
        .arg("--no-key-store")
        .arg("toolbox")
        .arg("packet")
        .arg("dump")
        .args(["--session-key", "1FE820EC21FB5D7E33D83367106D1D3747DCD48E6320C1AEC57EE7D18FC437D4AB"])
        .arg(artifact("messages/rsa.msg.pgp"))
        .assert()
        .success()
        .stdout(predicate::str::contains("Decryption failed"));
        Ok(())
    }

    #[test]
    fn session_key_wrong_length_with_prefix() -> Result<()> {
        // too short
        Command::cargo_bin("sq")
        .unwrap()
        .arg("--no-cert-store")
        .arg("--no-key-store")
        .arg("toolbox")
        .arg("packet")
        .arg("dump")
        .args(["--session-key", "1:1FE820EC21FB5D7E33D83367106D1D3747DCD48E6320C1AEC57EE7D18FC437"])
        .arg(artifact("messages/rsa.msg.pgp"))
        .assert()
        .success()
        .stdout(predicate::str::contains("Decryption failed"));

        // too long
        Command::cargo_bin("sq")
        .unwrap()
        .arg("--no-cert-store")
        .arg("--no-key-store")
        .arg("toolbox")
        .arg("packet")
        .arg("dump")
        .args(["--session-key", "1:1FE820EC21FB5D7E33D83367106D1D3747DCD48E6320C1AEC57EE7D18FC437D4AB"])
        .arg(artifact("messages/rsa.msg.pgp"))
        .assert()
        .success()
        .stdout(predicate::str::contains("Decryption failed"));
        Ok(())
    }

    #[test]
    fn session_key_wrong_key_with_prefix() -> Result<()> {
        Command::cargo_bin("sq")
        .unwrap()
        .arg("--no-cert-store")
        .arg("--no-key-store")
        .arg("toolbox")
        .arg("packet")
        .arg("dump")
        .args(["--session-key", "9:BB9CCB8EDE22DC222C83BD1C63AEB97335DDC7B696DB171BD16EAA5784CC0478"])
        .arg(artifact("messages/rsa.msg.pgp"))
        .assert()
        .success()
        .stdout(predicate::str::contains("Decryption failed"));
        Ok(())
    }
}