diff options
| -rw-r--r-- | src/db.rs | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/src/db.rs b/src/db.rs index c0e4cda..eeaa27a 100644 --- a/src/db.rs +++ b/src/db.rs @@ -68,7 +68,12 @@ impl Posts { hasher.update(body); let hash = format!("{:x}", hasher.finalize()); + #[cfg(test)] + let homedir = std::env::var("PWD")?; + + #[cfg(not(test))] let homedir = std::env::var("HOME")?; + let writepath = format!("{}/.clinte.sha256", homedir); fs::write(writepath, &hash)?; @@ -163,4 +168,22 @@ mod tests { all.delete(1); all.write(); } + + #[test] + fn check_hash() { + let strposts = fs::read_to_string(PATH).unwrap(); + let mut hasher = Sha256::new(); + hasher.update(strposts); + let rhs = format!("{:x}", hasher.finalize()); + + let all = Posts::get_all(PATH); + all.hash().unwrap(); + + let pwd = std::env::var("PWD").unwrap(); + let hashpath = format!("{}/.clinte.sha256", pwd); + let lhs = fs::read_to_string(hashpath.clone()).unwrap(); + fs::remove_file(hashpath).unwrap(); + + assert_eq!(lhs, rhs); + } } |