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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
use std::fs::File;
use std::path::Path;
use std::io;
use std::io::{Read, Write};
pub use io::Result;
pub fn get<P: AsRef<Path>>(path: P) -> io::Result<Vec<u8>> {
let mut file = File::open(path)?;
let mut data = Vec::new();
if let Ok(meta) = file.metadata() {
data.reserve(meta.len() as usize);
}
file.read_to_end(&mut data)?;
Ok(data)
}
pub fn put<P: AsRef<Path>, Bytes: AsRef<[u8]>>(path: P, data: Bytes) -> io::Result<()> {
let mut file = File::create(path)?;
file.write_all(data.as_ref())?;
Ok(())
}
pub fn get_text<P: AsRef<Path>>(path: P) -> io::Result<String> {
let bytes = get(path)?;
String::from_utf8(bytes).map_err(|_| {
io::Error::new(io::ErrorKind::InvalidData, "file did not contain valid UTF-8")
})
}
pub fn put_text<P: AsRef<Path>, S: AsRef<str>>(path: P, data: S) -> io::Result<()> {
put(path, data.as_ref().as_bytes())
}
#[test]
fn it_works() {
let s = String::from_utf8(get(file!()).unwrap()).unwrap();
assert!(s.contains("it_works()"));
let mut tmp_name = std::env::temp_dir();
tmp_name.push("tmp_file_should_not_exist");
assert!(get(tmp_name).is_err());
let mut tmp_name = std::env::temp_dir();
tmp_name.push("tmp_file_created");
let data = vec![0u8,1,2,3];
put(&tmp_name, &data).unwrap();
assert_eq!(data, get(&tmp_name).unwrap());
put(&tmp_name, data).unwrap();
std::fs::remove_file(tmp_name).ok();
}
#[test]
fn it_works_with_text() {
let s = String::from_utf8(get(file!()).unwrap()).unwrap();
assert!(s.contains("it_works()"));
let mut tmp_name = std::env::temp_dir();
tmp_name.push("hello");
put(&tmp_name, [0x80]).unwrap();
if let Err(e) = get_text(&tmp_name) {
assert_eq!(e.kind(), io::ErrorKind::InvalidData);
} else {
panic!("Should error on invalid UTF-8")
}
let text = "Hello, World!";
put_text(&tmp_name, text).unwrap();
assert_eq!(text, get_text(&tmp_name).unwrap());
std::fs::remove_file(tmp_name).ok();
}