41 lines
1.6 KiB
Rust
41 lines
1.6 KiB
Rust
use std::{io::{self, Read, Write}, fs::File, path::PathBuf};
|
|
use assert_cmd::{Command, cargo::CargoError};
|
|
use tempfile::{NamedTempFile, TempDir};
|
|
|
|
fn doby_cmd<>() -> Result<Command, CargoError> {
|
|
const PASSWORD: &str = "the password";
|
|
let mut cmd = Command::cargo_bin("doby")?;
|
|
cmd.arg("-p").arg(PASSWORD);
|
|
Ok(cmd)
|
|
}
|
|
|
|
#[test]
|
|
fn files() -> io::Result<()> {
|
|
const PLAINTEXT: &[u8] = b"the plaintext";
|
|
|
|
let tmp_dir = TempDir::new()?;
|
|
let tmp_path = PathBuf::from(tmp_dir.path());
|
|
|
|
let mut tmp_plaintext = NamedTempFile::new_in(&tmp_dir)?;
|
|
tmp_plaintext.write_all(PLAINTEXT)?;
|
|
let tmp_ciphertext = tmp_path.join("ciphertext");
|
|
doby_cmd().unwrap().arg(tmp_plaintext.path()).arg(&tmp_ciphertext).assert().success().stdout("").stderr("");
|
|
|
|
let tmp_decrypted = tmp_path.join("decryped");
|
|
doby_cmd().unwrap().arg(tmp_ciphertext).arg(&tmp_decrypted).assert().success().stdout("").stderr("");
|
|
|
|
let mut tmp_decrypted = File::open(tmp_decrypted).unwrap();
|
|
let mut buff = [0; PLAINTEXT.len()];
|
|
assert_eq!(tmp_decrypted.read(&mut buff)?, PLAINTEXT.len());
|
|
assert_eq!(buff, PLAINTEXT);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn argon2_params() -> io::Result<()> {
|
|
Command::cargo_bin("doby").unwrap().arg("-i").arg("0").assert().stderr("Invalid argon2 params: time cost is too small\n");
|
|
Command::cargo_bin("doby").unwrap().arg("-m").arg("0").assert().stderr("Invalid argon2 params: memory cost is too small\n");
|
|
Command::cargo_bin("doby").unwrap().arg("-t").arg("0").assert().stderr("Invalid argon2 params: too few lanes\n");
|
|
Ok(())
|
|
} |