blockworks-foundation
/
solana
mirror of https://github.com/blockworks-foundation/solana.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Use temp path for append_vec tests (#4765) 

* Use temp path for serialize test

* set account path
This commit is contained in:
Sathish 2019-06-25 16:11:57 -07:00 committed by GitHub
parent f20ba423ca
commit 1f950781c2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
runtime/src/append_vec.rs
View File

@ -289,13 +289,17 @@ pub mod test_utils {
} }
} }
pub fn get_append_vec_dir() -> String {
std::env::var("OUT_DIR").unwrap_or_else(|_| "target/append_vec_tests".to_string())
}
pub fn get_append_vec_path(path: &str) -> TempFile { pub fn get_append_vec_path(path: &str) -> TempFile {
let out_dir = let out_dir = get_append_vec_dir();
std::env::var("OUT_DIR").unwrap_or_else(|_| "target/append_vec_tests".to_string());
let mut buf = PathBuf::new();
let rand_string: String = thread_rng().sample_iter(&Alphanumeric).take(30).collect(); let rand_string: String = thread_rng().sample_iter(&Alphanumeric).take(30).collect();
buf.push(&format!("{}/{}{}", out_dir, path, rand_string)); let dir = format!("{}/{}", out_dir, rand_string);
create_dir_all(out_dir).expect("Create directory failed"); let mut buf = PathBuf::new();
buf.push(&format!("{}/{}", dir, path));
create_dir_all(dir).expect("Create directory failed");
TempFile { path: buf } TempFile { path: buf }
} }
@ -477,8 +481,8 @@ pub mod tests {
#[test] #[test]
fn test_append_vec_serialize() { fn test_append_vec_serialize() {
let path = Path::new("append_vec_serialize"); let path = get_append_vec_path("test_append_serialize");
let av: AppendVec = AppendVec::new(path, true, 1024 * 1024); let av: AppendVec = AppendVec::new(Path::new(&path.path), true, 1024 * 1024);
let account1 = create_test_account(1); let account1 = create_test_account(1);
let index1 = av.append_account_test(&account1).unwrap(); let index1 = av.append_account_test(&account1).unwrap();
assert_eq!(index1, 0); assert_eq!(index1, 0);
@ -493,12 +497,13 @@ pub mod tests {
let mut writer = Cursor::new(&mut buf[..]); let mut writer = Cursor::new(&mut buf[..]);
serialize_into(&mut writer, &av).unwrap(); serialize_into(&mut writer, &av).unwrap();
AppendVec::set_account_paths(&vec![get_append_vec_dir()]);
let mut reader = Cursor::new(&mut buf[..]); let mut reader = Cursor::new(&mut buf[..]);
let dav: AppendVec = deserialize_from(&mut reader).unwrap(); let dav: AppendVec = deserialize_from(&mut reader).unwrap();
assert_eq!(dav.get_account_test(index2).unwrap(), account2); assert_eq!(dav.get_account_test(index2).unwrap(), account2);
assert_eq!(dav.get_account_test(index1).unwrap(), account1); assert_eq!(dav.get_account_test(index1).unwrap(), account1);
std::fs::remove_file(path).unwrap(); drop(dav);
let mut reader = Cursor::new(&mut buf[..]); let mut reader = Cursor::new(&mut buf[..]);
let dav: AppendVec = deserialize_from(&mut reader).unwrap(); let dav: AppendVec = deserialize_from(&mut reader).unwrap();