-
Notifications
You must be signed in to change notification settings - Fork 316
/
Copy pathread_metadata_files.rs
42 lines (31 loc) · 1.26 KB
/
read_metadata_files.rs
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
40
41
42
use failure::Error as FailureError;
use crate::parampublish::support::session::ParamPublishSessionBuilder;
#[test]
fn fails_if_missing_metadata_file() -> Result<(), FailureError> {
// missing the corresponding .meta file
let filenames = vec!["v12-aaa.vk", "v12-aaa.params"];
let (mut session, _) = ParamPublishSessionBuilder::new()
.with_session_timeout_ms(1000)
.with_files(&filenames)
.build();
session.exp_string("found 2 param files in cache dir")?;
session.exp_string("no file triples found, exiting")?;
std::fs::remove_dir_all(session._cache_dir.path())?;
Ok(())
}
#[test]
fn fails_if_malformed_metadata_file() -> Result<(), FailureError> {
// A malformed v11-aaa.meta file.
let mut malformed: &[u8] = &[42];
let (mut session, _) = ParamPublishSessionBuilder::new()
.with_session_timeout_ms(1000)
.with_files(&["v11-aaa.vk", "v11-aaa.params"])
.with_file_and_bytes("v11-aaa.meta", &mut malformed)
.build();
session.exp_string("found 3 param files in cache dir")?;
session.exp_string("found 1 file triples")?;
session.exp_string("failed to parse .meta file")?;
session.exp_string("exiting")?;
std::fs::remove_dir_all(session._cache_dir.path())?;
Ok(())
}